Update app.py

app.py

@@ -13,12 +13,9 @@ CHAT_HISTORY_DIR = "chat_histories"  # Directory to save chat histories
 # Ensure the directory exists
 os.makedirs(CHAT_HISTORY_DIR, exist_ok=True)
 
-# Streamlit Configurations
+# Streamlit Configurations (must come first)
 st.set_page_config(page_title="🐶 DUBSChat", layout="wide")
 
-with st.sidebar:
-    dubs_key = st.text_input("Enter Dubs Key", key="chatbot_api_key", type="password")
-
 # Utility Functions
 def save_chat_history(session_name, messages):
     """
@@ -44,28 +41,28 @@ def get_saved_sessions():
 
 # Sidebar Configuration
 with st.sidebar:
-    st.title("📂 Sessions")
+    st.title("📂 Dubs Recall")
     saved_sessions = get_saved_sessions()
     
     # Display saved sessions
     if saved_sessions:
-        selected_session = st.radio("Load a past session:", saved_sessions)
+        selected_session = st.radio("Past Sessions:", saved_sessions)
         if st.button("Load Session"):
             st.session_state["messages"] = load_chat_history(f"{selected_session}.json")
+            st.session_state["session_name"] = selected_session
             st.success(f"Loaded session: {selected_session}")
     else:
-        st.write("No saved sessions yet.")
-    
-    # Create new session
-    if st.button("Start New Session"):
-        session_name = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+        st.write("No past sessions available.")
+
+    # Reset chat
+    if st.button("🔄 Reset Chat"):
         st.session_state["messages"] = [
             {"role": "system", "content": "You are DUBS, a helpful assistant capable of conversing in a friendly and knowledgeable way."},
             {"role": "assistant", "content": "Hello! How can I assist you today?"}
         ]
-        st.session_state["session_name"] = session_name
-        save_chat_history(session_name, st.session_state["messages"])
-        st.success(f"New session started: {session_name}")
+        st.session_state["session_name"] = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+        save_chat_history(st.session_state["session_name"], st.session_state["messages"])
+        st.success("Chat reset and new session started.")
 
 # Chat History Initialization
 if "messages" not in st.session_state:
@@ -106,7 +103,7 @@ if prompt := st.chat_input():
             response = requests.post(
                 SPACE_URL,
                 json={"prompt": chat_history},
-                headers={"Authorization": f"Bearer {dubs_key}"},
+                headers={"Authorization": f"Bearer {API_KEY}"},
                 timeout=60,
             )
             response.raise_for_status()
@@ -121,3 +118,4 @@ if prompt := st.chat_input():
 
     # Auto-save the updated chat history
     save_chat_history(st.session_state["session_name"], st.session_state["messages"])
+