import gradio as gr
from summarizer import TransformerSummarizer

title = "Summarizer"
description = """
This demo is GPT-2 based Summarizer, 
works with English, Ukrainian, and Russian (and a few other languages too, it`s GPT-2 after all).
"""


def start_fn(article_input: str) -> str:
    """
    GPT-2 based solution, input full text, output summarized text
    :param article_input:
    :return summarized article_output:
    """
    GPT2_model = TransformerSummarizer(transformer_type="GPT2", transformer_model_key="gpt2-medium")
    full = ''.join(GPT2_model(article_input, min_length=60))
    return full


face = gr.Interface(fn=start_fn,
                    inputs=gr.inputs.Textbox(lines=2, placeholder="Paste article here.", label='Input Article'),
                    outputs=gr.inputs.Textbox(lines=2, placeholder="Summarized article here.", label='Summarized '
                                                                                                     'Article'),
                    title=title,
                    description=description,)
#face.launch(server_name="0.0.0.0", share=True)

#launching the app
if __name__ == "__main__":
    face.launch(debug=True)