First Commit

app.py

@@ -19,7 +19,6 @@ DEFAULT_PROVIDER:Literal['hf-inference']="hf-inference"
 
 
 async def get_agent():
-    """Get or create the agent instance."""
     global agent_instance
     if agent_instance is None and HF_TOKEN:
         print("🔧 Creating new Agent instance ...")
@@ -27,19 +26,30 @@ async def get_agent():
         print(f"🤖 Model: {HF_MODEL}")
         print(f"Provider: {DEFAULT_PROVIDER}")
         try:
-            agent=Agent(model=HF_MODEL,provider="hf-inference",api_key=HF_TOKEN,
-                        servers=[{"type":"stdio","config":{
-                            "command":"python",
-                            "args":["mcp_server.py"],"cwd":".","env":{"HF_TOKEN": HF_TOKEN} if HF_TOKEN else {}}}])
+            agent = Agent(
+                model=HF_MODEL,
+                provider="hf-inference",
+                api_key=HF_TOKEN,
+                servers=[{
+                    "type": "stdio",
+                    "config": {
+                        "command": "python",
+                        "args": ["mcp_server.py"],
+                        "cwd": ".",
+                        "env": {"HF_TOKEN": HF_TOKEN} if HF_TOKEN else {}
+                    }
+                }]
+            )
             print("🚀 Agent instance created successfully")
             print("🔁 loading tools ...")
-            await agent_instance.load_tools()
+            await agent.load_tools()
+            agent_instance = agent
             print("✅ Tools loaded successfully")
-
         except Exception as e:
             print(f"❌ Error creating/loading agent: {str(e)}")
     return agent_instance
 
+
 @app.on_event("startup")
 async def startup_event():
     global agent_instance
@@ -80,10 +90,10 @@ def create_gradio_app():
         gr.Markdown("## 🤖 Model Card Chatbot\nAsk questions about Hugging Face model card")
         with gr.Row():
             model_id=gr.Textbox(label="MODEL ID", value="google/gemma-2-2b")
-            user_input=gr.Textbox(label="Your Question")
+            user_input=gr.Textbox(label="Your Question",value="Ask something about the model card .....")
             send=gr.Button("Ask")
             output=gr.Markdown()
-            send.click(fn=chat_function,inputs=[user_input,model_id],outputs=output)
+            send.click(fn=chat_function,inputs=[user_input,output,model_id],outputs=[output,user_input])
             return demo
 gradio_app=create_gradio_app()
 app=gr.mount_gradio_app(app,gradio_app,path="/")