reset

app.py

@@ -4,6 +4,7 @@ FastAPI URL Blink Application
 Production-grade API for receiving encrypted user data,
 decrypting it, and storing in SQLite database.
 """
+import os
 import logging
 from contextlib import asynccontextmanager
 from fastapi import FastAPI, Request
@@ -32,9 +33,16 @@ async def lifespan(app: FastAPI):
     """
     logger.info("Starting up - initializing database...")
     
-    # Startup: Download DB from Drive
-    logger.info("Startup: Attempting to download database from Google Drive...")
-    drive_service.download_db()
+    # Check for RESET_DB environment variable
+    if os.getenv("RESET_DB", "").lower() == "true":
+        logger.warning("RESET_DB is set to true. Skipping download and clearing local database.")
+        if os.path.exists("blink_data.db"):
+            os.remove("blink_data.db")
+            logger.info("Local database deleted.")
+    else:
+        # Startup: Download DB from Drive
+        logger.info("Startup: Attempting to download database from Google Drive...")
+        drive_service.download_db()
     
     await init_db()
     logger.info("Database initialized successfully")