import gradio as gr
from transformers import pipeline

pipe = pipeline("summarization", model="cloudqi/cqi_brain_memory_summarizer_large_pt_v0")

def main(in_text):
  print(in_text)
  answer = pipe(in_text, min_length=5, max_length=20)
  print(answer)
  return answer[0]["summary_text"]

with gr.Blocks() as demo:
  gr.Markdown("""## Sumarizer Open Version""")
  with gr.Row():
    with gr.Column():
      text1 = gr.Textbox(label="""Diálogo""",
            lines=1,value="""Bruno: Ei, você tem o número da Betty?
Kleber: Desculpe, não encontrei.
Bruno: Pesquise.
Kleber: De acordo com minha pesquisa o número é 35 9 9898-6142"""
        )
      output = gr.Textbox(label="Resultado")
      b1 = gr.Button("Summarize!")
      b1.click(main, inputs=[text1], outputs=output)
  gr.Markdown("""###### Model cqi_brain_memory_summarizer_large_pt_v0 designed by [cloudqi]""")


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