First Commit

app.py

@@ -56,35 +56,26 @@ async def startup_event():
     agent_instance = await get_agent()
 
 
-def chat_function(user_message,history,model_id):
-
-    """Handles a user question by prompting the agent to read a model card and answer.
-    Args:
-        user_message (str): The user's question.
-        history (list): Chat history for display.
-        model_id (str): Hugging Face repo ID
-        
-    Returns:
-        Tuple[list,str]: Updated Chat history and empty string for clearing input."""
-    
+async def chat_function(user_message, history, model_id):
     prompt=f"""You're an assistant helping with hugging face model cards.
-    First, run the tool `read_model_card` on repo_id `{model_id}` to get the model card.
-    Then answer this user question based on the model card:
-    User question: {user_message}"""
-    history=history+[(user_message,None)]
+First, run the tool `read_model_card` on repo_id `{model_id}` to get the model card.
+Then answer this user question based on the model card:
+User question: {user_message}"""
+    history = history + [(user_message, None)]
     try:
-        response=""
-        for output in agent_instance.run(prompt):
-            if hasattr(output,"content") and output.content:
-                response=output.content
-        final_response=response or "⚠️ Sorry, I couldn't generate a response."
-        history[-1]=(user_message,final_response)
+        response = ""
+        async for output in agent_instance.run(prompt):
+            if hasattr(output, "content") and output.content:
+                response = output.content
+        final_response = response or "⚠️ Sorry, I couldn't generate a response."
+        history[-1] = (user_message, final_response)
     except Exception as e:
-        history[-1]=(user_message,f"⚠️ Error: {str(e)}")
+        history[-1] = (user_message, f"⚠️ Error: {str(e)}")
     return history, ""
 
 
 
+
 def create_gradio_app():
     with gr.Blocks(title="Model Card Chatbot") as demo:
         gr.Markdown("## 🤖 Model Card Chatbot\nAsk questions about Hugging Face model card")
@@ -92,8 +83,8 @@ def create_gradio_app():
             model_id=gr.Textbox(label="MODEL ID", value="google/gemma-2-2b")
             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,output,model_id],outputs=[output,user_input])
+            chatbot=gr.Chatbot(label="chat")
+            send.click(fn=chat_function,inputs=[user_input,chatbot,model_id],outputs=[chatbot,user_input])
             return demo
 gradio_app=create_gradio_app()
 app=gr.mount_gradio_app(app,gradio_app,path="/")

mcp_server.py

@@ -12,7 +12,7 @@ def read_model_card(repo_id:str)->str:
     """Tool to read and return the full model card from hugging face."""
     try:
         card=ModelCard.load(repo_id)
-        return json.dumps({"status":"success","readme":str(card)})
+        return json.dumps({"status":"success","readme": card.text})
     
     except FileNotFoundError:
         return json.dumps({"status":"error","message":"This model does not have a model card"})