Update app.py

app.py

@@ -42,7 +42,7 @@ def generate_response(user_input, chat_history):
     )
 
     # Add user input to history
-    chat_history.append(user_input)
+    chat_history.append(("user", user_input))
 
     # Limit history length to the last 10 messages
     chat_history = chat_history[-10:]
@@ -54,15 +54,28 @@ def generate_response(user_input, chat_history):
             chat_session = model.start_chat()
 
             # Send the entire chat history as the first message
-            response = chat_session.send_message("\n".join(chat_history))
-            return response.text, chat_history
+            response = chat_session.send_message("\n".join([f"{role}: {msg}" for role, msg in chat_history]))
+            chat_history.append(("assistant", response.text))
+            return chat_history
 
         except Exception as e:
             if attempt < retry_attempts - 1:
                 continue
             else:
-                return f"Error after {retry_attempts} attempts: {str(e)}", chat_history
+                chat_history.append(("assistant", f"Error after {retry_attempts} attempts: {str(e)}"))
+                return chat_history
 
 # Build the Gradio interface using ChatInterface
-with gr.ChatInterface(fn=generate_response) as iface:
-    iface.launch()
+with gr.Blocks() as iface:
+    chatbot = gr.Chatbot()  # Create a Chatbot component
+    user_input = gr.Textbox(label="Talk to AI", placeholder="Enter your message here...")
+    chat_history_state = gr.State([])  # State input for chat history
+
+    # Define the layout and components
+    user_input.submit(
+        fn=generate_response,
+        inputs=[user_input, chat_history_state],
+        outputs=chatbot
+    )
+
+iface.launch()