Update app.py

app.py

@@ -19,9 +19,14 @@ def generate_response(input_text, image, openai_api_key, reasoning_effort="mediu
 
     # Prepare the messages for OpenAI API
     if model_choice == "o1":
-        messages = [
-            {"role": "user", "content": [{"type": "image_url", "image_url": {"url": input_text}}]}
-        ]
+        if image:
+            messages = [
+                {"role": "user", "content": [{"type": "image_url", "image_url": {"url": input_text}}]}
+            ]
+        else:
+            messages = [
+                {"role": "user", "content": [{"type": "text", "text": input_text}]}
+            ]
     elif model_choice == "o3-mini":
         messages = [
             {"role": "user", "content": [{"type": "text", "text": input_text}]}