| from fastapi import FastAPI, Request, HTTPException
|
| from pydantic import BaseModel
|
| import os
|
| import requests
|
|
|
| app = FastAPI(title="Hakayti Proxy API")
|
|
|
| UPSTREAM_URL = os.getenv("UPSTREAM_URL", "https://placeholder.invalid")
|
| API_KEY = os.getenv("API_KEY", "my_secret_key_123")
|
|
|
| class StoryRequest(BaseModel):
|
| name: str
|
| hobby: str
|
|
|
| @app.post("/generate")
|
| async def generate(req: StoryRequest, request: Request):
|
| header_key = request.headers.get("x-api-key") or request.headers.get("authorization")
|
| if API_KEY and header_key not in (API_KEY, f"Bearer {API_KEY}"):
|
| raise HTTPException(status_code=401, detail="Unauthorized")
|
|
|
| try:
|
| r = requests.post(
|
| f"{UPSTREAM_URL}/generate",
|
| headers={"x-api-key": API_KEY},
|
| json=req.dict(),
|
| timeout=120
|
| )
|
| r.raise_for_status()
|
| except Exception as e:
|
| raise HTTPException(status_code=502, detail=f"Kaggle upstream error: {e}")
|
|
|
| return r.json()
|
|
|
