Update app.py

app.py

@@ -21,6 +21,7 @@ DEFAULT_VISION_MODEL = "Qwen/Qwen2.5-VL-72B-Instruct"
 VISION_MODELS = [
     DEFAULT_VISION_MODEL,
     "Qwen/Qwen2.5-VL-7B-Instruct",
+    "mistralai/Pixtral-12B-2409",  # Mistral's vision model
 ]
 
 # Code generation models confirmed to work on Nebius
@@ -34,13 +35,35 @@ CODE_MODELS = [
     "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
     "meta-llama/Meta-Llama-3.1-70B-Instruct",
     "meta-llama/Meta-Llama-3.1-8B-Instruct",
+    "mistralai/Mixtral-8x22B-Instruct-v0.1",
+    "mistralai/Mixtral-8x7B-Instruct-v0.1",
+    "mistralai/Mistral-7B-Instruct-v0.3",
+    "mistralai/Mistral-Nemo-Instruct-2407",
+    "mistralai/Codestral-22B-v0.1",
+    "mistralai/Mistral-Large-Instruct-2407",
 ]
 
-# Performance recommendations based on working models
+# Model recommendations for different use cases
 MODEL_RECOMMENDATIONS = {
-    "fast": ["Qwen/Qwen2.5-7B-Instruct", "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B", "meta-llama/Meta-Llama-3.1-8B-Instruct"],
-    "balanced": ["Qwen/Qwen2.5-Coder-32B-Instruct", "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"],
-    "quality": ["Qwen/Qwen2.5-72B-Instruct", "deepseek-ai/DeepSeek-V3", "meta-llama/Meta-Llama-3.1-70B-Instruct"],
+    "fast": [
+        "Qwen/Qwen2.5-7B-Instruct", 
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B", 
+        "meta-llama/Meta-Llama-3.1-8B-Instruct",
+        "mistralai/Mistral-7B-Instruct-v0.3"
+    ],
+    "balanced": [
+        "Qwen/Qwen2.5-Coder-32B-Instruct", 
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+        "mistralai/Mixtral-8x7B-Instruct-v0.1",
+        "mistralai/Codestral-22B-v0.1"
+    ],
+    "quality": [
+        "Qwen/Qwen2.5-72B-Instruct", 
+        "deepseek-ai/DeepSeek-V3", 
+        "meta-llama/Meta-Llama-3.1-70B-Instruct",
+        "mistralai/Mixtral-8x22B-Instruct-v0.1",
+        "mistralai/Mistral-Large-Instruct-2407"
+    ],
 }
 
 # Timeouts and retries
