Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-📱 Instagram Caption Generator - Simplified Version (FIXED DROPDOWNS)
+📱 Instagram Caption Generator - Simplified Version
 ==================================================
 
 AI-Powered Instagram Content Creation Suite with SambaNova Integration
@@ -8,7 +8,7 @@ Multi-Modal AI Analysis (Vision + Text) + Multi-Language Support
 🚀 Key Features:
 - SambaNova Llama-4-Maverick Integration
 - Multi-Language Support (German, Chinese, French, Arabic via Hugging Face)
-- Advanced Gradio Interface with WORKING DROPDOWNS
+- Advanced Gradio Interface
 - Advanced Error Handling & Security
 
 Author: MCP Hackathon 2025 Participant
@@ -133,14 +133,14 @@ class AdvancedInstagramGenerator:
         print("✅ AI models setup completed!")
     
     def setup_huggingface_client(self):
-        """Initialize Hugging Face client for translations - FIXED"""
+        """Initialize Hugging Face client for translations"""
         try:
-            # Initialize Hugging Face client - FIXED INITIALIZATION
+            # Initialize Hugging Face client
             hf_token = os.environ.get("HF_TOKEN")
             if hf_token:
-                # FIXED: Use correct initialization method
                 self.hf_client = InferenceClient(
-                    token=hf_token,  # Changed from provider and api_key
+                    provider="hf-inference",
+                    api_key=hf_token,
                 )
                 print("✅ Hugging Face client initialized successfully!")
                 self.hf_client_working = True
@@ -418,41 +418,6 @@ class AdvancedInstagramGenerator:
         
         return f"🇸🇦 ARABIC VERSION (Fallback):\n{translated}"
 
-    def get_fallback_hindi_translation(self, text: str) -> str:
-        """Fallback Hindi translation when HF API fails"""
-        # Simple keyword-based translation for common Instagram terms
-        hindi_translations = {
-            "amazing": "अद्भुत",
-            "beautiful": "सुंदर", 
-            "love": "प्रेम",
-            "perfect": "परफेक्ट",
-            "awesome": "शानदार",
-            "incredible": "अविश्वसनीय",
-            "follow": "फॉलो",
-            "like": "लाइक",
-            "share": "शेयर",
-            "comment": "कमेंट",
-            "today": "आज",
-            "moment": "पल",
-            "life": "जीवन",
-            "inspiration": "प्रेरणा",
-            "community": "समुदाय",
-            "content": "कंटेंट",
-            "check out": "देखें",
-            "what do you think": "आप क्या सोचते हैं"
-        }
-        
-        # Basic word replacement (not perfect but functional fallback)
-        translated = text.lower()
-        for english, hindi in hindi_translations.items():
-            translated = translated.replace(english, hindi)
-        
-        # Add Hindi hashtags
-        if "#" in translated:
-            translated += " #हिंदी #भारत #सोशलमीडिया #कंटेंट"
-        
-        return f"🇮🇳 HINDI VERSION (Fallback):\n{translated}"
-
     def setup_trend_analysis(self):
         """Initialize basic trend analysis"""
         self.trending_cache = {}
