bug

local_database.py

@@ -8,11 +8,13 @@ INDEX_FILE = "auramind_faiss.index"
 def get_db_connection():
     """Establishes a connection to the database."""
     conn = None
+    print(f"Attempting to connect to database: {DB_FILE}")
     try:
         conn = sqlite3.connect(DB_FILE)
         conn.row_factory = sqlite3.Row
+        print(f"Successfully connected to database: {DB_FILE}")
     except Error as e:
-        print(e)
+        print(f"Error connecting to database: {e}")
     return conn
 
 def init_db():
@@ -20,9 +22,12 @@ def init_db():
     Initializes a more robust database schema for multimodal data.
     - 'documents' table tracks the source files.
     - 'chunks' table stores the individual encrypted text/image chunks.
+    - 'farm_analysis' table stores analysis data.
     """
+    print("Initializing database...")
     conn = get_db_connection()
     if conn is None:
+        print("Database connection failed, cannot initialize.")
         return
     cursor = conn.cursor()
 
@@ -32,7 +37,9 @@ def init_db():
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             name TEXT NOT NULL UNIQUE
         )
-    ''')
+    ''
+    )
+    print("Table 'documents' checked/created.")
 
     # Table to store each chunk of content (text or image)
     # The faiss_id will correspond to the row number in the FAISS index
@@ -45,7 +52,9 @@ def init_db():
             page_num INTEGER,
             FOREIGN KEY (doc_id) REFERENCES documents (id)
         )
-    ''')
+    ''
+    )
+    print("Table 'chunks' checked/created.")
 
     # Table for farm analysis data
     sql_create_table = """ CREATE TABLE IF NOT EXISTS farm_analysis (
@@ -64,9 +73,12 @@ def init_db():
                                     outcome_image_id TEXT
                                 ); """
     cursor.execute(sql_create_table)
+    print("Table 'farm_analysis' checked/created.")
 
     conn.commit()
+    print("Database changes committed.")
     conn.close()
+    print("Database connection closed.")
 
 def check_if_indexed():
     """Checks if the initial database and index file exist."""