import gradio as gr
from transformers import pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
def summarize_text(text, max_length, min_length):
    summary = summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)
    return summary[0]['summary_text']
demo = gr.Interface(
    fn=summarize_text,
    inputs=[
        gr.Textbox(
            label="요약할 텍스트를 입력하세요",
            placeholder="긴 텍스트를 입력하세요...",
            lines=10
        ),
        gr.Slider(
            minimum=50,
            maximum=500,
            value=130,
            step=10,
            label="최대 길이"
        ),
        gr.Slider(
            minimum=10,
            maximum=200,
            value=30,
            step=10,
            label="최소 길이"
        )
    ],
    outputs=gr.Textbox(label="요약 결과"),
    title="텍스트 요약기",
    theme="soft",
    examples=[
        ["""인공지능(AI)은 인간의 학습능력, 추론능력, 지각능력, 언어이해능력 등을 컴퓨터 프로그램으로 실현한 기술이다. 
        인공지능은 컴퓨터가 인간의 지능적인 행동을 모방할 수 있도록 하는 것을 목표로 한다. 
        이는 학습, 문제해결, 패턴인식, 음성인식, 시각적 지각, 의사결정 등을 포함한다. 
        최근에는 딥러닝과 기계학습의 발전으로 인공지능의 성능이 크게 향상되었으며, 
        다양한 분야에서 활용되고 있다.""", 100, 30],
    ]
)

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