import gradio as gr
from transformers import pipeline, set_seed

generator = pipeline("text-generation", model="gpt2")
set_seed(42)

def ask_librarian(user_input):
    prompt = f"사용자: {user_input}\n사서:"
    result = generator(prompt, max_new_tokens=60, do_sample=True, temperature=0.8)[0]["generated_text"]
    response = result.split("사서:")[-1].strip()
    return response

demo = gr.Interface(
    fn=ask_librarian,
    inputs=gr.Textbox(lines=3, placeholder="책 추천이나 궁금한 것을 입력해 보세요."),
    outputs="text",
    title="📚 AI 도서관 사서",
    description="책 추천, 도서 관련 질문, 간단한 상식에 답변하는 AI 사서 챗봇입니다."
)

if __name__ == "__main__":
    demo.launch()