Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-📱 Instagram Caption Generator - Simplified Version
+📱 Instagram Caption Generator - Simplified Version (FIXED DROPDOWNS)
 ==================================================
 
 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
+- Advanced Gradio Interface with WORKING DROPDOWNS
 - 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"""
+        """Initialize Hugging Face client for translations - FIXED"""
         try:
-            # Initialize Hugging Face client
+            # Initialize Hugging Face client - FIXED INITIALIZATION
             hf_token = os.environ.get("HF_TOKEN")
             if hf_token:
+                # FIXED: Use correct initialization method
                 self.hf_client = InferenceClient(
-                    provider="hf-inference",
-                    api_key=hf_token,
+                    token=hf_token,  # Changed from provider and api_key
                 )
                 print("✅ Hugging Face client initialized successfully!")
                 self.hf_client_working = True
@@ -1026,16 +1026,18 @@ Format:
 [Hashtags]"""
 
 
-# Global generator instance with caching
+# Global generator instance with caching - FIXED
 @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
@@ -1043,25 +1045,35 @@ except Exception as e:
     print(f"❌ Setup failed: {e}")
 
 
-# Gradio Interface Functions
-async def generate_advanced_caption_interface(uploaded_files, style, audience, 
-                                            custom_prompt):
-    """Advanced interface function for caption generation"""
+# 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"""
     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}", ""
     
-    result = await generator.generate_advanced_caption(
-        images, style, audience, custom_prompt
-    )
+    # 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)}", ""
     
     # Extract clean caption for multi-language processing
     caption_only = ""
@@ -1087,12 +1099,14 @@ async def generate_advanced_caption_interface(uploaded_files, style, audience,
     return result, caption_only
 
 
-async def generate_multiple_captions_interface(uploaded_files, style, audience, 
-                                             custom_prompt):
-    """Generate multiple caption variations using Meta-Llama-3.2-3B-Instruct"""
+def generate_multiple_captions_interface(uploaded_files, style, audience, custom_prompt):
+    """Generate multiple caption variations - FIXED"""
     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]:
@@ -1105,126 +1119,141 @@ async def generate_multiple_captions_interface(uploaded_files, style, audience,
     if not images:
         return "❌ Please upload at least one image to generate caption 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
+    # FIXED: Use asyncio.run for async functions
+    try:
+        import asyncio
         
-        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)
+        # First generate the main caption using Llama-4-Maverick
+        main_result = asyncio.run(generator.generate_advanced_caption(
+            images, style, audience, custom_prompt
+        ))
         
-        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"
+        # 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"
+        
+        # Generate 3 variations using Meta-Llama-3.2-3B-Instruct
+        variations = asyncio.run(generator.generate_mistral_variations(base_caption, count=3))
+        
+        # Format the results
+        formatted_result = "✨ ALTERNATIVE CAPTIONS:\n\n"
         formatted_result += "=" * 60 + "\n\n"
-    
-    return formatted_result
+        
+        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)}"
 
 
-async def translate_caption_interface(base_caption, selected_languages):
-    """Generate multi-language versions of captions"""
+def translate_caption_interface(base_caption, selected_languages):
+    """Generate multi-language versions of captions - FIXED"""
     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:
-        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:
+        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:
                     fallback_german = generator.get_fallback_german_translation(base_caption)
                     result += f"{fallback_german}\n\n"
                     result += "=" * 60 + "\n\n"
-            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:
+            
+            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:
                     fallback_chinese = generator.get_fallback_chinese_translation(base_caption)
                     result += f"{fallback_chinese}\n\n"
                     result += "=" * 60 + "\n\n"
-            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:
+                
+            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:
                     fallback_hindi = generator.get_fallback_hindi_translation(base_caption)
                     result += f"{fallback_hindi}\n\n"
                     result += "=" * 60 + "\n\n"
-            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:
+                
+            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:
                     fallback_arabic = generator.get_fallback_arabic_translation(base_caption)
                     result += f"{fallback_arabic}\n\n"
                     result += "=" * 60 + "\n\n"
-            else:
-                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"
     
     if any(lang in selected_languages for lang in ["🇩🇪 German", "🇨🇳 Chinese", "🇮🇳 Hindi", "🇸🇦 Arabic"]):
         hf_langs = []
@@ -1245,7 +1274,7 @@ async def translate_caption_interface(base_caption, selected_languages):
 
 
 def create_gradio_app():
-    """Create the Gradio app with SambaNova and Hugging Face integration"""
+    """Create the Gradio app with SambaNova and Hugging Face integration - FIXED DROPDOWNS"""
     
     # Championship-level CSS
     css = """
@@ -1399,7 +1428,12 @@ def create_gradio_app():
     }
     """
     
