Update app.py

app.py

@@ -40,7 +40,7 @@ def load_chat_history(session_id):
 def chat_history(session_id):
     return SQLChatMessageHistory(
         session_id=session_id,
-        connection=conn  # FIXED: Pass direct SQLite connection
+        connection="sqlite:///chat_history.db"  # ✅ FIXED: Use a connection string
     )
 
 # Generate unique session ID for each user
@@ -50,7 +50,7 @@ if "session_id" not in st.session_state:
 session_id = st.session_state.session_id
 chat_history_instance = chat_history(session_id)
 
-# Define Chat Prompt Template (FIXED)
+# Define Chat Prompt Template
 chat_prompt = ChatPromptTemplate(
     messages=[
         ('system', """You are an AI assistant specialized in Data Science tutoring. 
@@ -71,7 +71,7 @@ chain = chat_prompt | llm | out_parser
 # Define Runnable with message history
 chat = RunnableWithMessageHistory(
     chain,
-    lambda session: SQLChatMessageHistory(session, conn),  # FIXED: Pass connection correctly
+    lambda session: SQLChatMessageHistory(session, "sqlite:///chat_history.db"), 
     input_messages_key="prompt",
     history_messages_key="history"
 )