Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
 import os
-from smolagents import InferenceClientModel, CodeAgent
-from smolagents.mcp_client import MCPClient   # ✅ fixed import
 
-mcp_client = None
-demo = None
+from smolagents import InferenceClientModel, CodeAgent, MCPClient
+
 
 try:
     mcp_client = MCPClient(
@@ -13,11 +11,7 @@ try:
     tools = mcp_client.get_tools()
 
     model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
-    agent = CodeAgent(
-        tools=[*tools],
-        model=model,
-        additional_authorized_imports=["json", "ast", "urllib", "base64"]
-    )
+    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
 
     demo = gr.ChatInterface(
         fn=lambda message, history: str(agent.run(message)),
@@ -27,19 +21,6 @@ try:
         description="This is a simple agent that uses MCP tools to answer questions.",
     )
 
-except Exception as e:
-    print("❌ MCP connection failed:", e)
-    # Fallback UI so Hugging Face Space still runs
-    demo = gr.Interface(
-        fn=lambda text: f"MCP unavailable. Echo: {text}",
-        inputs="text",
-        outputs="text",
-        title="Fallback Mode",
-        description="MCP server not reachable. Running in fallback mode."
-    )
-finally:
-    if mcp_client:
-        mcp_client.disconnect()
-
-if demo:
     demo.launch()
+finally:
+    mcp_client.disconnect()