Update app.py

app.py

@@ -1,7 +1,8 @@
 import asyncio
-import re
 import os
+import re
 import gradio as gr
+
 from pydantic_ai import Agent
 from pydantic_ai.mcp import MCPServerHTTP
 from pydantic_ai.models.gemini import GeminiModel
@@ -11,42 +12,60 @@ from pydantic_ai.providers.google_gla import GoogleGLAProvider
 SSE_URL = "https://oppaai-job-search-mcp-server.hf.space/gradio_api/mcp/sse"
 server = MCPServerHTTP(url=SSE_URL)
 
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")  # Make sure your GEMINI_API_KEY is set in environment variables
+# Load Gemini API Key from environment
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
-# Use OpenRouter hosted LLM Qwen3-30B-A3B model
+# Define model
 model = GeminiModel(
-    'gemini-2.0-flash', 
-    provider=GoogleGLAProvider(api_key=GEMINI_API_KEY)
+    "gemini-2.0-flash",
+    provider=GoogleGLAProvider(api_key=GEMINI_API_KEY),
+    thinking_budget=0  # Optional: disables internal thinking token usage
 )
 
-# Create Agent with MCP Server
+# Define agent
 agent = Agent(
     model=model,
     mcp_servers=[server],
     instructions="""
-    Your name is Jobcy. You are an AI assistant designed to help users to 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.
+    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.
+    Return job listings in a structured format: job title, company, location, and a Google search link.
     """
 )
 
+# Async chat function with timeout protection
 async def chat_with_agent(user_input, history):
     history = history or []
-    result = await agent.run(user_input)
-    history.append(("User", user_input))
-    history.append(("Jobcy", result))
+    history.append(("You", user_input))
+    try:
+        result = await asyncio.wait_for(agent.run(user_input), timeout=30)
+        history.append(("Jobcy", result))
+    except asyncio.TimeoutError:
+        history.append(("Jobcy", "⏳ Sorry, I took too long to respond. Try again in a moment."))
+    except Exception as e:
+        history.append(("Jobcy", f"⚠️ Error: {type(e).__name__}: {e}"))
     return history, history
 
-async def main():
-    async with agent.run_mcp_servers():
-        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([])
+# Main app UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 💼 Jobcy – Your Remote Job Assistant")
+    chatbot = gr.Chatbot()
+    user_input = gr.Textbox(
+        placeholder="Ask Jobcy about remote jobs...",
+        label="Your Message",
+        lines=2
+    )
+    state = gr.State([])
 
-            user_input.submit(chat_with_agent, inputs=[user_input, state], outputs=[chatbot, state])
+    user_input.submit(chat_with_agent, inputs=[user_input, state], outputs=[chatbot, state])
 
-            demo.title = "Jobcy Remote Job Search Assistant"
-            demo.launch(share=True)
+    gr.Examples(
+        examples=[
+            "Find remote software engineer jobs in Canada",
+            "Show me part-time data analyst jobs I can do from home",
+            "List any remote marketing jobs with no degree required",
+        ],
+        inputs=user_input
+    )
 
-if __name__ == "__main__":
-    asyncio.run(main())
+    demo.queue(concurrency_count=2)
+    demo.launch(share=True)