Update app.py

app.py

@@ -8,4 +8,21 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_community.chat_message_histories import SQLChatMessageHistory
 from langchain_core.runnables.history import RunnableWithMessageHistory
 
-st.write('hello welecome to data science tutor')
+st.write('hello welecome to data science tutor')
+db=chat_history_data
+conn=st.connection('db',type='sql')
+cursor=conn.cursor()
+# Create a table to store the session history (if it doesn't already exist)
+cursor.execute('''CREATE TABLE IF NOT EXISTS message_history (
+                    session_id TEXT NOT NULL,
+                    user_message TEXT NOT NULL,
+                    assistant_message TEXT,
+                    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
+                    PRIMARY KEY (session_id, timestamp))''')
+
+# Commit the changes
+conn.commit()
+
+# Close the connection
+conn.close()
+