Update app.py

app.py

@@ -21,37 +21,51 @@ try:
             # Try to connect to your deployed Modal functions using the new API
             logger.info("🔄 Attempting to connect to Modal functions...")
             
-            generate_content_with_llm = modal.Function.from_name(
-                "content-creation-agent",
-                "generate_content_with_llm"
-            )
-            
-            # Test connection with health check
-            health_check_func = modal.Function.from_name(
-                "content-creation-agent", 
-                "health_check"
-            )
-            
-            # Test if Modal is working
-            test_result = health_check_func.remote()
-            logger.info(f"✅ Modal health check passed: {test_result}")
-            
-            MODAL_AVAILABLE = True
-            logger.info("✅ Modal successfully connected")
-            
-        except Exception as e:
-            logger.warning(f"⚠️ Modal main function connection failed: {e}")
-            # Try fallback function instead
+            # First try the main AI function
             try:
                 generate_content_with_llm = modal.Function.from_name(
                     "content-creation-agent",
-                    "generate_fallback_content"
+                    "generate_content_with_llm"
                 )
+                
+                # Test connection with health check
+                health_check_func = modal.Function.from_name(
+                    "content-creation-agent", 
+                    "health_check"
+                )
+                
+                # Test if Modal is working
+                test_result = health_check_func.remote()
+                logger.info(f"✅ Modal health check passed: {test_result}")
+                
                 MODAL_AVAILABLE = True
-                logger.info("✅ Modal connected using fallback function")
-            except Exception as e2:
-                logger.warning(f"⚠️ Modal fallback also failed: {e2}")
-                MODAL_AVAILABLE = False
+                logger.info("✅ Modal successfully connected to main AI function")
+                
+            except Exception as e:
+                logger.warning(f"⚠️ Modal main function connection failed: {e}")
+                # Try fallback function instead (this one we know works)
+                try:
+                    generate_content_with_llm = modal.Function.from_name(
+                        "content-creation-agent",
+                        "generate_fallback_content"
+                    )
+                    
+                    # Still test health check
+                    health_check_func = modal.Function.from_name(
+                        "content-creation-agent", 
+                        "health_check"
+                    )
+                    test_result = health_check_func.remote()
+                    logger.info(f"✅ Modal health check passed: {test_result}")
+                    
+                    MODAL_AVAILABLE = True
+                    logger.info("✅ Modal connected using fallback function (still high quality!)")
+                except Exception as e2:
+                    logger.warning(f"⚠️ Modal fallback also failed: {e2}")
+                    MODAL_AVAILABLE = False
+        except Exception as main_e:
+            logger.warning(f"⚠️ Modal connection failed: {main_e}")
+            MODAL_AVAILABLE = False
     else:
         logger.warning("⚠️ MODAL_TOKEN not found in environment variables")
         logger.info("💡 To fix this, add MODAL_TOKEN to your HuggingFace Space environment variables")