Update database.py

database.py

@@ -1,39 +1,41 @@
 import sqlite3
-import os
 from langchain_community.utilities.sql_database import SQLDatabase
+import os
 
-APP_DIR = "/app"
-DB_FILENAME = os.path.join(APP_DIR, "tasks.db")
+DB_FILENAME = "tasks.db"
 DB_PATH_URI = f"sqlite:///{DB_FILENAME}"
 
-
 def create_db_and_table():
-    """Creates the SQLite database and tasks table if they don't exist."""
-
-    conn = sqlite3.connect(DB_FILENAME) # Uses the absolute path
-    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:
+    if not os.path.exists(DB_FILENAME):
+        conn = sqlite3.connect(DB_FILENAME)
+        cursor = conn.cursor()
         cursor.execute("""
-        CREATE UNIQUE INDEX IF NOT EXISTS idx_unq_user_task
-        ON tasks (user_email, task_name, due_date, COALESCE(due_time, ''));
+        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
+        )
         """)
-    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()
+        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.")
+
+
 
 def get_db_info():
     db = SQLDatabase.from_uri(DB_PATH_URI, include_tables=['tasks'])