Update app.py

app.py

@@ -54,17 +54,25 @@ def chat_interface():
                 send_button = gr.Button("Send", elem_id="send-btn")
 
         def submit_input(user_input, chat_history):
-            chat_history.append((user_input, ""))
+            # Initialize the chat history if it's empty
+            if not chat_history:
+                chat_history = []
+            chat_history.append({"role": "user", "content": user_input})
             return "", chat_history  # Clear input field
 
         input_textbox.submit(submit_input, [input_textbox, chat_output], [input_textbox, chat_output])
         send_button.click(submit_input, [input_textbox, chat_output], [input_textbox, chat_output])
 
         def handle_response(user_input, chat_history):
-            chat_history[-1] = (user_input, "")  # Update the last chat with user input
+            # Initialize the chat history if it's empty
+            if not chat_history:
+                chat_history = []
+            chat_history.append({"role": "user", "content": user_input})
+            
             response_stream = get_response(user_input)
             for partial_response in response_stream:
-                chat_history[-1] = (user_input, partial_response)
+                chat_history[-1] = {"role": "user", "content": user_input}  # Update the last user message with content
+                chat_history.append({"role": "assistant", "content": partial_response})  # Add assistant's response
                 yield "", chat_history  # Return the updated chat history progressively
 
         input_textbox.submit(handle_response, [input_textbox, chat_output], [input_textbox, chat_output])
@@ -76,13 +84,16 @@ def chat_interface():
         bottom: 10px;
         width: 100%;
         padding: 10px;
-        background-color: #f5f5f5;
+        background-color: #333;  /* Dark background */
         border-top: 1px solid #ddd;
     }
     #chat-box {
         height: calc(100vh - 100px); /* Adjust the height of chat history */
         overflow-y: scroll;
     }
+    #user-input, #send-btn {
+        background-color: #333;  /* Keep dark gray background */
+    }
     """
 
     return demo