Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import time
-from fastapi import FastAPI, Request, HTTPException, status
+from fastapi import FastAPI, Request, HTTPException, status, Header
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import Response, JSONResponse, StreamingResponse, RedirectResponse
 import httpx
@@ -13,6 +13,8 @@ from urllib.parse import quote
 import uuid
 import base64
 from subscriptions import fetch_subscription
+from typing import Optional
+
 WAN_SPACE = "https://huggingface.co/spaces/Wan-AI/Wan2.1"
 WAN_API_BASE = f"{WAN_SPACE}/gradio_api"
 
@@ -919,16 +921,15 @@ async def genvideo_airforce(request: Request, prompt: str = None):
         },
     )
 
-@app.get("/subscription/{email}")
-async def get_subscription_by_email(email: str):
-    result = await fetch_subscription(email)
-    return result or {"email": email, "subscription": None}
+@app.get("/subscription")
+async def get_subscription(authorization: Optional[str] = Header(None)):
+    if not authorization or not authorization.startswith("Bearer "):
+        raise HTTPException(401, "Missing or invalid Authorization header")
+
+    jwt = authorization.split(" ", 1)[1]
 
-@app.post("/subscription")
-async def get_subscription_post(body: dict):
-    email = body.get("email")
-    if not email:
-        raise HTTPException(400, "email is required")
+    result = await fetch_subscription(jwt)
+    if "error" in result:
+        raise HTTPException(401, result["error"])
 
-    result = await fetch_subscription(email)
-    return result or {"email": email, "subscription": None}
+    return result