Update database/models.py

database/models.py

@@ -3,19 +3,20 @@ from datetime import datetime
 import mysql.connector
 from config import Config
 
-
 def create_tables():
+    # First, create the database if it doesn't exist
     conn = mysql.connector.connect(
-        host='127.0.0.1',
-        port=3306,  # explicitly specify port
-        user='root',
-        password='your_password',  # make sure this is your actual MySQL root password
-        database='ragchatbot',
-        auth_plugin='mysql_native_password'  # explicitly specify auth method
+        host=Config.MYSQL_HOST,
+        user=Config.MYSQL_USER,
+        password=Config.MYSQL_PASSWORD
     )
-
     cursor = conn.cursor()
     
+    # Create database if it doesn't exist
+    cursor.execute(f"CREATE DATABASE IF NOT EXISTS {Config.MYSQL_DB}")
+    cursor.execute(f"USE {Config.MYSQL_DB}")
+    
+    # Create table
     cursor.execute('''
         CREATE TABLE IF NOT EXISTS chat_messages (
             id INT AUTO_INCREMENT PRIMARY KEY,
@@ -28,5 +29,3 @@ def create_tables():
     conn.commit()
     cursor.close()
     conn.close()
-
-