Update app.py

app.py

@@ -21,33 +21,38 @@ sys_prompt = read('system_prompt.txt')
     
 def process_text(text_input, unit):
     if text_input:
-        response = client.chat.completions.create(
+        completion = client.chat.completions.create(
             model=MODEL,
             messages=[
                 {"role": "system", "content": f" You are a helpful {unit} doctor." + sys_prompt},
                 {"role": "user", "content": f"Hello! Could you solve {text_input}?"}
             ]
         )
-        return response.choices[0].message.content
+        return completion.choices[0].message.content
+    return ""
+
 
-def process_image(image_input,unit):
+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")
+        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}"}
+                }
+            ]}
+        ]
         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}"}
-                    }
-                ]}
-            ],
+            messages=messages,
             temperature=0.0,
         )
         return response.choices[0].message.content
+    return ""
+
 
 def main(text_input="", image_input=None, unit=""):
     if text_input and image_input is None: