Update auth.py

auth.py

@@ -1,44 +1,37 @@
 from fastapi import Header, HTTPException
-from firebase_admin import firestore
 import time
 
-db = firestore.client()
+# Demo API keys (later move to DB)
+API_KEYS = {
+    "free_key_123": {"plan": "free", "limit": 100},
+    "pro_key_456": {"plan": "pro", "limit": 10000},
+}
 
-# in-memory rate tracking (later move to Redis)
-REQUEST_LOG = {}
+request_log = {}
 
 
 def verify_api_key(x_api_key: str = Header(...)):
-    # 1. Check key exists
-    doc = db.collection("api_keys").document(x_api_key).get()
-
-    if not doc.exists:
-        raise HTTPException(status_code=401, detail="Invalid API key")
-
-    data = doc.to_dict()
-
-    # 2. Check active
-    if not data.get("active", False):
-        raise HTTPException(status_code=403, detail="API key disabled")
+    if x_api_key not in API_KEYS:
+        raise HTTPException(
+            status_code=401,
+            detail="Invalid API Key"
+        )
 
-    plan = data.get("plan", "pro")
-    limit = data.get("limit", 50000)
+    key_data = API_KEYS[x_api_key]
 
-    # 3. Rate limit
     now = int(time.time())
-    window = 60  # 1 min
+    window = 60  # 1 minute window
 
-    logs = REQUEST_LOG.get(x_api_key, [])
+    logs = request_log.get(x_api_key, [])
     logs = [t for t in logs if now - t < window]
 
-    if len(logs) >= limit:
+    if len(logs) >= key_data["limit"]:
         raise HTTPException(
             status_code=429,
-            detail=f"Rate limit exceeded ({plan} plan)"
+            detail=f"Rate limit exceeded for {key_data['plan']} plan"
         )
 
     logs.append(now)
-    REQUEST_LOG[x_api_key] = logs
-
-    return data
+    request_log[x_api_key] = logs
 
+    return key_data