Update app.py

app.py

@@ -3,18 +3,27 @@ import requests
 import os
 
 app = FastAPI()
-HF_TOKEN = os.getenv("HF_TOKEN")
+
+AUTH_TOKEN = os.getenv("AUTH_TOKEN")  # poprawiona zmienna
+URL = os.getenv("URL")  # URL do datasetu JSON
 
 def load_dataset():
-    url = os.getenv("URL")
-    r = requests.get(url)
+    if not URL:
+        raise HTTPException(status_code=500, detail="Dataset URL not set")
+    r = requests.get(URL)
     if r.status_code != 200:
         raise HTTPException(status_code=500, detail="Dataset fetch failed")
     return r.json()
 
 async def auth(authorization: str | None = Header(default=None)):
-    if AUTH_TOKEN and (not authorization or authorization.replace("Bearer ", "") != AUTH_TOKEN):
-        raise HTTPException(status_code=401, detail="Unauthorized")
+    if AUTH_TOKEN:
+        token = authorization.replace("Bearer ", "") if authorization else None
+        if token != AUTH_TOKEN:
+            raise HTTPException(status_code=401, detail="Unauthorized")
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}
 
 @app.get("/rows")
 async def get_rows(limit: int = 3, authorization: str | None = Header(default=None)):
@@ -27,9 +36,6 @@ async def get_row(id: str, authorization: str | None = Header(default=None)):
     await auth(authorization)
     data = load_dataset()
     for r in data:
-        if r["id"] == id:
+        if r.get("id") == id:
             return r
     raise HTTPException(status_code=404, detail="Not found")
-
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)