Create app.py

app.py

@@ -0,0 +1,31 @@
+from fastapi import FastAPI, Header, HTTPException
+import requests
+import os
+
+app = FastAPI()
+AUTH_TOKEN = os.getenv("API_TOKEN")
+
+def load_dataset():
+    url = "https://huggingface.co/datasets/OjciecTadeusz/test/raw/main/data.json"
+    return requests.get(url).json()
+
+async def auth(authorization: str | None = Header(default=None)):
+    if not authorization or authorization.replace("Bearer ", "") != AUTH_TOKEN:
+        raise HTTPException(status_code=401, detail="Unauthorized")
+
+@app.get("/rows")
+async def get_rows(limit: int = 100, authorization: str | None = Header(default=None)):
+    await auth(authorization)
+    if not (1 <= limit <= 100):
+        raise HTTPException(status_code=400, detail="Invalid limit")
+
+    return load_dataset()[:limit]
+
+@app.get("/row")
+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:
+            return r
+    raise HTTPException(status_code=404, detail="Not found")