first commit

app.py

@@ -24,9 +24,7 @@ DEFAULT_PROVIDER: Literal['hf-inference'] = "hf-inference"
 async def get_agent():
     global agent_instance
     if agent_instance is None and HF_TOKEN:
-        print("🔧 Creating new Agent instance ...")
-        print(f"✅ HF_TOKEN present: {bool(HF_TOKEN)}")
-        print(f"🤖 Model: {HF_MODEL}")
+        print("🔧 Creating Agent...")
         try:
             agent = Agent(
                 model=HF_MODEL,
@@ -38,21 +36,20 @@ async def get_agent():
                         "command": "python",
                         "args": ["mcp_server.py"],
                         "cwd": ".",
-                        "env": {"HF_TOKEN": HF_TOKEN} if HF_TOKEN else {}
+                        "env": {"HF_TOKEN": HF_TOKEN}
                     }
                 }]
             )
             await agent.load_tools()
             agent_instance = agent
-            print("✅ Agent is ready")
+            print("✅ Agent is ready!")
         except Exception as e:
-            print(f"❌ Error creating/loading agent: {str(e)}")
+            print(f"❌ Failed to create agent: {str(e)}")
     return agent_instance
 
 @app.on_event("startup")
 async def startup_event():
-    global agent_instance
-    agent_instance = await get_agent()
+    await get_agent()
 
 def chat_function(user_message, history, model_id):
     global agent_instance
@@ -69,7 +66,8 @@ User question: {user_message}
             if hasattr(output, "content") and output.content:
                 response = output.content
         if not response:
-            response = "⚠️ Sorry, I couldn't generate a response."
+            response = "⚠️ Sorry, no answer generated."
+        # Append as new OpenAI-style message
         history = history + [{"role": "user", "content": user_message}]
         history.append({"role": "assistant", "content": response})
     except Exception as e:
@@ -78,18 +76,22 @@ User question: {user_message}
     return history, ""
 
 def create_gradio_app():
-    with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet"), title="🤖 Model Card Chatbot") as demo:
+    with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), title="🤖 Model Card Chatbot") as demo:
         gr.Markdown("""
         # 🤖 **Model Card Chatbot**
-        Ask anything about a model's card on Hugging Face.
+        _Ask anything about a Hugging Face model's card_
         """)
+        chatbot = gr.Chatbot(label="Chat", type="messages", show_copy_button=True, height=400)
         with gr.Row():
             model_id = gr.Textbox(label="Model ID", value="google/gemma-2-2b", scale=2)
-            user_input = gr.Textbox(label="Your Question", placeholder="e.g., What is this model trained on?", scale=3)
-            send = gr.Button("🔍 Ask", scale=1)
-        chatbot = gr.Chatbot(label="Chat", type="messages")
+            user_input = gr.Textbox(
+                label="Your Question", 
+                placeholder="e.g., What dataset was it trained on?", 
+                scale=4
+            )
+            ask_button = gr.Button("🔍 Ask", scale=1)
         
-        send.click(
+        ask_button.click(
             fn=chat_function,
             inputs=[user_input, chatbot, model_id],
             outputs=[chatbot, user_input]