@@ -1026,18 +991,16 @@ Format:
 [Hashtags]"""
 
 
-# Global generator instance with caching - FIXED
+# Global generator instance with caching
 @functools.lru_cache(maxsize=1)
 def get_generator():
     """Get cached generator instance"""
     return AdvancedInstagramGenerator()
 
-# FIXED: Initialize generator with proper error handling
 try:
     generator = get_generator()
     setup_success = True
     setup_error = ""
-    print("✅ Generator initialized successfully!")
 except Exception as e:
     generator = None
     setup_success = False
@@ -1045,35 +1008,25 @@ except Exception as e:
     print(f"❌ Setup failed: {e}")
 
 
-# FIXED: Gradio Interface Functions with proper parameter handling
-def generate_advanced_caption_interface(uploaded_files, style, audience, custom_prompt):
-    """Advanced interface function for caption generation - FIXED"""
+# Gradio Interface Functions
+async def generate_advanced_caption_interface(uploaded_files, style, audience, 
+                                            custom_prompt):
+    """Advanced interface function for caption generation"""
     if not setup_success:
         return f"❌ Setup Error: {setup_error}", ""
     
-    # FIXED: Debug dropdown values
-    print(f"🔍 DEBUG - Style received: {style}")
-    print(f"🔍 DEBUG - Audience received: {audience}")
-    print(f"🔍 DEBUG - Custom prompt: {custom_prompt}")
-    
     images = []
     if uploaded_files:
         for file in uploaded_files[:3]:
             try:
                 image = Image.open(file.name)
                 images.append(image)
-                print(f"✅ Loaded image: {file.name}")
             except Exception as e:
                 return f"❌ Error processing file: {e}", ""
     
-    # FIXED: Use asyncio.run for async function
-    try:
-        import asyncio
-        result = asyncio.run(generator.generate_advanced_caption(
-            images, style, audience, custom_prompt
-        ))
-    except Exception as e:
-        return f"❌ Generation error: {str(e)}", ""
+    result = await generator.generate_advanced_caption(
+        images, style, audience, custom_prompt
+    )
     
     # Extract clean caption for multi-language processing
     caption_only = ""
@@ -1099,14 +1052,12 @@ def generate_advanced_caption_interface(uploaded_files, style, audience, custom_
     return result, caption_only
 
 
-def generate_multiple_captions_interface(uploaded_files, style, audience, custom_prompt):
-    """Generate multiple caption variations - FIXED"""
+async def generate_multiple_captions_interface(uploaded_files, style, audience, 
+                                             custom_prompt):
+    """Generate multiple caption variations using Meta-Llama-3.2-3B-Instruct"""
     if not setup_success:
         return f"❌ Setup Error: {setup_error}"
     
-    # FIXED: Debug dropdown values
-    print(f"🔍 DEBUG VARIATIONS - Style: {style}, Audience: {audience}")
-    
     images = []
     if uploaded_files:
         for file in uploaded_files[:3]:
@@ -1119,141 +1070,126 @@ def generate_multiple_captions_interface(uploaded_files, style, audience, custom
     if not images:
         return "❌ Please upload at least one image to generate caption variations."
     
-    # FIXED: Use asyncio.run for async functions
-    try:
-        import asyncio
-        
-        # First generate the main caption using Llama-4-Maverick
-        main_result = asyncio.run(generator.generate_advanced_caption(
-            images, style, audience, custom_prompt
-        ))
-        
-        # Extract just the caption text (without the header and footer)
-        base_caption = ""
-        if "✨ AI-GENERATED INSTAGRAM CONTENT:" in main_result:
-            lines = main_result.split('\n')
-            caption_lines = []
-            start_capturing = False
-            
-            for line in lines:
-                if "✨ AI-GENERATED INSTAGRAM CONTENT:" in line:
-                    start_capturing = True
-                    continue
-                elif "🤖 Powered by SambaNova" in line:
-                    break
-                elif start_capturing and line.strip():
-                    caption_lines.append(line)
-            
-            base_caption = '\n'.join(caption_lines).strip()
-        
-        if not base_caption:
-            return "❌ Failed to generate base caption for variations"
+    # First generate the main caption using Llama-4-Maverick
+    main_result = await generator.generate_advanced_caption(
+        images, style, audience, custom_prompt
+    )
+    
+    # Extract just the caption text (without the header and footer)
+    base_caption = ""
+    if "✨ AI-GENERATED INSTAGRAM CONTENT:" in main_result:
+        lines = main_result.split('\n')
+        caption_lines = []
+        start_capturing = False
         
-        # Generate 3 variations using Meta-Llama-3.2-3B-Instruct
-        variations = asyncio.run(generator.generate_mistral_variations(base_caption, count=3))
+        for line in lines:
+            if "✨ AI-GENERATED INSTAGRAM CONTENT:" in line:
+                start_capturing = True
+                continue
+            elif "🤖 Powered by SambaNova" in line:
+                break
+            elif start_capturing and line.strip():
+                caption_lines.append(line)
         
-        # Format the results
-        formatted_result = "✨ ALTERNATIVE CAPTIONS:\n\n"
+        base_caption = '\n'.join(caption_lines).strip()
+    
+    if not base_caption:
+        return "❌ Failed to generate base caption for variations"
+    
+    # Generate 3 variations using Meta-Llama-3.2-3B-Instruct
+    variations = await generator.generate_mistral_variations(base_caption, count=3)
+    
+    # Format the results
+    formatted_result = "✨ ALTERNATIVE CAPTIONS:\n\n"
+    formatted_result += "=" * 60 + "\n\n"
+    
+    for i, variation in enumerate(variations, 1):
+        formatted_result += f"📝 ALTERNATIVE {i}:\n"
+        formatted_result += f"{variation}\n"
         formatted_result += "=" * 60 + "\n\n"
-        
-        for i, variation in enumerate(variations, 1):
-            formatted_result += f"📝 ALTERNATIVE {i}:\n"
-            formatted_result += f"{variation}\n"
-            formatted_result += "=" * 60 + "\n\n"
-        
-        return formatted_result
-    except Exception as e:
-        return f"❌ Variation generation error: {str(e)}"
+    
+    return formatted_result
 
 
-def translate_caption_interface(base_caption, selected_languages):
-    """Generate multi-language versions of captions - FIXED"""
+async def translate_caption_interface(base_caption, selected_languages):
+    """Generate multi-language versions of captions"""
     if not base_caption.strip():
         return "❌ Please provide a caption to translate"
     
     if not selected_languages:
         return "❌ Please select at least one language"
     
-    # FIXED: Debug language selection
-    print(f"🔍 DEBUG TRANSLATION - Languages: {selected_languages}")
-    
     result = "🌍 MULTI-LANGUAGE CAPTION VERSIONS:\n\n"
     result += "=" * 60 + "\n\n"
     
-    # FIXED: Use asyncio.run for async translation functions
-    import asyncio
-    
     for language in selected_languages:
-        try:
-            if language == "🇩🇪 German":
-                # Use Hugging Face for German translation
-                if generator and generator.hf_client_working:
-                    try:
-                        german_translation = asyncio.run(generator.translate_to_german(base_caption))
-                        result += "🇩🇪 GERMAN VERSION (Hugging Face T5):\n"
-                        result += f"{german_translation}\n\n"
-                        result += "=" * 60 + "\n\n"
-                    except Exception as e:
-                        fallback_german = generator.get_fallback_german_translation(base_caption)
-                        result += f"{fallback_german}\n\n"
-                        result += "=" * 60 + "\n\n"
-                else:
+        if language == "🇩🇪 German":
+            # Use Hugging Face for German translation
+            if generator and generator.hf_client_working:
+                try:
+                    german_translation = await generator.translate_to_german(base_caption)
+                    result += "🇩🇪 GERMAN VERSION (Hugging Face T5):\n"
+                    result += f"{german_translation}\n\n"
+                    result += "=" * 60 + "\n\n"
+                except Exception as e:
                     fallback_german = generator.get_fallback_german_translation(base_caption)
                     result += f"{fallback_german}\n\n"
                     result += "=" * 60 + "\n\n"
-            
-            elif language == "🇨🇳 Chinese":
-                # Use Hugging Face for Chinese translation
-                if generator and generator.hf_client_working:
-                    try:
-                        chinese_translation = asyncio.run(generator.translate_to_chinese(base_caption))
-                        result += "🇨🇳 CHINESE VERSION (Hugging Face MT5):\n"
-                        result += f"{chinese_translation}\n\n"
-                        result += "=" * 60 + "\n\n"
-                    except Exception as e:
-                        fallback_chinese = generator.get_fallback_chinese_translation(base_caption)
-                        result += f"{fallback_chinese}\n\n"
-                        result += "=" * 60 + "\n\n"
-                else:
+            else:
+                fallback_german = generator.get_fallback_german_translation(base_caption)
+                result += f"{fallback_german}\n\n"
+                result += "=" * 60 + "\n\n"
+        
+        elif language == "🇨🇳 Chinese":
+            # Use Hugging Face for Chinese translation
+            if generator and generator.hf_client_working:
+                try:
+                    chinese_translation = await generator.translate_to_chinese(base_caption)
+                    result += "🇨🇳 CHINESE VERSION (Hugging Face MT5):\n"
+                    result += f"{chinese_translation}\n\n"
+                    result += "=" * 60 + "\n\n"
+                except Exception as e:
                     fallback_chinese = generator.get_fallback_chinese_translation(base_caption)
                     result += f"{fallback_chinese}\n\n"
                     result += "=" * 60 + "\n\n"
-                
-            elif language == "🇮🇳 Hindi":
-                # Use Hugging Face for Hindi translation
-                if generator and generator.hf_client_working:
-                    try:
-                        hindi_translation = asyncio.run(generator.translate_to_hindi(base_caption))
-                        result += "🇮🇳 HINDI VERSION (Hugging Face Helsinki-NLP):\n"
-                        result += f"{hindi_translation}\n\n"
-                        result += "=" * 60 + "\n\n"
-                    except Exception as e:
-                        fallback_hindi = generator.get_fallback_hindi_translation(base_caption)
-                        result += f"{fallback_hindi}\n\n"
-                        result += "=" * 60 + "\n\n"
-                else:
+            else:
+                fallback_chinese = generator.get_fallback_chinese_translation(base_caption)
+                result += f"{fallback_chinese}\n\n"
+                result += "=" * 60 + "\n\n"
+            
+        elif language == "🇮🇳 Hindi":
+            # Use Hugging Face for Hindi translation
+            if generator and generator.hf_client_working:
+                try:
+                    hindi_translation = await generator.translate_to_hindi(base_caption)
+                    result += "🇮🇳 HINDI VERSION (Hugging Face Helsinki-NLP):\n"
+                    result += f"{hindi_translation}\n\n"
+                    result += "=" * 60 + "\n\n"
+                except Exception as e:
                     fallback_hindi = generator.get_fallback_hindi_translation(base_caption)
                     result += f"{fallback_hindi}\n\n"
                     result += "=" * 60 + "\n\n"
-                
-            elif language == "🇸🇦 Arabic":
-                # Use Hugging Face for Arabic translation
-                if generator and generator.hf_client_working:
-                    try:
-                        arabic_translation = asyncio.run(generator.translate_to_arabic(base_caption))
-                        result += "🇸🇦 ARABIC VERSION (Hugging Face Marefa):\n"
-                        result += f"{arabic_translation}\n\n"
-                        result += "=" * 60 + "\n\n"
-                    except Exception as e:
-                        fallback_arabic = generator.get_fallback_arabic_translation(base_caption)
-                        result += f"{fallback_arabic}\n\n"
-                        result += "=" * 60 + "\n\n"
-                else:
+            else:
+                fallback_hindi = generator.get_fallback_hindi_translation(base_caption)
+                result += f"{fallback_hindi}\n\n"
+                result += "=" * 60 + "\n\n"
+            
+        elif language == "🇸🇦 Arabic":
+            # Use Hugging Face for Arabic translation
+            if generator and generator.hf_client_working:
+                try:
+                    arabic_translation = await generator.translate_to_arabic(base_caption)
+                    result += "🇸🇦 ARABIC VERSION (Hugging Face Marefa):\n"
+                    result += f"{arabic_translation}\n\n"
+                    result += "=" * 60 + "\n\n"
+                except Exception as e:
                     fallback_arabic = generator.get_fallback_arabic_translation(base_caption)
                     result += f"{fallback_arabic}\n\n"
                     result += "=" * 60 + "\n\n"
-        except Exception as e:
-            result += f"❌ Error translating to {language}: {str(e)}\n\n"
+            else:
+                fallback_arabic = generator.get_fallback_arabic_translation(base_caption)
+                result += f"{fallback_arabic}\n\n"
+                result += "=" * 60 + "\n\n"
     
     if any(lang in selected_languages for lang in ["🇩🇪 German", "🇨🇳 Chinese", "🇮🇳 Hindi", "🇸🇦 Arabic"]):
         hf_langs = []
@@ -1274,9 +1210,13 @@ def translate_caption_interface(base_caption, selected_languages):
 
 
 def create_gradio_app():
-    """Create the Gradio app with SambaNova and Hugging Face integration - FIXED DROPDOWNS"""
+    """Create the Gradio app with good UI styling"""
+    
+    # Status indicators  
+    hf_status = "✅ Connected" if generator and generator.hf_client_working else "⚠️ Fallback Mode"
+    sambanova_status = "✅ Connected" if generator and generator.sambanova_client_working else "⚠️ Fallback Mode"
     
-    # Championship-level CSS
+    # Enhanced CSS for better UI
     css = """
     @import url('https://fonts.googleapis.com/css2?family=Space+Grotesk:wght@300;400;500;600;700&display=swap');
     
