from fastapi import FastAPI, Request
from transformers import pipeline

print("FastAPI 시작됨!")
app = FastAPI()

print("모델 로딩 시작")
generator = pipeline("text-generation", model="gpt2")
print("모델 로딩 완료!")

@app.get("/")
def root():
    return {"message": "FastAPI 정상 작동 중!"}

@app.post("/generate")
async def generate_text(request: Request):
    body = await request.json()
    prompt = body.get("prompt", "")
    print(f"프롬프트 입력: {prompt}")
    if not prompt:
        return {"error": "Missing 'prompt' field"}
    result = generator(
    prompt,
    max_length=100,
    do_sample=True,         
    temperature=0.8,        
    top_k=50               
)
    return {"generated_text": result[0]["generated_text"]}