Create app.py

app.py

@@ -0,0 +1,92 @@
+import os, json, httpx, gradio as gr, uvicorn
+from fastapi import FastAPI, Request, Header, HTTPException
+from fastapi.responses import JSONResponse
+
+# ---------------------------------------------------------------------
+# Configuration
+# ---------------------------------------------------------------------
+BYTEZ_CHAT_URL   = "https://api.bytez.com/models/v2/openai/v1/chat/completions"
+BYTEZ_MODELS_URL = "https://api.bytez.com/models/v2/list/models"
+BYTEZ_AUTH  = os.getenv("BYTEZ_API_KEY")
+LOCAL_API_KEY = os.getenv("LOCAL_API_KEY")
+
+# ---------------------------------------------------------------------
+# FastAPI backend
+# ---------------------------------------------------------------------
+api = FastAPI(title="Bytez → OpenAI Proxy")
+
+def check_key(auth: str | None):
+    if not auth or not auth.startswith("Bearer "):
+        raise HTTPException(status_code=401, detail="Missing or invalid API key")
+    user_key = auth.split("Bearer ")[1].strip()
+    if LOCAL_API_KEY and user_key != LOCAL_API_KEY:
+        raise HTTPException(status_code=403, detail="Unauthorized API key")
+
+# ---------------------------------------------------------------------
+# Root / health
+# ---------------------------------------------------------------------
+@api.get("/")
+def root():
+    return {"status": "ok", "message": "Bytez proxy running"}
+
+# ---------------------------------------------------------------------
+# /v1/models  →  must look OpenAI-style
+# ---------------------------------------------------------------------
+@api.get("/v1/models")
+async def models(authorization: str = Header(None)):
+    check_key(authorization)
+    if not BYTEZ_AUTH:
+        raise HTTPException(status_code=500, detail="Server BYTEZ_API_KEY not configured")
+
+    async with httpx.AsyncClient(timeout=30) as c:
+        r = await c.get(BYTEZ_MODELS_URL, headers={"Authorization": BYTEZ_AUTH})
+    try:
+        data = r.json()
+    except json.JSONDecodeError:
+        raise HTTPException(status_code=502, detail="Upstream returned invalid JSON")
+
+    # Transform Bytez format → OpenAI format
+    models_list = [
+        {"id": m.get("id") or m.get("name"), "object": "model"}
+        for m in (data if isinstance(data, list) else data.get("data", []))
+    ]
+    return JSONResponse(
+        {"object": "list", "data": models_list},
+        headers={"Access-Control-Allow-Origin": "*"}
+    )
+
+# ---------------------------------------------------------------------
+# /v1/chat/completions
+# ---------------------------------------------------------------------
+@api.post("/v1/chat/completions")
+async def chat(request: Request, authorization: str = Header(None)):
+    check_key(authorization)
+    if not BYTEZ_AUTH:
+        raise HTTPException(status_code=500, detail="Server BYTEZ_API_KEY not configured")
+
+    payload = await request.json()
+    headers = {
+        "Authorization": BYTEZ_AUTH,
+        "Content-Type": "application/json",
+    }
+    async with httpx.AsyncClient(timeout=120) as c:
+        r = await c.post(BYTEZ_CHAT_URL, headers=headers, json=payload)
+    try:
+        data = r.json()
+    except json.JSONDecodeError:
+        raise HTTPException(status_code=502, detail="Upstream returned invalid JSON")
+
+    return JSONResponse(data, headers={"Access-Control-Allow-Origin": "*"})
+
+# ---------------------------------------------------------------------
+# Minimal Gradio UI (to make HF Space start)
+# ---------------------------------------------------------------------
+with gr.Blocks() as ui:
+    gr.Markdown("### ✅ Jwero Bytez → OpenAI Proxy\n"
+                "Endpoints: `/v1/models`, `/v1/chat/completions`")
+
+demo = gr.mount_gradio_app(api, ui, path="/")
+
+# local only
+if __name__ == "__main__":
+    uvicorn.run(demo, host="0.0.0.0", port=7860)