@@ -1284,258 +1224,146 @@ def create_gradio_app():
         background: linear-gradient(135deg, #667eea 0%, #764ba2 25%, #f093fb 50%, #f5576c 75%, #4facfe 100%);
         font-family: 'Space Grotesk', 'Inter', system-ui, sans-serif;
         min-height: 100vh;
-        position: relative;
-        overflow-x: hidden;
-    }
-    
-    .gradio-container::before {
-        content: '';
-        position: absolute;
-        top: 0;
-        left: 0;
-        right: 0;
-        bottom: 0;
-        background: radial-gradient(circle at 20% 80%, rgba(120, 119, 198, 0.3) 0%, transparent 50%),
-                    radial-gradient(circle at 80% 20%, rgba(255, 119, 198, 0.3) 0%, transparent 50%);
-        pointer-events: none;
-        z-index: 1;
     }
     
     .main-header {
         text-align: center;
         color: white;
-        margin-bottom: 40px;
-        padding: 40px;
-        background: rgba(255,255,255,0.1);
-        border-radius: 30px;
-        backdrop-filter: blur(30px);
-        border: 2px solid rgba(255,255,255,0.2);
-        box-shadow: 0 25px 50px rgba(0,0,0,0.1);
-        position: relative;
-        z-index: 2;
-        animation: float 6s ease-in-out infinite;
-    }
-    
-    @keyframes float {
-        0%, 100% { transform: translateY(0px); }
-        50% { transform: translateY(-10px); }
-    }
-    
-    .feature-card {
-        background: rgba(255,255,255,0.15);
-        border-radius: 25px;
+        margin-bottom: 30px;
         padding: 30px;
+        background: rgba(255,255,255,0.1);
+        border-radius: 20px;
         backdrop-filter: blur(20px);
-        border: 1px solid rgba(255,255,255,0.3);
+        border: 1px solid rgba(255,255,255,0.2);
         box-shadow: 0 20px 40px rgba(0,0,0,0.1);
-        transition: all 0.4s cubic-bezier(0.4, 0, 0.2, 1);
-        position: relative;
-        z-index: 2;
     }
     
