|
|
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()
|
|
|
|
