Update app.py

app.py

@@ -196,8 +196,36 @@ class BuiltInFloorManager:
 
 _agents_cache: Dict[str, OpenFloorAgent] = {}
 _floor_manager = BuiltInFloorManager()
+_initialized = False
+
+def _ensure_initialization():
+    """Ensure MCP server is properly initialized"""
+    global _initialized
+    if not _initialized:
+        print("🔧 Initializing MCP-OpenFloor Bridge...")
+        # Pre-discover agents from environment if available
+        env_agents = os.getenv("OPENFLOOR_AGENTS", "")
+        if env_agents:
+            env_urls = [url.strip() for url in env_agents.split(",") if url.strip()]
+            print(f"🌍 Pre-loading {len(env_urls)} agents from environment...")
+            for url in env_urls:
+                if url and url not in _agents_cache:
+                    try:
+                        agent = OpenFloorAgent(url)
+                        if agent.get_manifest():
+                            _agents_cache[url] = agent
+                            _floor_manager.register_agent_from_openfloor_agent(agent)
+                            print(f"✅ Pre-loaded agent: {agent.name} at {url}")
+                        else:
+                            print(f"❌ Failed to pre-load agent from: {url}")
+                    except Exception as e:
+                        print(f"❌ Error pre-loading agent {url}: {e}")
+        _initialized = True
+        print("🚀 MCP-OpenFloor Bridge initialized")
 
 def _discover_agents_from_env_and_headers(request: gr.Request) -> Dict[str, OpenFloorAgent]:
+    _ensure_initialization()
+    
     agent_urls = []
     
     # Environment variable
@@ -234,13 +262,16 @@ def _discover_agents_from_env_and_headers(request: gr.Request) -> Dict[str, Open
     # Register agents
     for url in unique_urls:
         if url and url not in _agents_cache:
-            agent = OpenFloorAgent(url)
-            if agent.get_manifest():
-                _agents_cache[url] = agent
-                _floor_manager.register_agent_from_openfloor_agent(agent)
-                print(f"✅ Discovered and registered agent: {agent.name} at {url}")
-            else:
-                print(f"❌ Failed to get manifest from: {url}")
+            try:
+                agent = OpenFloorAgent(url)
+                if agent.get_manifest():
+                    _agents_cache[url] = agent
+                    _floor_manager.register_agent_from_openfloor_agent(agent)
+                    print(f"✅ Discovered and registered agent: {agent.name} at {url}")
+                else:
+                    print(f"❌ Failed to get manifest from: {url}")
+            except Exception as e:
+                print(f"❌ Error discovering agent {url}: {e}")
     
     return _agents_cache