Update app.py

app.py

@@ -1,51 +1,41 @@
 import os
 import gradio as gr
 import asyncio
-from smolagents import Agent
-from smolagents.mcp import MCPServerHTTP
-from smolagents.models import HuggingFaceModel
-from smolagents.providers import HuggingFaceProvider
-
-HF_TOKEN = os.getenv("HF_TOKEN")
-
-# MCP Server SSE URL
-SSE_URL = "https://oppaai-job-search-mcp-server.hf.space/gradio_api/mcp/sse"
-server = MCPServerHTTP(url=SSE_URL)
-
-# Use HF Qwen3-8B model from Hugging Face Hub
-hf_model = HuggingFaceModel(
-    model_name="Qwen/Qwen-3-8B",
-    provider=HuggingFaceProvider(
-        api_token=HF_TOKEN
-    )
-)
-
-# Create Agent with MCP Server
-agent = Agent(
-    model=hf_model,
-    mcp_servers=[server],
-    instructions="""
-    Your name is Jobcy. You are an AI assistant designed to help users find remote jobs by searching through job listings from various sources, including the Jobicy API and other platforms.
-    You will list the job listings in a structured format, including the job title, company, location, and the Google search link.
-    """,
-    stream=False
-)
-
-async def chat_with_agent(user_input, history):
-    history = history or []
-    history.append(("user", user_input))
-
-    result = await agent.run(user_input)
-
-    history.append(("agent", result.output))  # result.output contains the answer text
-    return history, history
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    user_input = gr.Textbox(placeholder="Ask Jobcy about remote jobs or anything else...", label="Your Message")
-    state = gr.State([])
-
-    user_input.submit(chat_with_agent, inputs=[user_input, state], outputs=[chatbot, state])
-
-demo.title = "Jobcy Remote Job Search Assistant"
-demo.launch()
+from smolagents import CodeAgent, MCPClient, InferenceClientModel
+
+async def main():
+    mcp_client = None
+    try:
+        # Initialize MCPClient for your MCP server
+        mcp_client = MCPClient({"url": "https://oppaai-job-search-mcp-server.hf.space/gradio_api/mcp/sse"})
+        
+        # Fetch tools exposed by the MCP server
+        tools = await mcp_client.get_tools()
+        
+        # Initialize the model with HF token from env
+        model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
+        
+        # Create the agent with the fetched tools and model
+        agent = CodeAgent(tools=tools, model=model)
+        
+        async def chat_with_agent(message, history):
+            history = history or []
+            result = await agent.run(message)
+            history.append((message, result))
+            return history, history
+        
+        with gr.Blocks() as demo:
+            chatbot = gr.Chatbot()
+            user_input = gr.Textbox(placeholder="Ask Jobcy about remote jobs or anything else...", label="Your Message")
+            state = gr.State([])
+
+            user_input.submit(chat_with_agent, inputs=[user_input, state], outputs=[chatbot, state])
+
+        demo.title = "Jobcy Remote Job Search Assistant"
+        demo.launch()
+    finally:
+        if mcp_client:
+            await mcp_client.disconnect()
+
+if __name__ == "__main__":
+    asyncio.run(main())