Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import os
 import requests
 import json
+from streamlit_chat import message
 
 # Initialize a global variable to store the entire assistant response
 entire_assistant_response = ""
@@ -69,26 +70,67 @@ def get_streamed_response(message, history):
     return entire_assistant_response
 
 # Streamlit application
+st.set_page_config(page_title="Raxder unofficial AI", page_icon="🤖")
+
 st.sidebar.title("Raxder unofficial AI")
 st.sidebar.write("This is NOT an AI Therapist, use it at your OWN RISK! This might be the worst AI you have ever used.")
 
-# Initialize session state for history
+history = []
 if "history" not in st.session_state:
     st.session_state.history = []
 
-# Create a placeholder for the chat history
+# Create a container for the chat history
 chat_placeholder = st.container()
 
-# Bottom input box
+# Accept user input at the bottom
+st.markdown("""
+    <style>
+    .input-container {
+        position: fixed;
+        bottom: 0;
+        width: 100%;
+        background-color: #f0f2f6;
+        padding: 10px;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
 user_input = st.text_input("You:", key="user_input")
 send_button = st.button("Send")
 
 if send_button and user_input:
-    response = get_streamed_response(user_input, st.session_state.history)
-    st.session_state.history.append((user_input, response))
+    history = st.session_state.history
+    response = get_streamed_response(user_input, history)
+    history.append((user_input, response))
+    st.session_state.history = history
 
-# Display chat history
+# Display chat history with colors
 with chat_placeholder:
     for human, assistant in st.session_state.history:
-        st.write(f"You: {human}")
-        st.write(f"notDave: {assistant}")
+        message(human, is_user=True, key=f"user_{human}")
+        message(assistant, is_user=False, key=f"assistant_{assistant}")
+
+# Add user input container at the bottom
+st.markdown("""
+    <div class="input-container">
+        <input type="text" id="input_box" placeholder="You:">
+        <button onclick="send_message()">Send</button>
+    </div>
+""", unsafe_allow_html=True)
+
+def send_message():
+    user_input = st.session_state.get("user_input", "")
+    if user_input:
+        history = st.session_state.history
+        response = get_streamed_response(user_input, history)
+        history.append((user_input, response))
+        st.session_state.history = history
+        st.session_state.user_input = ""
+    
+    chat_placeholder.empty()
+    with chat_placeholder:
+        for human, assistant in st.session_state.history:
+            message(human, is_user=True, key=f"user_{human}")
+            message(assistant, is_user=False, key=f"assistant_{assistant}")
+
+st.button("Send", on_click=send_message)