-    .feature-card:hover {
-        transform: translateY(-8px) scale(1.02);
-        box-shadow: 0 30px 60px rgba(0,0,0,0.2);
-        background: rgba(255,255,255,0.2);
+    .feature-card {
+        background: rgba(255,255,255,0.1);
+        border-radius: 15px;
+        padding: 20px;
+        backdrop-filter: blur(15px);
+        border: 1px solid rgba(255,255,255,0.2);
+        box-shadow: 0 15px 30px rgba(0,0,0,0.1);
+        margin: 10px 0;
     }
     
     .gradio-button-primary {
         background: linear-gradient(45deg, #ff6b6b, #ee5a24, #ff9ff3, #54a0ff) !important;
-        background-size: 300% 300% !important;
-        animation: gradient-shift 4s ease infinite !important;
         border: none !important;
-        border-radius: 20px !important;
-        padding: 18px 35px !important;
-        font-weight: 700 !important;
-        text-transform: uppercase !important;
-        letter-spacing: 2px !important;
-        box-shadow: 0 15px 35px rgba(255, 107, 107, 0.4) !important;
+        border-radius: 15px !important;
+        padding: 15px 25px !important;
+        font-weight: 600 !important;
+        color: white !important;
+        box-shadow: 0 10px 20px rgba(255, 107, 107, 0.3) !important;
         transition: all 0.3s ease !important;
-        position: relative !important;
-        overflow: hidden !important;
-    }
-    
-    @keyframes gradient-shift {
-        0% { background-position: 0% 50%; }
-        50% { background-position: 100% 50%; }
-        100% { background-position: 0% 50%; }
     }
     
     .gradio-button-primary:hover {
-        transform: translateY(-3px) scale(1.05) !important;
-        box-shadow: 0 20px 40px rgba(255, 107, 107, 0.6) !important;
+        transform: translateY(-2px) !important;
+        box-shadow: 0 15px 30px rgba(255, 107, 107, 0.5) !important;
     }
     
     .gradio-button-secondary {
         background: linear-gradient(45deg, #feca57, #ff9ff3, #54a0ff, #5f27cd) !important;
-        background-size: 300% 300% !important;
-        animation: gradient-shift 4s ease infinite !important;
         border: none !important;
-        border-radius: 18px !important;
-        padding: 15px 30px !important;
+        border-radius: 12px !important;
+        padding: 12px 20px !important;
         font-weight: 600 !important;
-        text-transform: uppercase !important;
-        letter-spacing: 1.5px !important;
-        box-shadow: 0 12px 25px rgba(254, 202, 87, 0.4) !important;
-        transition: all 0.3s ease !important;
         color: white !important;
+        box-shadow: 0 8px 16px rgba(254, 202, 87, 0.3) !important;
     }
     
-    .gradio-button-secondary:hover {
-        transform: translateY(-2px) scale(1.03) !important;
-        box-shadow: 0 15px 30px rgba(254, 202, 87, 0.6) !important;
-    }
-    
-    .advanced-controls {
-        background: rgba(255,255,255,0.1);
-        border-radius: 20px;
-        padding: 25px;
-        backdrop-filter: blur(15px);
-        border: 1px solid rgba(255,255,255,0.2);
-        margin: 20px 0;
-    }
-    
-    .sambanova-status {
+    .status-badge {
         background: linear-gradient(90deg, #2ecc71, #27ae60);
         color: white;
-        padding: 15px 25px;
-        border-radius: 15px;
+        padding: 10px 20px;
+        border-radius: 10px;
         text-align: center;
         font-weight: 600;
-        box-shadow: 0 10px 20px rgba(46, 204, 113, 0.3);
-        animation: pulse 2s infinite;
-    }
-    
-    @keyframes pulse {
-        0% { box-shadow: 0 10px 20px rgba(46, 204, 113, 0.3); }
-        50% { box-shadow: 0 15px 30px rgba(46, 204, 113, 0.5); }
-        100% { box-shadow: 0 10px 20px rgba(46, 204, 113, 0.3); }
-    }
-    
-    .trending-indicator {
-        display: inline-block;
-        width: 8px;
-        height: 8px;
-        background: #ff6b6b;
-        border-radius: 50%;
-        animation: blink 1s infinite;
-        margin-right: 8px;
-    }
-    
-    @keyframes blink {
-        0%, 50% { opacity: 1; }
-        51%, 100% { opacity: 0.3; }
+        box-shadow: 0 8px 16px rgba(46, 204, 113, 0.3);
+        margin: 10px 0;
     }
     """
     
-    # FIXED: Use specific theme and create proper block structure
-    with gr.Blocks(
-        css=css, 
-        title="📱 Instagram Generator - MCP Hackathon 2025", 
-        theme=gr.themes.Soft()  # Changed from Glass to Soft for better dropdown compatibility
-    ) as app:
+    with gr.Blocks(css=css, title="📱 Instagram Generator", theme=gr.themes.Glass()) as app:
         
         # Main Header
-        gr.HTML("""
+        gr.HTML(f"""
         <div class="main-header">
-            <h1 style="font-size: 3rem; margin-bottom: 15px; font-weight: 800; background: linear-gradient(45deg, #ff6b6b, #feca57, #ff9ff3, #54a0ff); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
+            <h1 style="font-size: 2.5rem; margin-bottom: 15px; font-weight: 800; background: linear-gradient(45deg, #ff6b6b, #feca57, #ff9ff3, #54a0ff); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
                 📱 INSTAGRAM CAPTION GENERATOR
             </h1>
-            <h2 style="font-size: 1.4rem; margin-bottom: 20px; opacity: 0.9; font-weight: 500;">
-                🚀 MCP Hackathon 2025 Submission • SambaNova + Hugging Face Edition
+            <h2 style="font-size: 1.2rem; margin-bottom: 20px; opacity: 0.9;">
+                🚀 AI-Powered Content Creation • SambaNova + Hugging Face
             </h2>
-            <div style="display: flex; justify-content: center; gap: 20px; margin-top: 25px;">
-                <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 20px; font-size: 0.9rem;">🤖 SambaNova AI</span>
-                <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 20px; font-size: 0.9rem;">🌍 Multi-Language</span>
-                <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 20px; font-size: 0.9rem;">🔗 MCP Server</span>
+            <div style="display: flex; justify-content: center; gap: 20px; margin-top: 15px;">
+                <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 15px;">🤖 SambaNova: {sambanova_status}</span>
+                <span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 15px;">🤗 Hugging Face: {hf_status}</span>
             </div>
         </div>
         """)
         
-        # Status Indicators
-        hf_status = "✅ Connected" if generator and generator.hf_client_working else "⚠️ Fallback Mode"
-        sambanova_status = "✅ Connected" if generator and generator.sambanova_client_working else "⚠️ Fallback Mode"
-        
-        gr.HTML(f"""
-        <div class="sambanova-status">
-            <span class="trending-indicator"></span>
-            🚀 SambaNova: {sambanova_status} • 🤗 Hugging Face: {hf_status} • 🦙 Llama-3.2 • MCP Server Ready
-        </div>
-        """)
-        
         # Main Interface
-        with gr.Tab("🎯 Advanced Caption Generator"):
+        with gr.Tab("🎯 Caption Generator"):
             with gr.Row():
-                # Left Column - Enhanced Controls
+                # Left Column - Controls
                 with gr.Column(scale=2, elem_classes=["feature-card"]):
-                    gr.Markdown("### 🖼️ Multi-Image Upload & Analysis")
+                    gr.Markdown("### 🖼️ Upload Images")
                     gr.Markdown("*SambaNova AI vision analysis with quality scoring*")
                     
                     images = gr.File(
                         label="📸 Upload Images (Max 3)", 
                         file_count="multiple",
                         file_types=["image"],
-                        height=280
+                        height=200
                     )
                     
-                    with gr.Group(elem_classes=["advanced-controls"]):
-                        gr.Markdown("### ⚙️ AI Configuration")
-                        
-                        with gr.Row():
-                            # FIXED: Improved dropdown definition with explicit parameters
-                            caption_style = gr.Dropdown(
-                                label="🎨 Caption Style",
-                                choices=[
-                                    "🎯 Viral Engagement", 
-                                    "💼 Professional Brand", 
-                                    "😄 Casual Fun", 
-                                    "😂 Humor & Memes", 
-                                    "💪 Motivational", 
-                                    "📖 Storytelling",
-                                    "🌟 Luxury Lifestyle",
-                                    "🔥 Trending Culture"
-                                ],
-                                value="🎯 Viral Engagement",
-                                interactive=True,  # FIXED: Explicitly set interactive
-                                allow_custom_value=False,  # FIXED: Prevent custom values
-                                multiselect=False  # FIXED: Single selection only
-                            )
-                            
-                            target_audience = gr.Dropdown(
-                                label="👥 Target Audience",
-                                choices=[
-                                    "🌟 General Audience", 
-                                    "💼 Business Professionals", 
-                                    "✈️ Travel Enthusiasts", 
-                                    "🍕 Food Lovers", 
-                                    "💪 Fitness Community", 
-                                    "👗 Fashion Forward", 
-                                    "💻 Tech Innovators", 
-                                    "🎨 Creative Artists",
-                                    "🌱 Sustainability Advocates",
-                                    "🎵 Music Fans"
-                                ],
-                                value="🌟 General Audience",
-                                interactive=True,  # FIXED: Explicitly set interactive
-                                allow_custom_value=False,  # FIXED: Prevent custom values
-                                multiselect=False  # FIXED: Single selection only
-                            )
+                    gr.Markdown("### ⚙️ Configuration")
+                    
+                    with gr.Row():
+                        caption_style = gr.Dropdown(
+                            choices=[
+                                "🎯 Viral Engagement", 
+                                "💼 Professional Brand", 
+                                "😄 Casual Fun", 
+                                "😂 Humor & Memes", 
+                                "💪 Motivational", 
+                                "📖 Storytelling",
+                                "🌟 Luxury Lifestyle",
+                                "🔥 Trending Culture"
+                            ],
+                            value="🎯 Viral Engagement",
+                            label="🎨 Caption Style"
+                        )
                         
-                        custom_prompt = gr.Textbox(
-                            label="💬 Advanced Instructions",
-                            placeholder="e.g., 'Focus on sustainability messaging', 'Include product launch details', 'Emphasize community building'...",
-                            lines=3,
-                            interactive=True  # FIXED: Explicitly set interactive
+                        target_audience = gr.Dropdown(
+                            choices=[
+                                "🌟 General Audience", 
+                                "💼 Business Professionals", 
+                                "✈️ Travel Enthusiasts", 
+                                "🍕 Food Lovers", 
+                                "💪 Fitness Community", 
+                                "👗 Fashion Forward", 
+                                "💻 Tech Innovators", 
+                                "🎨 Creative Artists"
+                            ],
+                            value="🌟 General Audience",
+                            label="👥 Target Audience"
                         )
                     
+                    custom_prompt = gr.Textbox(
+                        label="💬 Additional Instructions",
+                        placeholder="e.g., 'Focus on sustainability', 'Include product details'...",
+                        lines=2
+                    )
+                    
                     generate_btn = gr.Button(
                         "🚀 Generate Caption", 
                         variant="primary", 
-                        size="lg",
-                        scale=2
+                        size="lg"
                     )
                 
-                
                 # Right Column - Results
                 with gr.Column(scale=3, elem_classes=["feature-card"]):
                     gr.Markdown("### 📊 Generated Content")
@@ -1545,8 +1373,7 @@ def create_gradio_app():
                         lines=15,
                         max_lines=20,
                         show_copy_button=True,
-                        placeholder="Upload images and generate your Instagram content...",
-                        interactive=False  # FIXED: Output should not be interactive
+                        placeholder="Upload images and generate your Instagram content..."
                     )
                     
                     with gr.Row():
@@ -1560,8 +1387,7 @@ def create_gradio_app():
                         label="✨ Alternative Captions",
                         lines=15,
                         show_copy_button=True,
-                        placeholder="Generate 3 different caption alternatives using Meta-Llama-3.2-3B-Instruct...",
-                        interactive=False  # FIXED: Output should not be interactive
+                        placeholder="Generate 3 different caption alternatives using Meta-Llama-3.2-3B-Instruct..."
                     )
         
         # Multi-Language Tab
@@ -1574,21 +1400,18 @@ def create_gradio_app():
                     base_caption_input = gr.Textbox(
                         label="📝 Base Caption",
                         placeholder="Paste your generated caption here...",
-                        lines=5,
-                        interactive=True  # FIXED: Input should be interactive
+                        lines=5
                     )
                     
-                    # FIXED: Improved checkbox group definition
                     language_selector = gr.CheckboxGroup(
-                        label="🌐 Select Languages",
                         choices=[
                             "🇩🇪 German",
                             "🇨🇳 Chinese", 
                             "🇮🇳 Hindi",
                             "🇸🇦 Arabic"
                         ],
-                        value=["🇩🇪 German", "🇮🇳 Hindi"],
-                        interactive=True  # FIXED: Explicitly set interactive
+                        label="🌐 Select Languages",
+                        value=["🇩🇪 German", "🇮🇳 Hindi"]
                     )
                     
                     translate_btn = gr.Button(
@@ -1601,8 +1424,7 @@ def create_gradio_app():
                         label="🗺️ Multi-Language Captions",
                         lines=20,
                         show_copy_button=True,
-                        placeholder="Culturally adapted captions for global audiences...",
-                        interactive=False  # FIXED: Output should not be interactive
+                        placeholder="Culturally adapted captions for global audiences..."
                     )
         
         # SambaNova Features Tab
@@ -1722,7 +1544,8 @@ variations = client.chat.completions.create(
 from huggingface_hub import InferenceClient
 
 client = InferenceClient(
-    token=os.environ["HF_TOKEN"],  # FIXED: Use token parameter
+    provider="hf-inference",
+    api_key=os.environ["HF_TOKEN"],
 )
 
 # German translation
@@ -1759,34 +1582,26 @@ arabic_result = client.translation(
                         label="🔧 Hugging Face Translation Code"
                     )
         
-        # FIXED: Event Handlers with proper error handling
-        print("🔗 Setting up event handlers...")
-        
-        # Main caption generation
+        # Event Handlers
         generate_btn.click(
             fn=generate_advanced_caption_interface,
             inputs=[images, caption_style, target_audience, custom_prompt],
-            outputs=[output, base_caption_input],
-            show_progress=True  # FIXED: Show progress indicator
+            outputs=[output, base_caption_input]
         )
         
         # Generate multiple alternatives
         alternatives_btn.click(
             fn=generate_multiple_captions_interface,
             inputs=[images, caption_style, target_audience, custom_prompt],
-            outputs=alternatives_output,
-            show_progress=True  # FIXED: Show progress indicator
+            outputs=alternatives_output
         )
         
         # Multi-language translation
         translate_btn.click(
             fn=translate_caption_interface,
             inputs=[base_caption_input, language_selector],
-            outputs=multilingual_output,
-            show_progress=True  # FIXED: Show progress indicator
+            outputs=multilingual_output
         )
-        
-        print("✅ Event handlers configured successfully!")
     
     return app
 
@@ -1800,7 +1615,6 @@ def main():
     if not setup_success:
         print(f"❌ Setup failed: {setup_error}")
         print("💡 Please check your API configuration")
-        return
     
     # Status messages
     sambanova_msg = "✅ SambaNova ready!" if generator and generator.sambanova_client_working else "⚠️ SambaNova fallback mode"
@@ -1813,20 +1627,7 @@ def main():
     
     # Create and launch the app
     app = create_gradio_app()
-    
-    # FIXED: Launch configuration for better compatibility
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        show_error=True,
-        debug=True,  # FIXED: Enable debug mode for troubleshooting
-        show_api=True,  # FIXED: Show API for debugging
-        inbrowser=False,
-        favicon_path=None,
-        ssl_verify=False,
-        quiet=False
-    )
+    app.launch(mcp_server=True)
 
 
 if __name__ == "__main__":