# app.py
from fastapi import FastAPI, Request
from transformers import pipeline

app = FastAPI()

# Modelo leve e gratuito que funciona em português
generator = pipeline("text-generation", model="pierreguillou/tiny-random-gpt2-portuguese", max_length=200)

@app.post("/gerar")
async def gerar_historia(req: Request):
    body = await req.json()
    relato = body.get("relato", "")
    estilo = body.get("estilo", "")

    prompt = f"Escreva uma história no estilo {estilo} baseada neste relato: {relato}"

    try:
        resultado = generator(prompt)[0]['generated_text']
        return {"historia": resultado}
    except Exception as e:
        return {"erro": str(e)}