Update app.py

app.py

@@ -1,21 +1,30 @@
 import streamlit as st
 from chatbot.conversation import chat_bot
-from chatbot.database import get_chat_history, save_chat
-
-
-# Generate or retrieve session ID
+from chatbot.database import get_chat_history, save_chat, get_chat_sessions
 from config import generate_session_id
 
-if "session_id" not in st.session_state:
-    st.session_state.session_id = generate_session_id()
-
+# User ID input
 st.title("DataScience-AI-Mentor")
 st.write("Your AI-powered Data Science Tutor. Ask me anything related to Data Science!")
 
+if "user_id" not in st.session_state:
+    user_input_id = st.text_input("Enter your User ID (or leave blank to generate a new one):")
+    if user_input_id:
+        st.session_state.user_id = user_input_id
+    else:
+        from config import generate_user_id
+        st.session_state.user_id = generate_user_id()
+        st.write(f"Your new User ID: {st.session_state.user_id}")
+
+# Select or create a chat session
+session_options = get_chat_sessions(st.session_state.user_id) + ["Start New Chat"]
+st.session_state.session_id = st.selectbox("Select a chat session:", session_options)
 
+if st.session_state.session_id == "Start New Chat":
+    st.session_state.session_id = generate_session_id()
 
 # Display chat history
-history = get_chat_history(st.session_state.session_id)
+history = get_chat_history(st.session_state.user_id, st.session_state.session_id)
 for user_msg, bot_resp in history:
     with st.chat_message("user"):
         st.write(user_msg)
@@ -25,7 +34,7 @@ for user_msg, bot_resp in history:
 # User input
 user_input = st.chat_input("Type your message...")
 if user_input:
-    response, session_id = chat_bot(user_input, st.session_state.session_id)
+    response = chat_bot(user_input, st.session_state.session_id)
     
     # Display messages in chat
     with st.chat_message("user"):
@@ -34,4 +43,4 @@ if user_input:
         st.write(response)
     
     # Save chat history
-    save_chat(session_id, user_input, response)
+    save_chat(st.session_state.user_id, st.session_state.session_id, user_input, response)