Update app.py

app.py

@@ -28,9 +28,9 @@ 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_input)
+    
+    # Add user input to history as a tuple
+    chat_history.append(("user", user_input))
 
     # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
@@ -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(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(response.text)
+            # Append the assistant's response to history as a tuple
+            chat_history.append(("assistant", response.text))
             return chat_history
 
         except Exception as e:
             if attempt < retry_attempts - 1:
                 continue
             else:
-                chat_history.append(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