from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
import requests
import os 

app = FastAPI()
app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])

# 🔱 Secrets වලින් Key එක ලබාගැනීම (Invisible to View Source)
API_KEY = os.getenv("OPENROUTER_KEY")
MODEL = "tencent/hy3-preview:free"

@app.post("/generate")
async def generate(request: Request):
    try:
        data = await request.json()
        headers = {
            "Authorization": f"Bearer {API_KEY}",
            "HTTP-Referer": "https://hf.space",
            "Content-Type": "application/json"
        }
        payload = {
            "model": MODEL,
            "messages": [{"role": "user", "content": data.get("prompt")}]
        }
        response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload)
        return response.json()
    except Exception as e:
        return {"error": str(e)}