Update src/database.py

src/database.py

@@ -10,35 +10,35 @@ class DatabaseManager:
         self.db_path = db_path
         self._init_db()
 
-def _init_db(self):
-        """Creates tables if they don't exist."""
-        conn = sqlite3.connect(self.db_path)
-        c = conn.cursor()
-        
-        # UPDATED: Added 'abstract' column
-        c.execute('''
-            CREATE TABLE IF NOT EXISTS documents (
-                doc_id TEXT PRIMARY KEY,
-                filename TEXT,
-                upload_date TEXT,
-                user_id TEXT, 
-                full_text TEXT,
-                abstract TEXT 
-            )
-        ''')
-        
-        # (Chunks table remains the same...)
-        c.execute('''
-            CREATE TABLE IF NOT EXISTS chunks (
-                chunk_id TEXT PRIMARY KEY,
-                doc_id TEXT,
-                text TEXT,
-                chunk_index INTEGER,
-                FOREIGN KEY(doc_id) REFERENCES documents(doc_id)
-            )
-        ''')
-        conn.commit()
-        conn.close()
+    def _init_db(self):
+            """Creates tables if they don't exist."""
+            conn = sqlite3.connect(self.db_path)
+            c = conn.cursor()
+            
+            # UPDATED: Added 'abstract' column
+            c.execute('''
+                CREATE TABLE IF NOT EXISTS documents (
+                    doc_id TEXT PRIMARY KEY,
+                    filename TEXT,
+                    upload_date TEXT,
+                    user_id TEXT, 
+                    full_text TEXT,
+                    abstract TEXT 
+                )
+            ''')
+            
+            # (Chunks table remains the same...)
+            c.execute('''
+                CREATE TABLE IF NOT EXISTS chunks (
+                    chunk_id TEXT PRIMARY KEY,
+                    doc_id TEXT,
+                    text TEXT,
+                    chunk_index INTEGER,
+                    FOREIGN KEY(doc_id) REFERENCES documents(doc_id)
+                )
+            ''')
+            conn.commit()
+            conn.close()
 
     # UPDATED: Added abstract argument
     def add_document(self, doc_id, filename, full_text, abstract="", user_id="default"):