Update app.py

app.py

@@ -9,20 +9,24 @@ from langchain_community.chat_message_histories import SQLChatMessageHistory
 from langchain_core.runnables.history import RunnableWithMessageHistory
 
 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))''')
+# Database file name
+DB_PATH = "database.db"
 
-# Commit the changes
-conn.commit()
-
-# Close the connection
-conn.close()
+# Function to create a database and table if not exists
+def create_database():
+    if not os.path.exists(DB_PATH):  # Check if database exists
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
 
+        # Create a sample table
+        cursor.execute("""
+            CREATE TABLE IF NOT EXISTS users (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                name TEXT NOT NULL,
+                age INTEGER NOT NULL
+            )
+        """)
+        conn.commit()
+        conn.close()
+        st.success("Database and table created successfully!")
+