Update database.py

database.py

@@ -1,12 +1,23 @@
 import sqlite3
+import os
 from langchain_community.utilities.sql_database import SQLDatabase
 
-DB_FILENAME = "tasks.db"
+PERSISTENT_STORAGE_PATH = os.environ.get("HF_PERSISTENT_STORAGE_PATH", "/data")
+DB_FILENAME = os.path.join(PERSISTENT_STORAGE_PATH, "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."""
+
+        os.makedirs(os.path.dirname(DB_FILENAME), exist_ok=True)
+        print(f"Database directory {os.path.dirname(DB_FILENAME)} ensured.")
+    except OSError as e:
+        print(f"Error creating database directory {os.path.dirname(DB_FILENAME)}: {e}")
+
     conn = sqlite3.connect(DB_FILENAME)
     cursor = conn.cursor()
+
     cursor.execute("""
     CREATE TABLE IF NOT EXISTS tasks (
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -29,14 +40,18 @@ def create_db_and_table():
         print(f"Warning: Could not create unique index idx_unq_user_task, it might exist or conflict: {e}")
     conn.commit()
     conn.close()
+    print(f"✅ Database and table checked/created at {DB_FILENAME}.")
+
 
+# ... (rest of your database.py, e.g., get_db_info, execute_select_query, etc.) ...
+# All functions like `sqlite3.connect(DB_FILENAME)` will now use the path in persistent storage.
 def get_db_info():
     db = SQLDatabase.from_uri(DB_PATH_URI, include_tables=['tasks'])
     return db.get_table_info()
 
 def get_task_by_id(task_id: int, user_email: str):
     query = "SELECT id, task_name, status, category, due_date, due_time, created_at FROM tasks WHERE id = ? AND user_email = ?"
-    conn = sqlite3.connect(DB_FILENAME)
+    conn = sqlite3.connect(DB_FILENAME) # Uses persistent path
     cursor = conn.cursor()
     try:
         cursor.execute(query, (task_id, user_email))
@@ -52,7 +67,7 @@ def get_task_by_id(task_id: int, user_email: str):
         conn.close()
 
 def execute_select_query(query: str, params: tuple = None):
-    conn = sqlite3.connect(DB_FILENAME)
+    conn = sqlite3.connect(DB_FILENAME) # Uses persistent path
     cursor = conn.cursor()
     try:
         cursor.execute(query, params or ())
@@ -66,7 +81,7 @@ def execute_select_query(query: str, params: tuple = None):
         conn.close()
 
 def execute_dml_query(query: str, params: tuple = None):
-    conn = sqlite3.connect(DB_FILENAME)
+    conn = sqlite3.connect(DB_FILENAME) # Uses persistent path
     cursor = conn.cursor()
     is_insert = query.strip().upper().startswith("INSERT")
     try:
@@ -75,21 +90,19 @@ def execute_dml_query(query: str, params: tuple = None):
         if is_insert:
             return cursor.lastrowid
         return cursor.rowcount
-    
-    except sqlite3.IntegrityError as e: 
-        error_code = getattr(e, 'sqlite_errorcode', None) 
-        if not error_code and hasattr(e, 'args') and len(e.args) > 0: 
-            if "UNIQUE constraint failed" in str(e.args[0]).upper(): 
-                 error_code = 2067 
+    except sqlite3.IntegrityError as e:
+        error_code = getattr(e, 'sqlite_errorcode', None)
+        if not error_code and hasattr(e, 'args') and len(e.args) > 0:
+            if "UNIQUE constraint failed" in str(e.args[0]).upper():
+                 error_code = 2067
         if error_code == 2067 or error_code == 1555 or \
            (isinstance(e, sqlite3.IntegrityError) and "UNIQUE constraint failed" in str(e).upper()):
-            raise ValueError(f"Task likely already exists with the same name, due date, and time. (Details: {e})")
+            raise ValueError(f"Task likely already exists. (Details: {e})")
         else:
-            print(f"Unhandled IntegrityError executing DML query: {query}\nCode: {error_code}, Error: {e}")
-            raise 
-
+            print(f"Unhandled IntegrityError: {query}\nCode: {error_code}, Error: {e}")
+            raise
     except sqlite3.Error as e:
-        print(f"General SQLite error executing DML query: {query}\n{e}")
+        print(f"General SQLite error: {query}\n{e}")
         raise
     finally:
-        conn.close()
+        conn.close()