Update app.py

app.py

@@ -1,25 +1,32 @@
+# app.py
 import os
 import gradio as gr
 import google.generativeai as genai
 
-# --- Gemini -------------------------------------------------------
+# Configure the Gemini client using an environment variable
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+
+# Create the model and start a stateful chat session
 model = genai.GenerativeModel("gemini-2.0-flash")
-chat  = model.start_chat()                     # keeps its own history
+chat = model.start_chat(history=[])
 
+# Define the function that Gradio will call on each user message
 def respond(message, history):
-    return chat.send_message(message).text     # the reply only
+    # Send the user's message to the chat session
+    response = chat.send_message(message)
+    # Return only the model's text response
+    return response.text
 
-# --- Gradio -------------------------------------------------------
+# Build the Gradio UI
+# The retry_btn, undo_btn, and clear_btn arguments have been removed
+# as this functionality is now integrated into the Chatbot component.
 iface = gr.ChatInterface(
-    fn          = respond,
-    title       = "Gemini Chatbot",
-    description = "Ask anything – powered by Google Gemini",
-    chatbot     = gr.Chatbot(height=600, type="messages"),  # <- new type
-    textbox     = gr.Textbox(placeholder="Type a message…"),
-    retry_btn   = "🔄 Retry",   # works on Gradio >= 4.19
-    clear_btn   = "🗑️ Clear",
+    fn=respond,
+    title="Gemini Chatbot",
+    chatbot=gr.Chatbot(height=600, type="messages"),
+    textbox=gr.Textbox(placeholder="Ask anything…", container=False, scale=7),
 )
 
+# Launch the application
 if __name__ == "__main__":
     iface.launch()