Update app.py

app.py

@@ -2,8 +2,9 @@ import gradio as gr
 import requests
 import json
 import os
+import base64
 
-# Set OpenRouter API key in the Space's secrets as "OPENROUTER_API_KEY"
+# Set your OpenRouter API key in the Space's secrets as "OPENROUTER_API_KEY"
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
 HEADERS = {
     "Authorization": f"Bearer {OPENROUTER_API_KEY}",
@@ -19,19 +20,16 @@ FREE_MODELS = {
     "Qwen: Qwen2.5 VL 72B Instruct (free)": ("qwen/qwen2.5-vl-72b-instruct:free", 131072),
 }
 
-
 def query_openrouter_model(model_id, prompt, image=None):
-    messages = [{"role": "user", "content": prompt}]
-
-    # If image is included, add it to the message content as a dict
+    # If there's an image, embed it in the prompt as text
     if image is not None:
         with open(image, "rb") as f:
-            image_bytes = f.read()
-            base64_image = base64.b64encode(image_bytes).decode("utf-8")
-        messages[0]["content"] = [
-            {"type": "text", "text": prompt},
-            {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{base64_image}"}}
-        ]
+            b64_img = base64.b64encode(f.read()).decode("utf-8")
+        prompt += f"\n[User attached an image: data:image/png;base64,{b64_img}]"
+
+    messages = [
+        {"role": "user", "content": prompt}
+    ]
 
     payload = {
         "model": model_id,
@@ -51,15 +49,13 @@ def query_openrouter_model(model_id, prompt, image=None):
     except Exception as e:
         return f"Error: {str(e)}\n{response.text}"
 
-
 def chat_interface(prompt, image, model_label):
     model_id, _ = FREE_MODELS[model_label]
     return query_openrouter_model(model_id, prompt, image)
 
-
 with gr.Blocks(title="CrispChat") as demo:
     gr.Markdown("""
-    # 🌟 CrispChat 
+    # 🌟 CrispChat
     Multi-modal chat with free OpenRouter models
     """)
 
@@ -79,4 +75,5 @@ with gr.Blocks(title="CrispChat") as demo:
     submit.click(fn=chat_interface, inputs=[prompt, image, model_choice], outputs=output)
 
 if __name__ == "__main__":
-    demo.launch()
+    # Hide the API documentation to avoid schema issues
+    demo.launch(show_api=False)