backend with database

backend/db.py

@@ -9,7 +9,7 @@ import logging
 logger = logging.getLogger(__name__)
 
 # Database file path
-DB_PATH = os.getenv("DATABASE_PATH", "./ai_tutor.db")
+DB_PATH = os.getenv("DATABASE_PATH", "/app/ai_tutor.db")
 
 
 class Database:

backend/db_cache.py

@@ -6,7 +6,7 @@ import logging
 import hashlib
 
 logger = logging.getLogger(__name__)
-DB_PATH = os.getenv("DATABASE_PATH", "./ai_tutor.db")
+DB_PATH = os.getenv("DATABASE_PATH", "/app/ai_tutor.db")
 
 class ApiCache:
     """Generic caching service using a dedicated database table."""

backend/db_init.py

@@ -7,6 +7,8 @@ import os
 import aiosqlite
 import logging
 from pathlib import Path
+import tempfile
+import atexit
 from typing import Dict, Any, List
 
 logger = logging.getLogger(__name__)
@@ -15,8 +17,32 @@ class DatabaseInitializer:
     """Handles database initialization and health checks"""
     
     def __init__(self, db_path: str = None):
-        self.db_path = db_path or os.getenv("DATABASE_PATH", "./ai_tutor.db")
+        # Check if we should use a temporary database
+        use_temp_db = os.getenv("USE_TEMP_DATABASE", "false").lower() == "true"
+
+        if db_path:
+            self.db_path = db_path
+        elif use_temp_db:
+            # Create a temporary database file that gets cleaned up on exit
+            temp_fd, temp_path = tempfile.mkstemp(suffix='.db', prefix='ai_tutor_')
+            os.close(temp_fd)  # Close the file descriptor, we'll reopen it with sqlite
+            self.db_path = temp_path
+            # Register cleanup on exit
+            atexit.register(self._cleanup_temp_db)
+            logger.info(f"Using temporary database: {self.db_path}")
+        else:
+            self.db_path = os.getenv("DATABASE_PATH", "/app/ai_tutor.db")
+
         self.schema_path = self._find_schema_file()
+
+    def _cleanup_temp_db(self):
+        """Clean up temporary database file"""
+        try:
+            if os.path.exists(self.db_path):
+                os.unlink(self.db_path)
+                logger.info(f"Cleaned up temporary database: {self.db_path}")
+        except Exception as e:
+            logger.error(f"Error cleaning up temporary database: {e}")
     
     def _find_schema_file(self) -> str:
         """Return the path to the schema.sql file.