Update src/streamlit_app.py

src/streamlit_app.py

@@ -149,20 +149,23 @@ welcome_messages = [
 
 # Page 1: Chatbot
 def chatbot_page():
-    # Initialize all session state variables
+    # Initialize session ID
     if "xane_id" not in st.session_state:
         st.session_state.xane_id = str(uuid.uuid4())
     
+    # Load all sessions from Supabase
     if "chat_sessions" not in st.session_state:
         st.session_state.chat_sessions = load_all_memory()
-        if not st.session_state.chat_sessions:
-            st.session_state.chat_sessions = {"Default": []}
     
+    # Set default chat if not present
     if "current_chat" not in st.session_state:
-        st.session_state.current_chat = "Default"
+        st.session_state.current_chat = (
+            list(st.session_state.chat_sessions.keys())[0] if st.session_state.chat_sessions else "Default"
+        )
     
-    if "messages" not in st.session_state:
-        st.session_state.messages = st.session_state.chat_sessions.get(st.session_state.current_chat, [])
+    # Load messages from selected chat
+    st.session_state.messages = st.session_state.chat_sessions.get(st.session_state.current_chat, [])
+
     
     if "show_greeting" not in st.session_state:
         st.session_state.show_greeting = True
@@ -171,7 +174,6 @@ def chatbot_page():
     if st.session_state.show_greeting:
         greeting = np.random.choice(welcome_messages)
         st.header(greeting)
-        st.session_state.show_greeting = False
     
     st.sidebar.title("💬 Chats")