Update app.py

app.py

@@ -29,21 +29,23 @@ def chat(user_input, history):
     # Combine knowledge base + chat history + current input
     context_text = f"Use the following knowledge base to answer the user's question:\n\n{knowledge_base}\n\n"
     context_text += "Previous chat history:\n"
-    for user_msg, bot_msg in history:
-        context_text += f"User: {user_msg}\nBot: {bot_msg}\n"
+    for msg in history:
+        context_text += f"{msg['role'].capitalize()}: {msg['content']}\n"
     context_text += f"User: {user_input}\nAnswer based on the knowledge above."
 
-    # Send everything to Gemini (as string)
     try:
         client = genai.Client(api_key=global_api_key)
         chat_session = client.chats.create(model="gemini-2.5-flash")
-        response = chat_session.send_message(context_text)  # send as string
+        response = chat_session.send_message(context_text)
         reply = response.text
     except Exception as e:
         reply = f"❌ Error calling Gemini model: {e}"
 
-    history.append((user_input, reply))  # store chat history
-    return history, ""
+    # Append messages in new Gradio format
+    history.append({"role": "user", "content": user_input})
+    history.append({"role": "assistant", "content": reply})
+
+    return history, ""  # "" clears the user input box
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
@@ -54,15 +56,17 @@ with gr.Blocks() as demo:
     set_key_btn = gr.Button("Set API Key")
     api_status = gr.Textbox(label="Status", interactive=False)
 
+    # Store chat history in session with initial greeting
+    chat_history = gr.State([
+        {"role": "system", "content": "👋 Welcome to Arduino Expert! Ask me anything about Arduino projects."}
+    ])
+
     # Chatbot interface
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(type="messages")
     user_input = gr.Textbox(label="Ask a question")
     send_btn = gr.Button("Send")
     clear_btn = gr.Button("Clear Chat")
 
-    # Store chat history in session
-    chat_history = gr.State([])
-
     # Set API key
     set_key_btn.click(fn=set_api_key, inputs=api_key, outputs=api_status)
 
@@ -71,12 +75,6 @@ with gr.Blocks() as demo:
     user_input.submit(fn=chat, inputs=[user_input, chat_history], outputs=[chatbot, user_input])
 
     # Clear chat
-    clear_btn.click(lambda: [], None, chatbot)
-
-    # Initial greeting message
-    def greeting():
-        return [("System", "👋 Welcome to Arduino Expert! Ask me anything about Arduino projects.")]
-
-    chatbot.update(greeting())
+    clear_btn.click(lambda: [{"role": "system", "content": "👋 Welcome to Arduino Expert! Ask me anything about Arduino projects."}], None, chatbot)
 
 demo.launch(share=True)