Update app.py

app.py

@@ -28,6 +28,12 @@ safety_settings = [
 
 def generate_response(user_input, chat_history):
     """Generates a response based on user input and chat history."""
+    
+    # Add user input to history
+    chat_history.append(("user", user_input))
+
+    # Limit history length to the last 10 messages
+    chat_history = chat_history[-10:]
 
     # Create the generative model
     model = genai.GenerativeModel(
@@ -37,12 +43,6 @@ def generate_response(user_input, chat_history):
         system_instruction="You are Shadow the Hedgehog and you must act like Shadow the Hedgehog's personality.",
     )
 
-    # Add user input to history
-    chat_history.append({"role": "user", "content": user_input})
-
-    # Limit history length to the last 10 messages
-    chat_history = chat_history[-10:]
-
     retry_attempts = 3
     for attempt in range(retry_attempts):
         try:
@@ -50,18 +50,18 @@ def generate_response(user_input, chat_history):
             chat_session = model.start_chat()
 
             # Format the history for the model
-            formatted_history = "\n".join([f"{entry['role']}: {entry['content']}" for entry in chat_history])
+            formatted_history = "\n".join([f"{role}: {msg}" for role, msg in chat_history])
             response = chat_session.send_message(formatted_history)
 
             # Append the assistant's response to history
-            chat_history.append({"role": "assistant", "content": response.text})
+            chat_history.append(("assistant", response.text))
             return chat_history
 
         except Exception as e:
             if attempt < retry_attempts - 1:
                 continue
             else:
-                chat_history.append({"role": "assistant", "content": f"Error after {retry_attempts} attempts: {str(e)}"})
+                chat_history.append(("assistant", f"Error after {retry_attempts} attempts: {str(e)}"))
                 return chat_history
 
 # Build the Gradio interface using Chatbot and Button
@@ -78,11 +78,14 @@ with gr.Blocks() as iface:
         outputs=chatbot
     )
 
-    # Optional: Add an event to clear the input after submission
+    # Optional: Clear the input box after submission
+    def clear_input():
+        return ""
+
     user_input.submit(
-        fn=lambda x: "",  # Clear input box
-        inputs=[user_input],
-        outputs=[user_input]
-    )
+        fn=generate_response,
+        inputs=[user_input, chat_history_state],
+        outputs=chatbot
+    ).then(clear_input, outputs=[user_input])
 
 iface.launch()