Update app.py

app.py

@@ -32,23 +32,32 @@ def process_text(text_input, unit):
         return completion.choices[0].message.content
     return ""
 
+def encode_image_to_base64(image):
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return img_str
 
 def process_image(image_input, unit):
     if image_input is not None:
-        with open(image_input.name, "rb") as f:
-            base64_image = base64.b64encode(f.read()).decode("utf-8")
+        #with open(image_input.name, "rb") as f:
+        #    base64_image = base64.b64encode(f.read()).decode("utf-8")
+        base64_image = encode_image_to_base64(image)
         response = client.chat.completions.create(
             model=MODEL,
             messages=[
                 {"role": "system", "content": f" You are a helpful {unit} doctor." + sys_prompt},
                 {"role": "user", "content": [
                     {"type": "text", "text": "Help me understand what this image"},
-                    {"type": "image_url", "image_url": {
-                        "url": f"data:image/png;base64,{base64_image}"}
+                    {"type": "image_url", 
+                     "image_url": {
+                        "url": f"data:image/jpeg;base64,{base64_image}",
+                        "detail":"low"}
                     }
                 ]}
             ],
             temperature=0.0,
+            max_tokens=1024,
         )
         print(response)
         return response.choices[0].message.content