Update src/chats.py

src/chats.py

@@ -2,32 +2,53 @@ import sqlite3
 import datetime
 import os
 
-# Use /data for persistent storage in Hugging Face Spaces
-DB_FILE = "/data/chats.db"  # Changed from os.getcwd()
+# Use /data for persistent storage
+DB_FILE = "/data/chats.db"
 
 def init_db():
-    # Ensure the /data directory exists (optional, as /data is typically available)
-    os.makedirs(os.path.dirname(DB_FILE), exist_ok=True)
-    with sqlite3.connect(DB_FILE) as conn:
-        c = conn.cursor()
-        c.execute('''
-            CREATE TABLE IF NOT EXISTS chats (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                title TEXT,
-                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-            )
-        ''')
-        c.execute('''
-            CREATE TABLE IF NOT EXISTS messages (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                chat_id INTEGER,
-                role TEXT,
-                content TEXT,
-                timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                FOREIGN KEY(chat_id) REFERENCES chats(id)
-            )
-        ''')
-        conn.commit()
+    # Debug: Print the database path and check directory permissions
+    print(f"Database path: {DB_FILE}")
+    db_dir = os.path.dirname(DB_FILE)
+    print(f"Database directory: {db_dir}")
+    if not os.path.exists(db_dir):
+        print(f"Directory {db_dir} does not exist, attempting to create it")
+        try:
+            os.makedirs(db_dir, exist_ok=True)
+        except Exception as e:
+            print(f"Failed to create directory {db_dir}: {str(e)}")
+            raise RuntimeError(f"Cannot create directory {db_dir}: {str(e)}")
+    if not os.access(db_dir, os.W_OK):
+        print(f"Directory {db_dir} is not writable")
+        raise RuntimeError(f"Directory {db_dir} is not writable")
+
+    # Attempt to connect to the database
+    print(f"Attempting to connect to database at {DB_FILE}")
+    try:
+        with sqlite3.connect(DB_FILE) as conn:
+            print("Successfully connected to database")
+            c = conn.cursor()
+            c.execute('''
+                CREATE TABLE IF NOT EXISTS chats (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    title TEXT,
+                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+                )
+            ''')
+            c.execute('''
+                CREATE TABLE IF NOT EXISTS messages (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    chat_id INTEGER,
+                    role TEXT,
+                    content TEXT,
+                    timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                    FOREIGN KEY(chat_id) REFERENCES chats(id)
+                )
+            ''')
+            conn.commit()
+            print("Database tables created successfully")
+    except sqlite3.OperationalError as e:
+        print(f"SQLite error: {str(e)}")
+        raise
 
 # Rest of the code remains unchanged
 def create_new_chat():