Update app.py

app.py

@@ -40,13 +40,19 @@ def load_chat_history(session_id):
 def chat_history(session_id):
     return SQLChatMessageHistory(
         session_id=session_id,
-        connection="sqlite:///chat_history.db"  # ✅ FIXED: Use a connection string
+        connection="sqlite:///chat_history.db"
     )
 
 # Generate unique session ID for each user
 if "session_id" not in st.session_state:
     st.session_state.session_id = str(uuid.uuid4())
 
+# "New Chat" button
+if st.button("🆕 New Chat"):
+    st.session_state.session_id = str(uuid.uuid4())  # Generate new session ID
+    st.session_state.messages = []  # Clear chat history
+    st.rerun()  # Refresh the app
+
 session_id = st.session_state.session_id
 chat_history_instance = chat_history(session_id)
 
@@ -71,7 +77,7 @@ chain = chat_prompt | llm | out_parser
 # Define Runnable with message history
 chat = RunnableWithMessageHistory(
     chain,
-    lambda session: SQLChatMessageHistory(session, "sqlite:///chat_history.db"), 
+    lambda session: SQLChatMessageHistory(session, "sqlite:///chat_history.db"),
     input_messages_key="prompt",
     history_messages_key="history"
 )
@@ -80,16 +86,20 @@ chat = RunnableWithMessageHistory(
 st.title("Conversational AI Data Science Tutor")
 st.write("Ask me anything about Data Science!")
 
-# Load chat history
-st.session_state.setdefault("messages", load_chat_history(session_id))
+# Load chat history and keep it sticky
+if "messages" not in st.session_state:
+    st.session_state.messages = load_chat_history(session_id)
+
+chat_container = st.container()
 
-# Display chat history
-for role, content in st.session_state.messages:
-    with st.chat_message(role):
-        st.markdown(content)
+# Display chat history in a sticky format
+with chat_container:
+    for role, content in st.session_state.messages:
+        with st.chat_message(role):
+            st.markdown(content)
 
-# User input
-user_input = st.text_input("You:", "", key="user_input")
+# User input stays at the bottom
+user_input = st.text_input("Type your message here:", "", key="user_input")
 
 if user_input:
     save_message(session_id, "user", user_input)
@@ -102,5 +112,10 @@ if user_input:
     save_message(session_id, "assistant", response)
     st.session_state.messages.append(("assistant", response))
     
-    with st.chat_message("assistant"):
-        st.markdown(response)
+    # Refresh the chat container but keep history sticky
+    with chat_container:
+        with st.chat_message("assistant"):
+            st.markdown(response)
+
+    # Clear input field
+    st.session_state.user_input = ""