app.py

from fastapi import FastAPI, Request
from transformers import pipeline
import uvicorn

app = FastAPI()

generator = pipeline("text2text-generation", model="pierreguillou/t5-base-pt-small")

@app.post("/gerar")
async def gerar_historia(req: Request):
    body = await req.json()
    relato = body.get("relato", "")
    estilo = body.get("estilo", "")
    prompt = f"Crie uma história no estilo {estilo}. Baseie-se neste relato: {relato}"

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