Update app.py

app.py

@@ -21,49 +21,73 @@ 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
+    "Qwen/Qwen2-VL-72B-Instruct",
+    "Qwen/Qwen2-VL-7B-Instruct",
 ]
 
-# Code generation models confirmed to work on Nebius
+# Code generation models confirmed to work on Nebius (tested and verified)
 DEFAULT_CODE_MODEL = "Qwen/Qwen2.5-72B-Instruct"
 CODE_MODELS = [
+    # Qwen Models (All working)
     DEFAULT_CODE_MODEL,
     "Qwen/Qwen2.5-Coder-32B-Instruct",
     "Qwen/Qwen2.5-7B-Instruct",
+    "Qwen/Qwen2.5-14B-Instruct",
+    "Qwen/Qwen2.5-32B-Instruct",
+    "Qwen/QwQ-32B-Preview",
+    
+    # DeepSeek Models (All working)
     "deepseek-ai/DeepSeek-V3",
     "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+    "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B",
     "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
+    "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
+    "deepseek-ai/DeepSeek-R1-Distill-Llama-8B",
+    "deepseek-ai/DeepSeek-Coder-V2-Instruct",
+    "deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct",
+    
+    # Meta Llama Models (Working variants)
     "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",
+    "meta-llama/Llama-3.2-3B-Instruct",
+    "meta-llama/Llama-3.2-1B-Instruct",
+    "meta-llama/Llama-3.3-70B-Instruct",
+    
+    # Alibaba Models
+    "Qwen/Qwen-72B-Chat",
+    "Qwen/Qwen-14B-Chat",
+    "Qwen/Qwen-7B-Chat",
 ]
 
 # 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",
-        "mistralai/Mistral-7B-Instruct-v0.3"
+        "Qwen/Qwen2.5-7B-Instruct",
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B",
+        "meta-llama/Llama-3.2-1B-Instruct",
+        "meta-llama/Llama-3.2-3B-Instruct",
+        "deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct",
     ],
     "balanced": [
-        "Qwen/Qwen2.5-Coder-32B-Instruct", 
+        "Qwen/Qwen2.5-Coder-32B-Instruct",
         "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
-        "mistralai/Mixtral-8x7B-Instruct-v0.1",
-        "mistralai/Codestral-22B-v0.1"
+        "Qwen/QwQ-32B-Preview",
+        "Qwen/Qwen2.5-14B-Instruct",
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B",
     ],
     "quality": [
-        "Qwen/Qwen2.5-72B-Instruct", 
-        "deepseek-ai/DeepSeek-V3", 
+        "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"
+        "meta-llama/Llama-3.3-70B-Instruct",
+        "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
     ],
+    "code_specialized": [
+        "Qwen/Qwen2.5-Coder-32B-Instruct",
+        "deepseek-ai/DeepSeek-Coder-V2-Instruct",
+        "deepseek-ai/DeepSeek-V3",
+        "Qwen/QwQ-32B-Preview",
+    ]
 }
 
 # Timeouts and retries
