Update main.py

main.py

@@ -1,59 +1,53 @@
-# main.py — ASGI entrypoint for BubbleGuard Web (HF-ready, with GDrive fetch)
-# Coder: Amir
-# ------------------------------------------------
-# - Downloads model bundles from Google Drive (via download_assets.py)
-# - Starts Uvicorn serving app_server:app
-# - Respects HOST/PORT env (PORT is injected by Hugging Face Spaces)
-
-import os
-import sys
-import traceback
-
-def _truthy(val: str) -> bool:
-    return str(val).strip().lower() in {"1", "true", "yes", "on"}
-
-def main() -> None:
-    # --- Env & logging setup ---
-    host = os.getenv("HOST", "0.0.0.0")
-    port = int(os.getenv("PORT", "8000"))
-    debug = _truthy(os.getenv("DEBUG", "0"))
-
-    print(f"[BubbleGuard] Starting server on {host}:{port}")
-    folder_id = os.getenv("GDRIVE_FOLDER_ID", "").strip()
-    if folder_id:
-        print(f"[BubbleGuard] Using Google Drive folder: {folder_id}")
-    else:
-        print("[BubbleGuard] No GDRIVE_FOLDER_ID set (will try per-file IDs if provided).")
-
-    # --- Ensure assets (idempotent) before importing the app ---
-    try:
-        from download_assets import ensure_all_assets
-        ensure_all_assets()
-    except Exception as e:
-        print("[BubbleGuard] ERROR while preparing assets:", file=sys.stderr)
-        traceback.print_exc()
-        raise SystemExit(1)
-
-    # --- Run Uvicorn ---
-    try:
-        import uvicorn
-        # Use string import so app is imported after assets are ready
-        uvicorn.run(
-            "app_server:app",
-            host=host,
-            port=port,
-            reload=False,                 # safer with large models
-            workers=1,                    # avoid duplicating model memory
-            log_level="debug" if debug else "info",
-            proxy_headers=True,           # play nice behind proxies (HF Spaces)
-            forwarded_allow_ips="*",
-            timeout_keep_alive=10,
-            access_log=True,
-        )
-    except Exception:
-        print("[BubbleGuard] Uvicorn failed to start:", file=sys.stderr)
-        traceback.print_exc()
-        raise SystemExit(1)
-
-if __name__ == "__main__":
-    main()
+# main.py — ASGI entrypoint for BubbleGuard Web (HF-ready, with GDrive fetch)
+# Author: Amir
+
+import os
+import sys
+import traceback
+
+def _truthy(val: str) -> bool:
+    return str(val).strip().lower() in {"1", "true", "yes", "on"}
+
+def main() -> None:
+    host = os.getenv("HOST", "0.0.0.0")
+    port = int(os.getenv("PORT", "8000"))
+    debug = _truthy(os.getenv("DEBUG", "0"))
+
+    print(f"[BubbleGuard] Starting server on {host}:{port}")
+    folder_id = os.getenv("GDRIVE_FOLDER_ID", "").strip()
+    if folder_id:
+        print(f"[BubbleGuard] Using Google Drive folder: {folder_id}")
+    else:
+        print("[BubbleGuard] No GDRIVE_FOLDER_ID set (will fail to load assets).")
+
+    # Ensure assets before importing the app
+    try:
+        from download_assets import ensure_all_assets
+        ensure_all_assets()
+    except Exception as e:
+        print("[BubbleGuard] ERROR while preparing assets:", file=sys.stderr)
+        traceback.print_exc()
+        raise SystemExit(1)
+
+    # Start Uvicorn after assets are ready
+    try:
+        import uvicorn
+        uvicorn.run(
+            "app_server:app",
+            host=host,
+            port=port,
+            reload=False,
+            workers=1,
+            log_level="debug" if debug else "info",
+            proxy_headers=True,
+            forwarded_allow_ips="*",
+            timeout_keep_alive=10,
+            access_log=True,
+        )
+    except Exception:
+        print("[BubbleGuard] Uvicorn failed to start:", file=sys.stderr)
+        traceback.print_exc()
+        raise SystemExit(1)
+
+if __name__ == "__main__":
+    main()