Update app.py

app.py

@@ -114,18 +114,27 @@ MODELS = [
         ("Hugging Face: Zephyr 7B", "huggingfaceh4/zephyr-7b-beta:free", 4096),
         ("MythoMax 13B", "gryphe/mythomax-l2-13b:free", 4096),
     ]},
-    
+
     # Vision-capable Models
     {"category": "Vision Models", "models": [
-        ("Meta: Llama 3.2 11B Vision Instruct", "meta-llama/llama-3.2-11b-vision-instruct:free", 131072),
-        ("Qwen: Qwen2.5 VL 72B Instruct", "qwen/qwen2.5-vl-72b-instruct:free", 131072),
-        ("Qwen: Qwen2.5 VL 32B Instruct", "qwen/qwen2.5-vl-32b-instruct:free", 8192),
-        ("Qwen: Qwen2.5-VL 7B Instruct", "qwen/qwen-2.5-vl-7b-instruct:free", 64000),
-        ("Qwen: Qwen2.5 VL 3B Instruct", "qwen/qwen2.5-vl-3b-instruct:free", 64000),
         ("Google: Gemini Pro 2.0 Experimental", "google/gemini-2.0-pro-exp-02-05:free", 2000000),
-        ("Google: Gemini Pro 2.5 Experimental", "google/gemini-2.5-pro-exp-03-25:free", 1000000),
-        ("Google: Gemini 2.0 Flash Thinking Experimental", "google/gemini-2.0-flash-thinking-exp:free", 1048576),
+        ("Google: Gemini 2.0 Flash Thinking Experimental 01-21", "google/gemini-2.0-flash-thinking-exp:free", 1048576),
         ("Google: Gemini Flash 2.0 Experimental", "google/gemini-2.0-flash-exp:free", 1048576),
+        ("Google: Gemini Pro 2.5 Experimental", "google/gemini-2.5-pro-exp-03-25:free", 1000000),
+        ("Google: Gemini Flash 1.5 8B Experimental", "google/gemini-flash-1.5-8b-exp", 1000000),
+        ("Google: Gemma 3 4B", "google/gemma-3-4b-it:free", 131072),
+        ("Google: Gemma 3 12B", "google/gemma-3-12b-it:free", 131072),
+        ("Qwen: Qwen2.5 VL 72B Instruct", "qwen/qwen2.5-vl-72b-instruct:free", 131072),
+        ("Meta: Llama 3.2 11B Vision Instruct", "meta-llama/llama-3.2-11b-vision-instruct:free", 131072),
+        ("Mistral: Mistral Small 3.1 24B", "mistralai/mistral-small-3.1-24b-instruct:free", 96000),
+        ("Google: Gemma 3 27B", "google/gemma-3-27b-it:free", 96000),
+        ("Qwen: Qwen2.5 VL 3B Instruct", "qwen/qwen2.5-vl-3b-instruct:free", 64000),
+        ("Qwen: Qwen2.5-VL 7B Instruct", "qwen/qwen-2.5-vl-7b-instruct:free", 64000),
+        ("Google: LearnLM 1.5 Pro Experimental", "google/learnlm-1.5-pro-experimental:free", 40960),
+        ("Google: Gemini 2.0 Flash Thinking Experimental", "google/gemini-2.0-flash-thinking-exp-1219:free", 40000),
+        ("Bytedance: UI-TARS 72B", "bytedance-research/ui-tars-72b:free", 32768),
+        ("Google: Gemma 3 1B", "google/gemma-3-1b-it:free", 32768),
+        ("Qwen: Qwen2.5 VL 32B Instruct", "qwen/qwen2.5-vl-32b-instruct:free", 8192),
         ("AllenAI: Molmo 7B D", "allenai/molmo-7b-d:free", 4096),
     ]},
 ]
@@ -349,7 +358,7 @@ def ask_ai(message, chatbot, model_choice, temperature, max_tokens, top_p,
             headers={
                 "Content-Type": "application/json",
                 "Authorization": f"Bearer {OPENROUTER_API_KEY}",
-                "HTTP-Referer": "https://huggingface.co/spaces"
+                "HTTP-Referer": "https://huggingface.co/spaces/cstr/CrispStrobe"
             },
             json=payload,
             timeout=180,  # Longer timeout for document processing and streaming
@@ -382,13 +391,33 @@ def ask_ai(message, chatbot, model_choice, temperature, max_tokens, top_p,
         
         elif response.status_code == 200:
             # Handle normal response
-            result = response.json()
-            ai_response = result.get("choices", [{}])[0].get("message", {}).get("content", "")
-            chatbot = chatbot + [[message, ai_response]]
-            
-            # Log token usage if available
-            if "usage" in result:
-                logger.info(f"Token usage: {result['usage']}")
+            try:
+                result = response.json()
+                # Add detailed logging to debug what's in the response
+                logger.info(f"Response content: {result}")
+                
+                ai_response = ""
+                if "choices" in result and len(result["choices"]) > 0:
+                    if "message" in result["choices"][0]:
+                        ai_response = result["choices"][0]["message"].get("content", "")
+                    elif "delta" in result["choices"][0]:
+                        ai_response = result["choices"][0]["delta"].get("content", "")
+                    else:
+                        logger.error(f"Unexpected response structure: {result}")
+                        ai_response = "Error: Unexpected response structure from API"
+                else:
+                    logger.error(f"No choices in response: {result}")
+                    ai_response = "Error: No response received from the model"
+                
+                chatbot = chatbot + [[message, ai_response]]
+                
+                # Log token usage if available
+                if "usage" in result:
+                    logger.info(f"Token usage: {result['usage']}")
+            except Exception as e:
+                logger.error(f"Error processing response: {str(e)}")
+                logger.error(f"Response raw text: {response.text}")
+                chatbot = chatbot + [[message, f"Error processing response: {str(e)}"]]
         else:
             response_text = response.text
             logger.info(f"Error response body: {response_text}")
@@ -435,16 +464,29 @@ def update_context_display(model_name):
     return "Unknown"
 
 # Helper function for model info display
+# Update the model info display function to indicate vision capability
 def update_model_info(model_name):
     for model in ALL_MODELS:
         if model[0] == model_name:
             name, model_id, context_size = model
+            
+            # Check if this is a vision model
+            is_vision_model = False
+            for cat in MODELS:
+                if cat["category"] == "Vision Models":
+                    if any(m[0] == model_name for m in cat["models"]):
+                        is_vision_model = True
+                        break
+            
+            vision_badge = '<span style="background-color: #4CAF50; color: white; padding: 3px 6px; border-radius: 3px; font-size: 0.8em; margin-left: 5px;">Vision</span>' if is_vision_model else ''
+            
             return f"""
             <div class="model-info">
-                <h3>{name}</h3>
+                <h3>{name} {vision_badge}</h3>
                 <p><strong>Model ID:</strong> {model_id}</p>
                 <p><strong>Context Size:</strong> {context_size:,} tokens</p>
                 <p><strong>Provider:</strong> {model_id.split('/')[0]}</p>
+                {f'<p><strong>Features:</strong> Supports image understanding</p>' if is_vision_model else ''}
             </div>
             """
     return "<p>Model information not available</p>"