Update app/main.py

app/main.py

@@ -5,24 +5,19 @@ import hashlib
 app = FastAPI(
     title="Codex ReflexGuard Enterprise API",
     version="1.0.0",
-    description="Enterprise Reflex Intelligence & Control API"
+    description="Enterprise Reflex Intelligence API"
 )
 
-# -------------------------
-# SECURITY
-# -------------------------
+# ---------------- SECURITY ----------------
 VALID_KEYS = {
-    "codex-enterprise-demo-key",
-    "codex-internal-key"
+    "codex-enterprise-demo-key"
 }
 
-def verify_key(x_api_key: str | None):
-    if x_api_key not in VALID_KEYS:
+def verify_key(key: str | None):
+    if key not in VALID_KEYS:
         raise HTTPException(status_code=401, detail="Invalid API Key")
 
-# -------------------------
-# ROOT (THIS FIXES YOUR ISSUE)
-# -------------------------
+# ---------------- ROOT ----------------
 @app.get("/")
 def root():
     return {
@@ -33,19 +28,12 @@ def root():
         "endpoint": "/v1/reflex/check"
     }
 
-# -------------------------
-# HEALTH CHECK
-# -------------------------
+# ---------------- HEALTH ----------------
 @app.get("/health")
 def health():
-    return {
-        "status": "ok",
-        "timestamp": datetime.utcnow().isoformat()
-    }
+    return {"status": "ok", "time": datetime.utcnow().isoformat()}
 
-# -------------------------
-# CORE ENTERPRISE API
-# -------------------------
+# ---------------- CORE API ----------------
 @app.post("/v1/reflex/check")
 def reflex_check(payload: dict, x_api_key: str = Header(None)):
     verify_key(x_api_key)
@@ -56,12 +44,10 @@ def reflex_check(payload: dict, x_api_key: str = Header(None)):
     score = min(len(scenario) / 100.0, 1.0)
     state = "BLOCK" if score > 0.6 else "ALLOW"
 
-    scenario_hash = hashlib.sha256(scenario.encode()).hexdigest()
-
     return {
         "timestamp": datetime.utcnow().isoformat(),
-        "scenario_hash": scenario_hash,
+        "scenario_hash": hashlib.sha256(scenario.encode()).hexdigest(),
         "score": score,
-        "state": state,
+        "decision": state,
         "source": source
     }