Update app.py

app.py

@@ -1,41 +1,54 @@
 import os
 import gradio as gr
-import asyncio
-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())
+from huggingface_hub import InferenceClient
+from tinyagents import Agent
+
+# Set your Hugging Face API token
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Define the MCP server URL
+MCP_URL = "https://oppaai-job-search-mcp-server.hf.space/gradio_api/mcp/sse"
+
+# Initialize the MCP client
+mcp_client = InferenceClient(
+    url=MCP_URL,
+    token=HF_TOKEN
+)
+
+# Fetch the tools exposed by the MCP server
+tools = mcp_client.get_tools()
+
+# Define the model to be used
+model = InferenceClient(
+    model="Qwen/Qwen3-8B",
+    token=HF_TOKEN
+)
+
+# Create the agent
+agent = Agent(
+    model=model,
+    tools=tools
+)
+
+# Define the function to handle user input and generate responses
+def chat_fn(user_message, history):
+    history = history or []
+    history.append(("user", user_message))
+    response = agent.run(user_message)
+    history.append(("agent", response))
+    return history, history
+
+# Set up the Gradio interface
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    user_input = gr.Textbox(placeholder="Ask Jobcy about remote jobs or anything else...")
+    state = gr.State([])
+
+    user_input.submit(chat_fn, inputs=[user_input, state], outputs=[chatbot, state])
+
+demo.title = "Jobcy Remote Job Search Assistant (tinyagents)"
+demo.launch()
+
+# Disconnect the MCP client when done
+import atexit
+atexit.register(mcp_client.disconnect)