Update app.py

app.py

@@ -6,30 +6,81 @@ import gradio as gr
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Check Modal availability
+# Robust Modal authentication setup
+def setup_modal_authentication():
+    """Configure Modal authentication properly"""
+    modal_token = os.environ.get("MODAL_TOKEN")
+    
+    if not modal_token:
+        logger.warning("⚠️ MODAL_TOKEN environment variable not found")
+        return False
+    
+    # Set multiple environment variables that Modal might check
+    try:
+        # Method 1: Set MODAL_TOKEN_ID (most common)
+        os.environ["MODAL_TOKEN_ID"] = modal_token
+        
+        # Method 2: Set MODAL_TOKEN_SECRET (alternative)  
+        os.environ["MODAL_TOKEN_SECRET"] = modal_token
+        
+        # Method 3: Set the base MODAL_TOKEN (redundant but safe)
+        os.environ["MODAL_TOKEN"] = modal_token
+        
+        logger.info("✅ Modal environment variables configured")
+        return True
+        
+    except Exception as e:
+        logger.error(f"❌ Failed to configure Modal environment: {e}")
+        return False
+
+# Check Modal availability with robust authentication
 MODAL_AVAILABLE = False
 generate_content_with_llm = None
 
-try:
-    import modal
-    logger.info(f"Modal version: {modal.__version__}")
-    
-    # Check for Modal token
-    if os.environ.get("MODAL_TOKEN"):
+if setup_modal_authentication():
+    try:
+        import modal
+        logger.info(f"Modal version: {modal.__version__}")
+        
+        # Try to authenticate and connect
         try:
+            # Test 1: Try direct function lookup
             generate_content_with_llm = modal.Function.from_name(
                 "content-creation-agent",
                 "generate_content_with_llm"
             )
+            
+            # Test 2: Try a simple health check to verify authentication
+            health_check_func = modal.Function.from_name(
+                "content-creation-agent", 
+                "health_check"
+            )
+            
+            # Attempt a test call to verify authentication works
+            test_result = health_check_func.remote()
+            logger.info(f"✅ Modal authentication test passed: {test_result}")
+            
             MODAL_AVAILABLE = True
-            logger.info("✅ Modal successfully connected")
-        except Exception as e:
-            logger.warning(f"⚠️ Modal function loading failed: {e}")
-    else:
-        logger.warning("⚠️ MODAL_TOKEN not found in environment")
-        
-except ImportError:
-    logger.warning("⚠️ Modal package not available")
+            logger.info("✅ Modal successfully connected and authenticated")
+            
+        except Exception as auth_error:
+            logger.warning(f"⚠️ Modal authentication failed: {auth_error}")
+            
+            # Try fallback function
+            try:
+                generate_content_with_llm = modal.Function.from_name(
+                    "content-creation-agent",
+                    "generate_fallback_content"  # Use fallback function instead
+                )
+                MODAL_AVAILABLE = True
+                logger.info("✅ Modal connected using fallback function")
+            except Exception as fallback_error:
+                logger.error(f"❌ Modal fallback connection failed: {fallback_error}")
+                
+    except ImportError:
+        logger.warning("⚠️ Modal package not available")
+else:
+    logger.error("❌ Modal authentication setup failed")
 
 logger.info(f"Modal Status: {'✅ Available' if MODAL_AVAILABLE else '❌ Unavailable'}")
 
@@ -146,7 +197,7 @@ Give it a listen and support independent creators!
 {hashtags['gab']}"""
         }
     else:
-        # Anonymous content
+        # Anonymous content (same structure, different text)
         return {
             'youtube': f"""🎵 {title} 🎵
 
@@ -207,7 +258,7 @@ Give it a listen and support independent creators!
         }
 
 def generate_all_content(lyrics, artist, title, use_modal):
-    """Generate content with Modal toggle"""
+    """Generate content with Modal toggle and robust error handling"""
     if not lyrics.strip() or not title.strip():
         error_msg = "❌ Please provide both lyrics and song title"
         return [""] * 6 + [error_msg]
@@ -224,10 +275,24 @@ def generate_all_content(lyrics, artist, title, use_modal):
     
     try:
         if use_modal and MODAL_AVAILABLE and generate_content_with_llm:
