from fastapi import FastAPI
from transformers import pipeline
import uvicorn

app = FastAPI()

print("Caricamento modello amogaddy/GenerAI...")
generator = pipeline("text-generation", model="amogaddy/GenerAI")

@app.get("/")
def health_check():
    return {"status": "running", "model": "GenerAI"}

@app.post("/ask")
def ask(prompt: str):
    output = generator(prompt, max_length=100, num_return_sequences=1)
    return {"result": output[0]['generated_text']}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=7860)