Update StarReliabilityApp_Refined.py

StarReliabilityApp_Refined.py

@@ -33,15 +33,17 @@ class StarMaintAI:
 
         self.load_long_term_memory()
 
-    def ensure_database_exists(self):
+           def ensure_database_exists(self):
         """
         Ensure the database file exists and create required tables if not.
         """
         if not os.path.exists(self.db_path):
             print(f"Database not found at {self.db_path}. Initializing new database.")
-            connection = sqlite3.connect(self.db_path)
-            cursor = connection.cursor()
+        
+        connection = sqlite3.connect(self.db_path)
+        cursor = connection.cursor()
 
+        try:
             # Create necessary tables
             cursor.execute("""
                 CREATE TABLE IF NOT EXISTS long_term_memory (
@@ -63,8 +65,42 @@ class StarMaintAI:
             """)
 
             connection.commit()
+        except sqlite3.Error as e:
+            print(f"Error during database initialization: {e}")
+        finally:
             connection.close()
 
+        # Refresh connection to ensure database is ready
+        self.connection = sqlite3.connect(self.db_path)
+
+    def load_long_term_memory(self):
+        """
+        Load persistent memory from the 'long_term_memory' table.
+        """
+        try:
+            cursor = self.connection.cursor()
+            cursor.execute("SELECT key, value FROM long_term_memory")
+            self.long_term_memory = {row[0]: row[1] for row in cursor.fetchall()}
+        except sqlite3.OperationalError as e:
+            print(f"Error loading long-term memory: {e}. Reinitializing database.")
+            self.ensure_database_exists()
+            self.long_term_memory = {}
+
+
+    def load_long_term_memory(self):
+        """
+        Load persistent memory from the 'long_term_memory' table.
+        """
+        try:
+            cursor = self.connection.cursor()
+            cursor.execute("SELECT key, value FROM long_term_memory")
+            self.long_term_memory = {row[0]: row[1] for row in cursor.fetchall()}
+        except sqlite3.OperationalError as e:
+            print(f"Error loading long-term memory: {e}. Reinitializing database.")
+            self.ensure_database_exists()
+            self.long_term_memory = {}
+
+
     def load_long_term_memory(self):
         """
         Load persistent memory from the 'long_term_memory' table.