@@ -104,7 +127,7 @@ def call_chat_completions(
                         # Try a fallback model
                         fallback_models = {
                             "vision": ["Qwen/Qwen2.5-VL-7B-Instruct"],
-                            "code": ["Qwen/Qwen2.5-7B-Instruct"],
+                            "code": ["Qwen/Qwen2.5-7B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"],
                         }
                         
                         # Detect model type and use appropriate fallback
@@ -570,19 +593,6 @@ def export_html_to_file(html_code: str) -> Optional[str]:
         return None
 
 
-def get_model_recommendations(performance_tier: str) -> Tuple[str, str]:
-    """
-    Get recommended models based on performance tier.
-    Returns (vision_model, code_model)
-    """
-    tier_map = {
-        "Fast (7B-8B models)": ("Qwen/Qwen2.5-VL-7B-Instruct", "Qwen/Qwen2.5-7B-Instruct"),
-        "Balanced (32B models)": ("Qwen/Qwen2.5-VL-7B-Instruct", "Qwen/Qwen2.5-Coder-32B-Instruct"),
-        "Quality (70B+ models)": ("Qwen/Qwen2.5-VL-72B-Instruct", "Qwen/Qwen2.5-72B-Instruct"),
-    }
-    return tier_map.get(performance_tier, (DEFAULT_VISION_MODEL, DEFAULT_CODE_MODEL))
-
-
 # =========================
 # Gradio UI
 # =========================
@@ -670,6 +680,14 @@ custom_css = f"""
         border-color: {THEME_PRIMARY} !important;
         box-shadow: 0 0 0 3px rgba(108,92,231,0.15) !important;
     }}
+    
+    .model-info {{
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        color: white;
+        padding: 12px;
+        border-radius: 8px;
+        margin: 8px 0;
+    }}
 """
 
 with gr.Blocks(
@@ -683,8 +701,8 @@ with gr.Blocks(
 Transform website screenshots into functional HTML code using Nebius AI models.
 
 ### Key Features:
-- Vision Analysis with Qwen VL models
-- Code Generation with Qwen, DeepSeek, and Llama models  
+- Vision Analysis with Qwen VL and Mistral Pixtral models
+- Code Generation with Qwen, DeepSeek, Llama, and Mistral models  
 - Single-file HTML output with inline CSS
 - Direct CodeSandbox deployment
 - Automatic fallback for unavailable models
@@ -696,23 +714,17 @@ Transform website screenshots into functional HTML code using Nebius AI models.
         gr.Markdown(
             """
             Configure your API settings and model preferences. A default API key is provided for testing.
+            Select from a wide range of Mistral, Qwen, DeepSeek, and Llama models.
             """,
             elem_classes=["muted"]
         )
         
-        with gr.Row():
-            nebius_key = gr.Textbox(
-                label="Nebius API Key",
-                type="password",
-                placeholder="Enter your Nebius API key or use default",
-                value=DEFAULT_NEBIUS_API_KEY,
-            )
-            performance_tier = gr.Radio(
-                label="Performance Tier",
-                choices=["Fast (7B-8B models)", "Balanced (32B models)", "Quality (70B+ models)"],
-                value="Balanced (32B models)",
-                info="Select based on speed vs quality preference"
-            )
+        nebius_key = gr.Textbox(
+            label="Nebius API Key",
+            type="password",
+            placeholder="Enter your Nebius API key or use default",
+            value=DEFAULT_NEBIUS_API_KEY,
+        )
         
         with gr.Row():
             vision_model_dd = gr.Dropdown(
@@ -720,7 +732,7 @@ Transform website screenshots into functional HTML code using Nebius AI models.
                 choices=VISION_MODELS,
                 value=DEFAULT_VISION_MODEL,
                 allow_custom_value=True,
-                info="Model for image analysis",
+                info="Model for image analysis (Qwen VL or Mistral Pixtral)",
             )
             code_model_dd = gr.Dropdown(
                 label="Code Model",
@@ -747,12 +759,6 @@ Transform website screenshots into functional HTML code using Nebius AI models.
                 value=0.7,
                 info="Creativity level",
             )
-        
-        performance_tier.change(
-            fn=get_model_recommendations,
-            inputs=[performance_tier],
-            outputs=[vision_model_dd, code_model_dd]
-        )
 
     with gr.Tab("Generate"):
         with gr.Row():
@@ -827,26 +833,43 @@ Transform website screenshots into functional HTML code using Nebius AI models.
     with gr.Tab("Models"):
         gr.Markdown(
             """
-            ## Working Models on Nebius
+            ## Available Models on Nebius
             
             ### Vision Models
             - **Qwen/Qwen2.5-VL-72B-Instruct** - High quality vision analysis
             - **Qwen/Qwen2.5-VL-7B-Instruct** - Fast vision processing
+            - **mistralai/Pixtral-12B-2409** - Mistral's multimodal vision model
+            
+            ### Code Generation Models
             
-            ### Code Generation Models  
+            #### Qwen Models
             - **Qwen/Qwen2.5-72B-Instruct** - Best overall quality
             - **Qwen/Qwen2.5-Coder-32B-Instruct** - Optimized for code
             - **Qwen/Qwen2.5-7B-Instruct** - Fast generation
+            
+            #### DeepSeek Models
             - **deepseek-ai/DeepSeek-V3** - Advanced code generation
             - **deepseek-ai/DeepSeek-R1-Distill-Qwen-32B** - Balanced performance
             - **deepseek-ai/DeepSeek-R1-Distill-Qwen-7B** - Fast alternative
+            
+            #### Mistral Models
+            - **mistralai/Mistral-Large-Instruct-2407** - Mistral's flagship model
+            - **mistralai/Mixtral-8x22B-Instruct-v0.1** - Large MOE model
+            - **mistralai/Mixtral-8x7B-Instruct-v0.1** - Efficient MOE architecture
+            - **mistralai/Codestral-22B-v0.1** - Specialized for code generation
+            - **mistralai/Mistral-Nemo-Instruct-2407** - Balanced performance
+            - **mistralai/Mistral-7B-Instruct-v0.3** - Fast and efficient
+            
+            #### Meta Llama Models
             - **meta-llama/Meta-Llama-3.1-70B-Instruct** - High quality
             - **meta-llama/Meta-Llama-3.1-8B-Instruct** - Fast Llama option
             
-            ### Tips
-            - Start with Balanced tier for best results
-            - Use Fast tier for quick prototypes
-            - Quality tier for production-ready code
+            ### Recommendations by Use Case
+            
+            **For Speed:** Qwen 7B, Mistral 7B, DeepSeek 7B
+            **For Quality:** Qwen 72B, Mistral Large, DeepSeek V3
+            **For Code:** Qwen Coder, Codestral, DeepSeek models
+            **For Balance:** Mixtral models, Qwen 32B models
             """,
             elem_classes=["section"]
         )