-            # Try Modal AI generation
-            result = generate_content_with_llm.remote(lyrics, artist, title)
-            content = result
-            status_msg = f"✅ Generated with Modal AI in {time.time() - start_time:.2f}s"
+            # Try Modal generation with additional error handling
+            try:
+                logger.info("🔄 Attempting Modal AI generation...")
+                result = generate_content_with_llm.remote(lyrics, artist, title)
+                
+                # Validate the result
+                if isinstance(result, dict) and len(result) > 0:
+                    content = result
+                    status_msg = f"✅ Generated with Modal AI in {time.time() - start_time:.2f}s"
+                    logger.info("✅ Modal AI generation successful")
+                else:
+                    raise ValueError("Invalid result from Modal function")
+                    
+            except Exception as modal_error:
+                logger.warning(f"⚠️ Modal AI generation failed: {modal_error}")
+                # Fall back to local generation
+                content = generate_fallback_content(lyrics, artist, title)
+                status_msg = f"⚠️ Modal failed, used local fallback in {time.time() - start_time:.2f}s"
         else:
             # Use local fallback
             if use_modal and not MODAL_AVAILABLE:
@@ -237,9 +302,9 @@ def generate_all_content(lyrics, artist, title, use_modal):
                 status_msg = f"✅ Generated with local templates in {time.time() - start_time:.2f}s"
     
     except Exception as e:
-        logger.warning(f"⚠️ Modal generation failed: {e}")
+        logger.error(f"❌ Unexpected error: {e}")
         content = generate_fallback_content(lyrics, artist, title)
-        status_msg = f"⚠️ Modal failed, used local fallback in {time.time() - start_time:.2f}s"
+        status_msg = f"❌ Error occurred, used local fallback in {time.time() - start_time:.2f}s"
     
     # Final yield - actual content for each platform + status
     platforms = ['youtube', 'twitter', 'instagram', 'facebook', 'minds', 'gab']
@@ -288,7 +353,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as de
                     ("🤖 AI Generation (Modal)", True),
                     ("📝 Local Templates", False)
                 ],
-                value=MODAL_AVAILABLE,  # Default to Modal if available
+                value=False,  # Default to Local to avoid authentication issues
                 label="Generation Method",
                 info="Choose between AI-powered generation or local templates"
             )
@@ -323,13 +388,13 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as de
                 "Here's an example with some sample lyrics\nAbout dreams and aspirations\nReaching for the stars tonight\nNothing's gonna stop this fight\nWe'll keep on climbing higher\nUntil we touch the sky",
                 "",  # Empty artist name
                 "Dreams Tonight",
-                True  # Use Modal if available
+                False  # Default to local generation
             ],
             [
                 "Love is in the air tonight\nHearts beating as one\nDancing under moonlight\nUntil the morning sun\nThis feeling never ends\nLove conquers all",
                 "Romantic Vibes",
                 "Moonlight Dance",
-                False  # Use local generation
+                True  # Try Modal if available
             ]
         ],
         inputs=[lyrics_input, artist_input, title_input, modal_toggle],
@@ -337,13 +402,15 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🎵 Music Content Creator") as de
     )
     
     # Add footer with info
-    gr.Markdown("""
+    gr.Markdown(f"""
     ---
     **💡 Tips:**
-    - **AI Generation:** Uses GPT-2 on Modal for creative, context-aware content
-    - **Local Templates:** Fast, rule-based generation that works offline
+    - **AI Generation:** Uses GPT-2 on Modal for creative, context-aware content {"(Currently experiencing authentication issues)" if not MODAL_AVAILABLE else ""}
+    - **Local Templates:** Fast, rule-based generation that always works
     - **Modal Credits:** Can be used for LLM inference and GPU-intensive functions ([Modal Docs](https://modal.com/docs))
     - **Best Results:** Include rich, descriptive lyrics for better theme detection
+    
+    **🔧 Current Status:** Modal connection {"✅ Working" if MODAL_AVAILABLE else "❌ Authentication Issue - Using Local Generation"}
     """)
 
 if __name__ == "__main__":