first commit

app.py

@@ -62,7 +62,6 @@ First, run the tool `read_model_card` on repo_id `{model_id}` to get the model c
 Then answer this user question based on the model card:
 User question: {user_message}
 """
-    history = history + [(user_message, None)]
     try:
         response = ""
         outputs = agent_instance.run(prompt)
@@ -71,13 +70,15 @@ User question: {user_message}
                 response = output.content
         if not response:
             response = "⚠️ Sorry, I couldn't generate a response."
-        history[-1] = (user_message, response)
+        history = history + [{"role": "user", "content": user_message}]
+        history.append({"role": "assistant", "content": response})
     except Exception as e:
-        history[-1] = (user_message, f"⚠️ Error: {str(e)}")
+        history = history + [{"role": "user", "content": user_message}]
+        history.append({"role": "assistant", "content": f"⚠️ Error: {str(e)}"})
     return history, ""
 
 def create_gradio_app():
-    with gr.Blocks(theme=gr.themes.Soft(), title="🤖 Model Card Chatbot") as demo:
+    with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet"), title="🤖 Model Card Chatbot") as demo:
         gr.Markdown("""
         # 🤖 **Model Card Chatbot**
         Ask anything about a model's card on Hugging Face.
@@ -86,7 +87,7 @@ def create_gradio_app():
             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")
+        chatbot = gr.Chatbot(label="Chat", type="messages")
         
         send.click(
             fn=chat_function,