Update StarReliabilityApp_Refined.py

StarReliabilityApp_Refined.py

@@ -7,6 +7,8 @@ import speech_recognition as sr  # For local STT (if desired)
 
 class StarMaintAI:
     def __init__(self, db_path):
+        self.db_path = db_path
+        self.ensure_database_exists()
         self.connection = sqlite3.connect(db_path)
         self.short_term_memory = []
         self.medium_term_memory = []
@@ -31,6 +33,38 @@ class StarMaintAI:
 
         self.load_long_term_memory()
 
+    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()
+
+            # Create necessary tables
+            cursor.execute("""
+                CREATE TABLE IF NOT EXISTS long_term_memory (
+                    key TEXT PRIMARY KEY,
+                    value TEXT
+                )
+            """)
+            cursor.execute("""
+                CREATE TABLE IF NOT EXISTS prompts (
+                    title TEXT PRIMARY KEY,
+                    description TEXT
+                )
+            """)
+            cursor.execute("""
+                CREATE TABLE IF NOT EXISTS functions (
+                    function_name TEXT PRIMARY KEY,
+                    description TEXT
+                )
+            """)
+
+            connection.commit()
+            connection.close()
+
     def load_long_term_memory(self):
         """
         Load persistent memory from the 'long_term_memory' table.