Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Simplified Modal authentication - no complex setup needed
+# Modal setup with proper secret handling
 MODAL_AVAILABLE = False
 generate_content_with_llm = None
 
@@ -14,17 +14,23 @@ try:
     import modal
     logger.info(f"Modal version: {modal.__version__}")
     
-    # Check for Modal token in environment (set by HuggingFace)
-    if os.environ.get("MODAL_TOKEN"):
+    # Check for Modal token - this should be set in HuggingFace environment
+    modal_token = os.environ.get("MODAL_TOKEN")
+    if modal_token:
         try:
-            # Try to connect to your Modal functions
-            generate_content_with_llm = modal.Function.from_name(
+            # Set up Modal client with token
+            os.environ["MODAL_TOKEN"] = modal_token
+            
+            # Try to connect to your deployed Modal functions
+            logger.info("🔄 Attempting to connect to Modal functions...")
+            
+            generate_content_with_llm = modal.Function.lookup(
                 "content-creation-agent",
                 "generate_content_with_llm"
             )
             
             # Test connection with health check
-            health_check_func = modal.Function.from_name(
+            health_check_func = modal.Function.lookup(
                 "content-creation-agent", 
                 "health_check"
             )
@@ -37,10 +43,10 @@ try:
             logger.info("✅ Modal successfully connected")
             
         except Exception as e:
-            logger.warning(f"⚠️ Modal connection failed: {e}")
+            logger.warning(f"⚠️ Modal main function connection failed: {e}")
             # Try fallback function instead
             try:
-                generate_content_with_llm = modal.Function.from_name(
+                generate_content_with_llm = modal.Function.lookup(
                     "content-creation-agent",
                     "generate_fallback_content"
                 )
@@ -48,11 +54,14 @@ try:
                 logger.info("✅ Modal connected using fallback function")
             except Exception as e2:
                 logger.warning(f"⚠️ Modal fallback also failed: {e2}")
+                MODAL_AVAILABLE = False
     else:
-        logger.warning("⚠️ MODAL_TOKEN not found in environment")
+        logger.warning("⚠️ MODAL_TOKEN not found in environment variables")
         
 except ImportError:
     logger.warning("⚠️ Modal package not available")
+except Exception as e:
+    logger.warning(f"⚠️ Modal setup failed: {e}")
 
 logger.info(f"Modal Status: {'✅ Available' if MODAL_AVAILABLE else '❌ Unavailable'}")