Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import httpx
-from fastmcp import FastMCP
+from fastmcp import FastMCP, Client
 
 mcp = FastMCP("Indeed Job Search Agent")
 
@@ -29,32 +29,34 @@ async def search_jobs_tool(query: str, location: str, date_created: str = None,
         except httpx.RequestError as e:
             return {"error": str(e)}
 
-def search_jobs_ui(query, location, date_created=None, salary=None, job_type=None):
-    # Use call_tool to call the async MCP tool synchronously
-    result = mcp.call_tool("search_jobs", {
-        "query": query,
-        "location": location,
-        "date_created": date_created,
-        "salary": salary,
-        "job_type": job_type
-    })
-
-    if "error" in result:
-        return f"❌ Error: {result['error']}"
+async def search_jobs_ui(query, location, date_created=None, salary=None, job_type=None):
+    async with Client(mcp) as client:
+        try:
+            result = await client.call_tool("search_jobs", {
+                "query": query,
+                "location": location,
+                "date_created": date_created,
+                "salary": salary,
+                "job_type": job_type
+            })
+            if "error" in result:
+                return f"❌ Error: {result['error']}"
 
-    jobs = result.get("results", [])
-    if not jobs:
-        return "No jobs found for your search."
+            jobs = result.get("results", [])
+            if not jobs:
+                return "No jobs found for your search."
 
-    output = ""
-    for job in jobs:
-        title = job.get("title", "No Title")
-        company = job.get("company", "Unknown Company")
-        job_location = job.get("location", "Unknown Location")
-        job_url = job.get("url", "#")
-        output += f"**{title}** at *{company}*\n📍 {job_location}\n[Apply Here]({job_url})\n\n"
+            output = ""
+            for job in jobs:
+                title = job.get("title", "No Title")
+                company = job.get("company", "Unknown Company")
+                job_location = job.get("location", "Unknown Location")
+                job_url = job.get("url", "#")
+                output += f"**{title}** at *{company}*\n📍 {job_location}\n[Apply Here]({job_url})\n\n"
 
-    return output
+            return output
+        except Exception as e:
+            return f"❌ An error occurred: {str(e)}"
 
 app = gr.Interface(
     fn=search_jobs_ui,
@@ -71,4 +73,4 @@ app = gr.Interface(
 )
 
 if __name__ == "__main__":
-    app.launch(share=True, mcp_server=True)
+    app.launch(mcp_server=True)