-    with gr.Blocks(css=css, title="📱 Instagram Generator - MCP Hackathon 2025", theme=gr.themes.Glass()) as app:
+    # 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:
         
         # Main Header
         gr.HTML("""
@@ -1448,7 +1482,9 @@ def create_gradio_app():
                         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", 
@@ -1460,10 +1496,13 @@ def create_gradio_app():
                                     "🔥 Trending Culture"
                                 ],
                                 value="🎯 Viral Engagement",
-                                label="🎨 Caption Style"
+                                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", 
@@ -1477,13 +1516,16 @@ def create_gradio_app():
                                     "🎵 Music Fans"
                                 ],
                                 value="🌟 General Audience",
-                                label="👥 Target Audience"
+                                interactive=True,  # FIXED: Explicitly set interactive
+                                allow_custom_value=False,  # FIXED: Prevent custom values
+                                multiselect=False  # FIXED: Single selection only
                             )
                         
                         custom_prompt = gr.Textbox(
                             label="💬 Advanced Instructions",
                             placeholder="e.g., 'Focus on sustainability messaging', 'Include product launch details', 'Emphasize community building'...",
-                            lines=3
+                            lines=3,
+                            interactive=True  # FIXED: Explicitly set interactive
                         )
                     
                     generate_btn = gr.Button(
@@ -1503,7 +1545,8 @@ def create_gradio_app():
                         lines=15,
                         max_lines=20,
                         show_copy_button=True,
-                        placeholder="Upload images and generate your Instagram content..."
+                        placeholder="Upload images and generate your Instagram content...",
+                        interactive=False  # FIXED: Output should not be interactive
                     )
                     
                     with gr.Row():
@@ -1517,7 +1560,8 @@ 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..."
+                        placeholder="Generate 3 different caption alternatives using Meta-Llama-3.2-3B-Instruct...",
+                        interactive=False  # FIXED: Output should not be interactive
                     )
         
         # Multi-Language Tab
@@ -1530,18 +1574,21 @@ def create_gradio_app():
                     base_caption_input = gr.Textbox(
                         label="📝 Base Caption",
                         placeholder="Paste your generated caption here...",
-                        lines=5
+                        lines=5,
+                        interactive=True  # FIXED: Input should be interactive
                     )
                     
+                    # FIXED: Improved checkbox group definition
                     language_selector = gr.CheckboxGroup(
+                        label="🌐 Select Languages",
                         choices=[
                             "🇩🇪 German",
                             "🇨🇳 Chinese", 
                             "🇮🇳 Hindi",
                             "🇸🇦 Arabic"
                         ],
-                        label="🌐 Select Languages",
-                        value=["🇩🇪 German", "🇮🇳 Hindi"]
+                        value=["🇩🇪 German", "🇮🇳 Hindi"],
+                        interactive=True  # FIXED: Explicitly set interactive
                     )
                     
                     translate_btn = gr.Button(
@@ -1554,7 +1601,8 @@ def create_gradio_app():
                         label="🗺️ Multi-Language Captions",
                         lines=20,
                         show_copy_button=True,
-                        placeholder="Culturally adapted captions for global audiences..."
+                        placeholder="Culturally adapted captions for global audiences...",
+                        interactive=False  # FIXED: Output should not be interactive
                     )
         
         # SambaNova Features Tab
@@ -1674,8 +1722,7 @@ variations = client.chat.completions.create(
 from huggingface_hub import InferenceClient
 
 client = InferenceClient(
-    provider="hf-inference",
-    api_key=os.environ["HF_TOKEN"],
+    token=os.environ["HF_TOKEN"],  # FIXED: Use token parameter
 )
 
 # German translation
@@ -1712,26 +1759,34 @@ arabic_result = client.translation(
                         label="🔧 Hugging Face Translation Code"
                     )
         
-        # Event Handlers
+        # FIXED: Event Handlers with proper error handling
+        print("🔗 Setting up event handlers...")
+        
+        # Main caption generation
         generate_btn.click(
             fn=generate_advanced_caption_interface,
             inputs=[images, caption_style, target_audience, custom_prompt],
-            outputs=[output, base_caption_input]
+            outputs=[output, base_caption_input],
+            show_progress=True  # FIXED: Show progress indicator
         )
         
         # Generate multiple alternatives
         alternatives_btn.click(
             fn=generate_multiple_captions_interface,
             inputs=[images, caption_style, target_audience, custom_prompt],
-            outputs=alternatives_output
+            outputs=alternatives_output,
+            show_progress=True  # FIXED: Show progress indicator
         )
         
         # Multi-language translation
         translate_btn.click(
             fn=translate_caption_interface,
             inputs=[base_caption_input, language_selector],
-            outputs=multilingual_output
+            outputs=multilingual_output,
+            show_progress=True  # FIXED: Show progress indicator
         )
+        
+        print("✅ Event handlers configured successfully!")
     
     return app
 
@@ -1745,6 +1800,7 @@ 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"
@@ -1757,7 +1813,20 @@ def main():
     
     # Create and launch the app
     app = create_gradio_app()
-    app.launch(mcp_server=True)
+    
+    # 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
+    )
 
 
 if __name__ == "__main__":