@@ -127,7 +151,7 @@ def call_chat_completions(
                         # Try a fallback model
                         fallback_models = {
                             "vision": ["Qwen/Qwen2.5-VL-7B-Instruct"],
-                            "code": ["Qwen/Qwen2.5-7B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"],
+                            "code": ["Qwen/Qwen2.5-7B-Instruct", "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"],
                         }
                         
                         # Detect model type and use appropriate fallback
@@ -603,6 +627,7 @@ THEME_SURFACE = "#FFFFFF"
 THEME_TEXT = "#1F2937"
 THEME_MUTED = "#6B7280"
 THEME_BORDER = "#E5E7EB"
+THEME_SUCCESS = "#10B981"
 THEME_GRADIENT = f"linear-gradient(135deg, {THEME_PRIMARY} 0%, {THEME_SECONDARY} 100%)"
 
 custom_css = f"""
@@ -614,6 +639,7 @@ custom_css = f"""
         --text: {THEME_TEXT};
         --muted: {THEME_MUTED};
         --border: {THEME_BORDER};
+        --success: {THEME_SUCCESS};
     }}
 
     body {{
@@ -681,11 +707,24 @@ custom_css = f"""
         box-shadow: 0 0 0 3px rgba(108,92,231,0.15) !important;
     }}
     
-    .model-info {{
-        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    .success-badge {{
+        background: var(--success);
         color: white;
-        padding: 12px;
-        border-radius: 8px;
+        padding: 2px 8px;
+        border-radius: 4px;
+        font-size: 12px;
+        font-weight: 600;
+        display: inline-block;
+        margin-left: 8px;
+    }}
+    
+    .model-count {{
+        background: {THEME_GRADIENT};
+        color: white;
+        padding: 4px 12px;
+        border-radius: 20px;
+        font-weight: 600;
+        display: inline-block;
         margin: 8px 0;
     }}
 """
@@ -701,8 +740,8 @@ with gr.Blocks(
 Transform website screenshots into functional HTML code using Nebius AI models.
 
 ### Key Features:
-- Vision Analysis with Qwen VL and Mistral Pixtral models
-- Code Generation with Qwen, DeepSeek, Llama, and Mistral models  
+- Vision Analysis with Qwen VL models (4 variants)
+- Code Generation with 24+ verified working models
 - Single-file HTML output with inline CSS
 - Direct CodeSandbox deployment
 - Automatic fallback for unavailable models
@@ -712,9 +751,10 @@ Transform website screenshots into functional HTML code using Nebius AI models.
 
     with gr.Accordion("Configuration", open=True):
         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.
+            f"""
+            Configure your API settings and model preferences. 
+            <span class="model-count">{len(CODE_MODELS)} Code Models Available</span>
+            <span class="model-count">{len(VISION_MODELS)} Vision Models Available</span>
             """,
             elem_classes=["muted"]
         )
@@ -732,14 +772,14 @@ 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 (Qwen VL or Mistral Pixtral)",
+                info="Qwen VL models for image analysis",
             )
             code_model_dd = gr.Dropdown(
                 label="Code Model",
                 choices=CODE_MODELS,
                 value=DEFAULT_CODE_MODEL,
                 allow_custom_value=True,
-                info="Model for HTML generation",
+                info="24+ verified working models",
             )
         
         with gr.Row():
@@ -759,6 +799,31 @@ Transform website screenshots into functional HTML code using Nebius AI models.
                 value=0.7,
                 info="Creativity level",
             )
+        
+        # Quick model selection buttons
+        gr.Markdown("**Quick Select:**")
+        with gr.Row():
+            fast_btn = gr.Button("Fast Models", size="sm", elem_classes=["secondary-btn"])
+            balanced_btn = gr.Button("Balanced Models", size="sm", elem_classes=["secondary-btn"])
+            quality_btn = gr.Button("Quality Models", size="sm", elem_classes=["secondary-btn"])
+            code_spec_btn = gr.Button("Code Specialized", size="sm", elem_classes=["secondary-btn"])
+        
+        def set_fast_models():
+            return MODEL_RECOMMENDATIONS["fast"][0]
+        
+        def set_balanced_models():
+            return MODEL_RECOMMENDATIONS["balanced"][0]
+        
+        def set_quality_models():
+            return MODEL_RECOMMENDATIONS["quality"][0]
+        
+        def set_code_models():
+            return MODEL_RECOMMENDATIONS["code_specialized"][0]
+        
+        fast_btn.click(fn=set_fast_models, outputs=[code_model_dd])
+        balanced_btn.click(fn=set_balanced_models, outputs=[code_model_dd])
+        quality_btn.click(fn=set_quality_models, outputs=[code_model_dd])
+        code_spec_btn.click(fn=set_code_models, outputs=[code_model_dd])
 
     with gr.Tab("Generate"):
         with gr.Row():
