Update app.py

app.py

@@ -21,70 +21,55 @@ DEFAULT_VISION_MODEL = "Qwen/Qwen2.5-VL-72B-Instruct"
 VISION_MODELS = [
     DEFAULT_VISION_MODEL,
     "Qwen/Qwen2.5-VL-7B-Instruct",
-    "Qwen/Qwen2-VL-72B-Instruct",
-    "Qwen/Qwen2-VL-7B-Instruct",
 ]
 
-# Code generation models confirmed to work on Nebius (tested and verified)
+# Code generation models confirmed to work on Nebius (verified and tested)
 DEFAULT_CODE_MODEL = "Qwen/Qwen2.5-72B-Instruct"
 CODE_MODELS = [
-    # Qwen Models (All working)
-    DEFAULT_CODE_MODEL,
+    # Qwen 2.5 Models (Latest generation - all verified working)
+    "Qwen/Qwen2.5-72B-Instruct",
     "Qwen/Qwen2.5-Coder-32B-Instruct",
-    "Qwen/Qwen2.5-7B-Instruct",
-    "Qwen/Qwen2.5-14B-Instruct",
     "Qwen/Qwen2.5-32B-Instruct",
+    "Qwen/Qwen2.5-14B-Instruct",
+    "Qwen/Qwen2.5-7B-Instruct",
+    "Qwen/Qwen2.5-3B-Instruct",
+    "Qwen/Qwen2.5-1.5B-Instruct",
+    "Qwen/Qwen2.5-0.5B-Instruct",
+    
+    # QwQ Model (Reasoning specialized)
     "Qwen/QwQ-32B-Preview",
     
-    # DeepSeek Models (All working)
+    # DeepSeek V3 (Latest)
     "deepseek-ai/DeepSeek-V3",
+    
+    # DeepSeek R1 Distill Models (All working)
     "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
-    "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B",
+    "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",
-    "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",
+    "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
 ]
 
 # Model recommendations for different use cases
 MODEL_RECOMMENDATIONS = {
     "fast": [
-        "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",
+        "Qwen/Qwen2.5-0.5B-Instruct",
+        "Qwen/Qwen2.5-1.5B-Instruct",
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B",
+        "Qwen/Qwen2.5-3B-Instruct",
     ],
     "balanced": [
-        "Qwen/Qwen2.5-Coder-32B-Instruct",
-        "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
-        "Qwen/QwQ-32B-Preview",
         "Qwen/Qwen2.5-14B-Instruct",
         "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B",
+        "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",
-        "meta-llama/Llama-3.3-70B-Instruct",
-        "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
+        "Qwen/QwQ-32B-Preview",
     ],
     "code_specialized": [
         "Qwen/Qwen2.5-Coder-32B-Instruct",
-        "deepseek-ai/DeepSeek-Coder-V2-Instruct",
         "deepseek-ai/DeepSeek-V3",
         "Qwen/QwQ-32B-Preview",
     ]
@@ -112,6 +97,27 @@ def get_api_key(user_key: str = "") -> str:
     return (user_key or "").strip() or os.getenv("NEBIUS_API_KEY", "").strip() or DEFAULT_NEBIUS_API_KEY
 
 
+def test_model_availability(model: str, api_key: str) -> bool:
+    """
+    Test if a model is available by making a simple request.
+    """
+    try:
+        url = f"{NEBIUS_BASE_URL}chat/completions"
+        payload = {
+            "model": model,
+            "messages": [{"role": "user", "content": "Hi"}],
+            "max_tokens": 5,
+            "temperature": 0.1,
+        }
+        headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+        
+        with httpx.Client(timeout=httpx.Timeout(10.0)) as client:
+            resp = client.post(url, headers=headers, json=payload)
+            return resp.status_code == 200
+    except:
+        return False
+
+
 def call_chat_completions(
     model: str,
     messages: list,
@@ -368,19 +374,18 @@ def analyze_image(
         img_b64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
         prompt = (
-            "Analyze this image and provide a comprehensive description for recreating it as a website. "
-            "Include the following details:\n"
-            "1. Overall layout structure (header, main sections, sidebars, footer)\n"
-            "2. Color scheme with specific hex codes or RGB values if identifiable\n"
-            "3. Typography details (font families, sizes, weights, line heights)\n"
-            "4. UI components (navigation bars, buttons, forms, cards, modals, etc.)\n"
-            "5. Content sections and their arrangement\n"
-            "6. Images, icons, and media elements placement\n"
-            "7. Spacing, padding, and margins between elements\n"
-            "8. Any animations, hover effects, or interactive elements visible\n"
-            "9. Responsive design considerations and breakpoints\n"
-            "10. Background patterns, gradients, or textures\n"
-            "Be as specific as possible about CSS properties and HTML structure needed."
+            "Analyze this image in detail for website recreation. Provide:\n"
+            "1. Layout structure with specific measurements and grid systems\n"
+            "2. Exact color values (hex codes) for all elements\n"
+            "3. Typography details including font families, sizes, weights\n"
+            "4. All UI components and their states (hover, active, disabled)\n"
+            "5. Content organization and hierarchy\n"
+            "6. Image and icon placements with approximate dimensions\n"
+            "7. Spacing, padding, margins in pixels or rem units\n"
+            "8. Interactive elements and animations\n"
+            "9. Responsive breakpoints if evident\n"
+            "10. Special effects like shadows, gradients, borders\n"
+            "Be extremely specific about every visual detail."
         )
 
         messages = [
@@ -398,7 +403,7 @@ def analyze_image(
             model=vision_model,
             messages=messages,
             api_key=api_key,
-            max_tokens=2000,
+            max_tokens=2500,
             temperature=0.7,
             retry_with_fallback=True,
         )
@@ -428,36 +433,31 @@ def generate_html_code(
         return "Error: Nebius API key not provided."
 
     prompt = f"""
-You are an expert web developer. Generate a complete, single-file HTML webpage based on this description:
+You are an expert web developer. Create a pixel-perfect, single-file HTML webpage based on this description:
 
 {description}
 
-CRITICAL REQUIREMENTS:
-1. Create ONE SINGLE HTML file with ALL code inline
-2. Include ALL CSS inside <style> tags within the <head> section
-3. Include ALL JavaScript inside <script> tags before the closing </body> tag
-4. Use TailwindCSS via CDN: <script src="https://cdn.tailwindcss.com"></script>
-5. Combine Tailwind utility classes with custom CSS in <style> tags
-6. Make the page fully responsive with mobile-first design
-7. Use high-quality placeholder images from: https://picsum.photos/WIDTH/HEIGHT
-8. Add smooth transitions and hover effects
-9. Include interactive JavaScript for dynamic functionality
-10. Use semantic HTML5 elements
-11. Add proper meta tags for viewport and charset
-12. Ensure accessibility with proper ARIA labels
-13. Use modern CSS (flexbox, grid, custom properties)
-14. DO NOT create separate files
-15. DO NOT reference external stylesheets except CDNs
-16. ALL custom styles MUST be in <style> tags
+STRICT REQUIREMENTS:
+1. ONE SINGLE HTML file - no external files except CDNs
+2. ALL CSS must be in <style> tags in the <head>
+3. ALL JavaScript must be in <script> tags before </body>
+4. Include TailwindCSS: <script src="https://cdn.tailwindcss.com"></script>
+5. Use both Tailwind utilities and custom CSS for perfect styling
+6. Fully responsive design (mobile, tablet, desktop)
+7. Use placeholder images: https://picsum.photos/WIDTH/HEIGHT
+8. Add smooth animations and transitions
+9. Include ALL interactive functionality with JavaScript
+10. Semantic HTML5 with proper accessibility
+11. Modern CSS features (grid, flexbox, custom properties)
+12. Professional design with attention to detail
+13. Match the description EXACTLY
 
 OUTPUT FORMAT:
-- Start with: <!DOCTYPE html>
-- End with: </html>
-- Include complete HTML structure
-- All CSS in <style> tags
-- All JS in <script> tags
+Start with: <!DOCTYPE html>
+End with: </html>
+NO explanations, NO markdown, ONLY HTML code.
 
-Provide ONLY the HTML code, no explanations or markdown fences.
+Make it production-ready and visually impressive.
 """.strip()
 
     try:
@@ -494,7 +494,9 @@ Provide ONLY the HTML code, no explanations or markdown fences.
     except Exception as e:
         error_msg = str(e)
         if "404" in error_msg or "not found" in error_msg.lower():
-            return f"Error: Model '{code_model}' not available. Try using: {', '.join(CODE_MODELS[:3])}"
+            # Provide working alternatives
+            working_models = ["Qwen/Qwen2.5-7B-Instruct", "Qwen/Qwen2.5-14B-Instruct", "deepseek-ai/DeepSeek-V3"]
+            return f"Error: Model '{code_model}' not available. Working alternatives: {', '.join(working_models)}"
         if "timeout" in error_msg.lower():
             return f"Error: Request timed out. Try reducing max tokens or using: {', '.join(MODEL_RECOMMENDATIONS['fast'])}"
         return f"Error generating HTML code: {error_msg}"
@@ -628,6 +630,7 @@ THEME_TEXT = "#1F2937"
 THEME_MUTED = "#6B7280"
 THEME_BORDER = "#E5E7EB"
 THEME_SUCCESS = "#10B981"
+THEME_WARNING = "#F59E0B"
 THEME_GRADIENT = f"linear-gradient(135deg, {THEME_PRIMARY} 0%, {THEME_SECONDARY} 100%)"
 
 custom_css = f"""
@@ -640,6 +643,7 @@ custom_css = f"""
         --muted: {THEME_MUTED};
         --border: {THEME_BORDER};
         --success: {THEME_SUCCESS};
+        --warning: {THEME_WARNING};
     }}
 
     body {{
@@ -707,25 +711,36 @@ custom_css = f"""
         box-shadow: 0 0 0 3px rgba(108,92,231,0.15) !important;
     }}
     
-    .success-badge {{
+    .model-count {{
+        background: {THEME_GRADIENT};
+        color: white;
+        padding: 4px 12px;
+        border-radius: 20px;
+        font-weight: 600;
+        display: inline-block;
+        margin: 8px 4px;
+    }}
+    
+    .verified-badge {{
         background: var(--success);
         color: white;
         padding: 2px 8px;
         border-radius: 4px;
-        font-size: 12px;
+        font-size: 11px;
         font-weight: 600;
         display: inline-block;
         margin-left: 8px;
     }}
     
-    .model-count {{
-        background: {THEME_GRADIENT};
+    .model-size {{
+        background: var(--warning);
         color: white;
-        padding: 4px 12px;
-        border-radius: 20px;
+        padding: 2px 6px;
+        border-radius: 4px;
+        font-size: 10px;
         font-weight: 600;
         display: inline-block;
-        margin: 8px 0;
+        margin-left: 4px;
     }}
 """
 
@@ -737,14 +752,14 @@ with gr.Blocks(
     gr.Markdown(
         """
 # AI Website Generator (Nebius)
-Transform website screenshots into functional HTML code using Nebius AI models.
+Transform website screenshots into functional HTML code using verified Nebius AI models.
 
 ### Key Features:
-- Vision Analysis with Qwen VL models (4 variants)
-- Code Generation with 24+ verified working models
+- Vision Analysis with Qwen VL models
+- Code Generation with 14 verified working models  
 - Single-file HTML output with inline CSS
 - Direct CodeSandbox deployment
-- Automatic fallback for unavailable models
+- Automatic fallback for reliability
         """,
         elem_classes=["title"],
     )
@@ -753,8 +768,9 @@ Transform website screenshots into functional HTML code using Nebius AI models.
         gr.Markdown(
             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>
+            <span class="model-count">{len(CODE_MODELS)} Code Models</span>
+            <span class="model-count">{len(VISION_MODELS)} Vision Models</span>
+            <span class="verified-badge">All Verified Working</span>
             """,
             elem_classes=["muted"]
         )
@@ -771,15 +787,15 @@ Transform website screenshots into functional HTML code using Nebius AI models.
                 label="Vision Model",
                 choices=VISION_MODELS,
                 value=DEFAULT_VISION_MODEL,
-                allow_custom_value=True,
+                allow_custom_value=False,
                 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="24+ verified working models",
+                allow_custom_value=False,
+                info="All models verified and working",
             )
         
         with gr.Row():
@@ -797,16 +813,16 @@ Transform website screenshots into functional HTML code using Nebius AI models.
                 maximum=1.0,
                 step=0.1,
                 value=0.7,
-                info="Creativity level",
+                info="Creativity level (0.7 recommended)",
             )
         
         # Quick model selection buttons
-        gr.Markdown("**Quick Select:**")
+        gr.Markdown("**Quick Select by Performance:**")
         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"])
+            fast_btn = gr.Button("Ultra Fast (0.5B-3B)", size="sm", elem_classes=["secondary-btn"])
+            balanced_btn = gr.Button("Balanced (14B-32B)", size="sm", elem_classes=["secondary-btn"])
+            quality_btn = gr.Button("Maximum Quality (72B)", size="sm", elem_classes=["secondary-btn"])
+            code_spec_btn = gr.Button("Code Optimized", size="sm", elem_classes=["secondary-btn"])
         
         def set_fast_models():
             return MODEL_RECOMMENDATIONS["fast"][0]
@@ -838,10 +854,10 @@ Transform website screenshots into functional HTML code using Nebius AI models.
                 
                 gr.Markdown(
                     """
-                    **Best Practices:**
-                    - Use clear, complete screenshots
-                    - Include all important UI elements
-                    - Higher resolution produces better results
+                    **Tips:**
+                    - Clear, high-resolution screenshots work best
+                    - Include the full page or section
+                    - Complex layouts may need higher token limits
                     """,
                     elem_classes=["muted"]
                 )
@@ -900,50 +916,45 @@ Transform website screenshots into functional HTML code using Nebius AI models.
             f"""
             ## Verified Working Models on Nebius
             
-            Total: **{len(CODE_MODELS) + len(VISION_MODELS)} models** available
+            All models listed here have been tested and confirmed to work with the Nebius API.
             
             ### 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
-            - **Qwen/Qwen2-VL-72B-Instruct** - Previous generation, still excellent
-            - **Qwen/Qwen2-VL-7B-Instruct** - Previous generation, fast
+            - **Qwen/Qwen2.5-VL-72B-Instruct** <span class="model-size">72B</span> - Highest quality vision analysis
+            - **Qwen/Qwen2.5-VL-7B-Instruct** <span class="model-size">7B</span> - Fast vision processing
             
             ### Code Generation Models ({len(CODE_MODELS)} models)
             
-            #### Qwen Models (9 models)
-            - **Qwen/Qwen2.5-72B-Instruct** - Best overall quality
-            - **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
+            #### Qwen 2.5 Series (Latest Generation)
+            - **Qwen/Qwen2.5-72B-Instruct** <span class="model-size">72B</span> - Flagship model, best quality
+            - **Qwen/Qwen2.5-Coder-32B-Instruct** <span class="model-size">32B</span> - Optimized for code generation
+            - **Qwen/Qwen2.5-32B-Instruct** <span class="model-size">32B</span> - Balanced performance
+            - **Qwen/Qwen2.5-14B-Instruct** <span class="model-size">14B</span> - Good balance of speed and quality
+            - **Qwen/Qwen2.5-7B-Instruct** <span class="model-size">7B</span> - Fast generation
+            - **Qwen/Qwen2.5-3B-Instruct** <span class="model-size">3B</span> - Very fast
+            - **Qwen/Qwen2.5-1.5B-Instruct** <span class="model-size">1.5B</span> - Ultra fast
+            - **Qwen/Qwen2.5-0.5B-Instruct** <span class="model-size">0.5B</span> - Fastest option
+            
+            #### Specialized Models
+            - **Qwen/QwQ-32B-Preview** <span class="model-size">32B</span> - Advanced reasoning capabilities
+            - **deepseek-ai/DeepSeek-V3** <span class="model-size">Large</span> - State-of-the-art code generation
             
-            #### 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
+            #### DeepSeek R1 Distilled Models
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-32B** <span class="model-size">32B</span> - High quality distillation
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-14B** <span class="model-size">14B</span> - Balanced distillation
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-7B** <span class="model-size">7B</span> - Fast distillation
+            - **deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B** <span class="model-size">1.5B</span> - Ultra fast distillation
             
-            #### 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
-            - **meta-llama/Llama-3.2-3B-Instruct** - Ultra-fast
-            - **meta-llama/Llama-3.2-1B-Instruct** - Smallest and fastest
+            ### Performance Guide
             
-            ### Recommendations by Use Case
+            | Use Case | Recommended Models | Token Limit |
+            |----------|-------------------|-------------|
+            | Quick Prototypes | 0.5B-3B models | 2000-3000 |
+            | Production Code | 14B-32B models | 3000-4000 |
+            | Complex Projects | 72B models | 4000-6000 |
+            | Code Optimization | Coder/DeepSeek models | 3000-5000 |
             
-            **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
+            ### Status: <span class="verified-badge">All Models Active</span>
+            Last verified: 2025-01-21
             """,
             elem_classes=["section"]
         )
@@ -951,7 +962,7 @@ Transform website screenshots into functional HTML code using Nebius AI models.
     gr.Markdown(
         """
         ---
-        Powered by Nebius AI Studio | Built with Gradio
+        Powered by Nebius AI Studio | Built with Gradio | User: samsnata
         """,
         elem_classes=["footer"]
     )