Update src/db_connector.py

src/db_connector.py

@@ -9,16 +9,17 @@ class Database:
         load_dotenv()
         self.db_uri = os.getenv("DB_URI")
         
-        # ✅ FIX: Use /tmp/chatbot.db which is ALWAYS writable
+        # ✅ CRITICAL FIX: Use /tmp which is always writable
         self.local_db_path = "/tmp/chatbot.db"
         
+        # Always default to internal DB if URI is missing or fails
         if not self.db_uri:
-            print(f"⚠️ DB_URI not found. Using internal safe storage: {self.local_db_path}")
+            print(f"⚠️ DB_URI not found. Using safe internal storage: {self.local_db_path}")
             self.db_uri = f"sqlite:///{self.local_db_path}"
             self.type = "sqlite"
             self.db_path = self.local_db_path
             
-            # Auto-build the user's data
+            # Auto-build the user's data every time
             self._init_sqlite_data()
         else:
             parsed = urlparse(self.db_uri)
@@ -48,15 +49,15 @@ class Database:
 
     def _init_sqlite_data(self):
         """Creates the User's specific schema inside the Safe Temp folder."""
-        # We don't check os.path.exists here because /tmp gets cleared on restart.
-        # We ALWAYS rebuild it to ensure your data is there.
-        
+        # We REBUILD it on every startup to ensure the schema matches your code.
         print("⚡ Building Database in /tmp/chatbot.db ...")
+        
+        conn = None
         try:
             conn = sqlite3.connect(self.local_db_path)
             cursor = conn.cursor()
             
-            # 1. Clean old data to prevent conflicts
+            # 1. Clean old data to avoid conflicts
             cursor.execute("DROP TABLE IF EXISTS sales")
             cursor.execute("DROP TABLE IF EXISTS employees")
 
@@ -123,7 +124,6 @@ class Database:
                 cursor.execute("SHOW TABLES")
                 tables = [list(row.values())[0] for row in cursor.fetchall()]
             
-            # Print to logs so we can verify
             print(f"🔍 TABLES IN DB: {tables}")
             return tables
         except Exception as e:
@@ -141,7 +141,6 @@ class Database:
                 cursor.execute(f"PRAGMA table_info({table_name})")
                 rows = cursor.fetchall()
                 for row in rows:
-                    # SQLite PRAGMA returns: (cid, name, type, notnull, dflt_value, pk)
                     col_name = row['name'] if isinstance(row, sqlite3.Row) else row[1]
                     col_type = row['type'] if isinstance(row, sqlite3.Row) else row[2]
                     columns.append(f"{col_name} ({col_type})")