Upload 2 files

app.py

@@ -0,0 +1,64 @@
+# app.py — FastAPI бекенд за Hugging Face Spaces
+import os
+import httpx
+from fastapi import FastAPI, Request, UploadFile, File, Response
+from fastapi.responses import JSONResponse
+from fastapi.staticfiles import StaticFiles
+
+HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACEHUB_API_TOKEN") or os.getenv("HUGGING_FACE_HUB_TOKEN")
+HF_MODEL = os.getenv("HF_MODEL", "nateraw/food101")
+HF_URL = f"https://api-inference.huggingface.co/models/{HF_MODEL}"
+
+app = FastAPI()
+
+@app.get("/health")
+async def health():
+    ok = bool(HF_TOKEN)
+    return {"ok": ok, "model": HF_MODEL, "token_present": ok}
+
+@app.post("/api/classify")
+async def classify_raw(request: Request):
+    body = await request.body()
+    if not body:
+        return JSONResponse({"error": "No image body"}, status_code=400)
+    if not HF_TOKEN:
+        return JSONResponse({"error": "Missing HF_TOKEN in Space secrets"}, status_code=500)
+
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Accept": "application/json",
+        "X-Wait-For-Model": "true",
+    }
+    async with httpx.AsyncClient(timeout=60) as client:
+        r = await client.post(HF_URL, headers=headers, content=body)
+
+    try:
+        data = r.json()
+        return JSONResponse(data, status_code=r.status_code)
+    except Exception:
+        return Response(content=r.text, status_code=r.status_code,
+                        media_type=r.headers.get("content-type", "text/plain"))
+
+@app.post("/api/classify-multipart")
+async def classify_multipart(file: UploadFile = File(...)):
+    if not HF_TOKEN:
+        return JSONResponse({"error": "Missing HF_TOKEN in Space secrets"}, status_code=500)
+
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Accept": "application/json",
+        "X-Wait-For-Model": "true",
+    }
+    files = {"file": (file.filename, await file.read(), file.content_type)}
+    async with httpx.AsyncClient(timeout=60) as client:
+        r = await client.post(HF_URL, headers=headers, files=files)
+
+    try:
+        data = r.json()
+        return JSONResponse(data, status_code=r.status_code)
+    except Exception:
+        return Response(content=r.text, status_code=r.status_code,
+                        media_type=r.headers.get("content-type", "text/plain"))
+
+if os.path.isdir("frontend"):
+    app.mount("/", StaticFiles(directory="frontend", html=True), name="static")

requirements.txt

@@ -0,0 +1,4 @@
+fastapi==0.111.0
+uvicorn==0.30.1
+httpx==0.27.0
+starlette==0.37.2