from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline

app = FastAPI()

generator = pipeline("text2text-generation", model="memorease/memorease-quizgen")

class Memory(BaseModel):
    description: str

@app.post("/generate")
def generate(memory: Memory):
    prompt = f"Soru üret: {memory.description}"
    result = generator(prompt, max_length=64, do_sample=False)
    return {"question": result[0]["generated_text"]}