import gradio as gr

from krwordrank.word import summarize_with_keywords

# import pip
# pip.main(['install', 'krwordrank'])


# main
def inference(prompt):

    prompt = prompt.split('\n')

    keywords = summarize_with_keywords(
        prompt, 
        min_count=5, 
        max_length=10,
        beta=0.85, 
        max_iter=10, 
        verbose=True
    )
    #keywords = summarize_with_keywords(prompt)

    return keywords


demo = gr.Interface(
    fn=inference, 
    inputs="text", 
    outputs="text" #return 값
    ).launch() # launch(share=True)를 설정하면 외부에서 접속 가능한 링크가 생성됨

demo.launch()