Update database.py

database.py

@@ -1,41 +1,34 @@
 import sqlite3
 from langchain_community.utilities.sql_database import SQLDatabase
-import os
 
 DB_FILENAME = "tasks.db"
 DB_PATH_URI = f"sqlite:///{DB_FILENAME}"
 
 def create_db_and_table():
-    if not os.path.exists(DB_FILENAME):
-        conn = sqlite3.connect(DB_FILENAME)
-        cursor = conn.cursor()
+    conn = sqlite3.connect(DB_FILENAME)
+    cursor = conn.cursor()
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS tasks (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        user_name TEXT,
+        user_email TEXT,
+        task_name TEXT NOT NULL,
+        status TEXT DEFAULT 'pending',
+        category TEXT,
+        created_at TEXT DEFAULT (strftime('%Y-%m-%d %H:%M:%S', 'now', 'localtime')),
+        due_date TEXT,
+        due_time TEXT
+    )
+    """)
+    try:
         cursor.execute("""
-        CREATE TABLE IF NOT EXISTS tasks (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            user_name TEXT,
-            user_email TEXT,
-            task_name TEXT NOT NULL,
-            status TEXT DEFAULT 'pending',
-            category TEXT,
-            created_at TEXT DEFAULT (strftime('%Y-%m-%d %H:%M:%S', 'now', 'localtime')),
-            due_date TEXT,
-            due_time TEXT
-        )
+        CREATE UNIQUE INDEX IF NOT EXISTS idx_unq_user_task
+        ON tasks (user_email, task_name, due_date, COALESCE(due_time, ''));
         """)
-        try:
-            cursor.execute("""
-            CREATE UNIQUE INDEX IF NOT EXISTS idx_unq_user_task
-            ON tasks (user_email, task_name, due_date, COALESCE(due_time, ''));
-            """)
-        except sqlite3.OperationalError as e:
-            print(f"Warning: Could not create unique index idx_unq_user_task, it might exist or conflict: {e}")
-        conn.commit()
-        conn.close()
-        print("Database and table created.")
-    else:
-        print("Database already exists, connecting to the existing database.")
-
-
+    except sqlite3.OperationalError as e:
+        print(f"Warning: Could not create unique index idx_unq_user_task, it might exist or conflict: {e}")
+    conn.commit()
+    conn.close()
 
 def get_db_info():
     db = SQLDatabase.from_uri(DB_PATH_URI, include_tables=['tasks'])
@@ -99,4 +92,4 @@ def execute_dml_query(query: str, params: tuple = None):
         print(f"General SQLite error executing DML query: {query}\n{e}")
         raise
     finally:
-        conn.close()
+        conn.close()