Update main.py

main.py

@@ -5,7 +5,6 @@ import uuid
 import re
 import time
 import json
-import base64
 import traceback
 import inflect
 from datetime import datetime, timezone
@@ -49,45 +48,52 @@ IDX_FACES      = "enterprise-faces"
 IDX_OBJECTS    = "enterprise-objects"
 
 # ════════════════════════════════════════════════════════════════
-#  GRAFANA LOKI  —  async, fire-and-forget, never crashes the API
+#  SUPABASE LOGGING  —  async, fire-and-forget, never crashes API
 #  HF Space Secrets needed:
-#    LOKI_URL       → https://logs-prod-006.grafana.net   (no trailing slash)
-#    LOKI_USERNAME  → your Grafana Cloud numeric user ID
-#    LOKI_PASSWORD  → your Grafana Cloud API token (Logs:Write scope)
+#    SUPABASE_URL          → https://xxxx.supabase.co
+#    SUPABASE_SERVICE_KEY  → your Supabase service_role key (not anon!)
 # ════════════════════════════════════════════════════════════════
-LOKI_URL      = os.getenv("LOKI_URL",      "")
-LOKI_USERNAME = os.getenv("LOKI_USERNAME", "")
-LOKI_PASSWORD = os.getenv("LOKI_PASSWORD", "")
+SUPABASE_URL         = os.getenv("SUPABASE_URL",         "")
+SUPABASE_SERVICE_KEY = os.getenv("SUPABASE_SERVICE_KEY", "")
 
-async def _loki_push(level: str, event: str, data: dict):
-    """Fire-and-forget push to Grafana Loki. Silent on failure."""
-    if not (LOKI_URL and LOKI_USERNAME and LOKI_PASSWORD):
+async def _supabase_log_push(level: str, event: str, data: dict):
+    """Fire-and-forget insert into Supabase app_logs table."""
+    if not (SUPABASE_URL and SUPABASE_SERVICE_KEY):
         return
     try:
         import aiohttp
-        ts_ns    = str(int(time.time() * 1e9))
-        line     = json.dumps({"timestamp": datetime.now(timezone.utc).isoformat(),
-                               "level": level.upper(), "service": "enterprise-lens",
-                               "event": event, **data}, default=str)
-        payload  = {"streams": [{"stream": {"service": "enterprise-lens",
-                                            "level": level.lower(),
-                                            "event": event,
-                                            "env": os.getenv("ENVIRONMENT", "production")},
-                                 "values": [[ts_ns, line]]}]}
-        creds    = base64.b64encode(f"{LOKI_USERNAME}:{LOKI_PASSWORD}".encode()).decode()
-        headers  = {"Content-Type": "application/json", "Authorization": f"Basic {creds}"}
+        row = {
+            "level":       level.upper(),
+            "event":       event,
+            "user_id":     str(data.get("user_id", "anonymous")),
+            "ip":          str(data.get("ip", "")),
+            "mode":        str(data.get("mode", "")),
+            "page":        str(data.get("page", "")),
+            "duration_ms": int(data["duration_ms"]) if "duration_ms" in data else None,
+            "error":       str(data["error"]) if "error" in data else None,
+            "data":        data,
+        }
+        headers = {
+            "Content-Type":  "application/json",
+            "apikey":        SUPABASE_SERVICE_KEY,
+            "Authorization": f"Bearer {SUPABASE_SERVICE_KEY}",
+            "Prefer":        "return=minimal",
+        }
         async with aiohttp.ClientSession() as s:
-            async with s.post(f"{LOKI_URL}/loki/api/v1/push",
-                              json=payload, headers=headers,
-                              timeout=aiohttp.ClientTimeout(total=5)) as r:
-                if r.status not in (200, 204):
-                    _log_fn("WARNING", f"Loki returned {r.status}")
+            async with s.post(
+                f"{SUPABASE_URL}/rest/v1/app_logs",
+                json=row, headers=headers,
+                timeout=aiohttp.ClientTimeout(total=5)
+            ) as r:
+                if r.status not in (200, 201):
+                    body = await r.text()
+                    _log_fn("WARNING", f"Supabase log insert failed {r.status}: {body[:200]}")
     except Exception as exc:
-        _log_fn("DEBUG", f"Loki push skipped: {exc}")
+        _log_fn("DEBUG", f"Supabase log push skipped: {exc}")
 
 def log(level: str, event: str, **data):
     """
-    Log to console + Grafana Loki (background task).
+    Log to console + Supabase app_logs table (background task).
     Usage: log("INFO", "upload.complete", user_id="x", files=3, duration_ms=340)
     """
     clean = {k: v for k, v in data.items()}
@@ -95,7 +101,7 @@ def log(level: str, event: str, **data):
     try:
         loop = asyncio.get_event_loop()
         if loop.is_running():
-            asyncio.create_task(_loki_push(level, event, data))
+            asyncio.create_task(_supabase_log_push(level, event, data))
     except Exception:
         pass