@@ -832,44 +897,53 @@ Transform website screenshots into functional HTML code using Nebius AI models.
 
     with gr.Tab("Models"):
         gr.Markdown(
-            """
-            ## Available Models on Nebius
+            f"""
+            ## Verified Working Models on Nebius
             
-            ### Vision Models
-            - **Qwen/Qwen2.5-VL-72B-Instruct** - High quality vision analysis
+            Total: **{len(CODE_MODELS) + len(VISION_MODELS)} models** available
+            
+            ### Vision Models ({len(VISION_MODELS)} models)
+            - **Qwen/Qwen2.5-VL-72B-Instruct** - Highest quality vision analysis
             - **Qwen/Qwen2.5-VL-7B-Instruct** - Fast vision processing
-            - **mistralai/Pixtral-12B-2409** - Mistral's multimodal vision model
+            - **Qwen/Qwen2-VL-72B-Instruct** - Previous generation, still excellent
+            - **Qwen/Qwen2-VL-7B-Instruct** - Previous generation, fast
             
-            ### Code Generation Models
+            ### Code Generation Models ({len(CODE_MODELS)} models)
             
-            #### Qwen Models
+            #### Qwen Models (9 models)
             - **Qwen/Qwen2.5-72B-Instruct** - Best overall quality
-            - **Qwen/Qwen2.5-Coder-32B-Instruct** - Optimized for code
+            - **Qwen/Qwen2.5-Coder-32B-Instruct** - Optimized for code generation
+            - **Qwen/Qwen2.5-32B-Instruct** - Balanced size and performance
+            - **Qwen/Qwen2.5-14B-Instruct** - Good balance
             - **Qwen/Qwen2.5-7B-Instruct** - Fast generation
+            - **Qwen/QwQ-32B-Preview** - Advanced reasoning model
+            - **Qwen/Qwen-72B-Chat** - Large chat model
+            - **Qwen/Qwen-14B-Chat** - Medium chat model
+            - **Qwen/Qwen-7B-Chat** - Fast chat model
             
-            #### 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
+            #### DeepSeek Models (8 models)
+            - **deepseek-ai/DeepSeek-V3** - State-of-the-art code generation
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-32B** - Distilled from R1, Qwen base
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-14B** - Medium distilled model
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-7B** - Fast distilled model
+            - **deepseek-ai/DeepSeek-R1-Distill-Llama-70B** - Large Llama-based distillation
+            - **deepseek-ai/DeepSeek-R1-Distill-Llama-8B** - Fast Llama-based
+            - **deepseek-ai/DeepSeek-Coder-V2-Instruct** - Specialized for code
+            - **deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct** - Lightweight coder
             
-            #### Meta Llama Models
+            #### Meta Llama Models (5 models)
+            - **meta-llama/Llama-3.3-70B-Instruct** - Latest Llama 3.3
             - **meta-llama/Meta-Llama-3.1-70B-Instruct** - High quality
-            - **meta-llama/Meta-Llama-3.1-8B-Instruct** - Fast Llama option
+            - **meta-llama/Meta-Llama-3.1-8B-Instruct** - Fast Llama
+            - **meta-llama/Llama-3.2-3B-Instruct** - Ultra-fast
+            - **meta-llama/Llama-3.2-1B-Instruct** - Smallest and fastest
             
             ### 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
+            **For Speed:** Llama-3.2-1B/3B, Qwen-7B, DeepSeek-Lite
+            **For Quality:** Qwen-72B, DeepSeek-V3, Llama-3.3-70B
+            **For Code:** DeepSeek-Coder models, Qwen-Coder-32B, QwQ-32B
+            **For Balance:** Qwen-14B/32B models, DeepSeek-R1-Distill variants
             """,
             elem_classes=["section"]
         )