from transformers import pipeline
import gradio as gr

get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")

def summarize(input):
    output = get_completion(input)
    return output[0]['summary_text']

# Test the function:
print(summarize("The tower is 324 metres (1,063 ft) tall, about the same height as an 81-storey building, and the tallest structure in Paris. Its base is square, measuring 125 metres (410 ft) on each side. During its construction, the Eiffel Tower surpassed the Washington Monument to become the tallest man-made structure in the world, a title it held for 41 years until the Chrysler Building in New York City was finished in 1930. It was the first structure to reach a height of 300 metres. Due to the addition of a broadcasting aerial at the top of the tower in 1957, it is now taller than the Chrysler Building by 5.2 metres (17 ft). Excluding transmitters, the Eiffel Tower is the second tallest free-standing structure in France after the Millau Viaduct."))

demo = gr.Interface(
    fn=summarize, 
    inputs=gr.components.Textbox(lines=5, label="Insérer votre texte ici"),
    outputs=gr.components.Textbox(label="Voilà votre résumé"),
    title="Outil pour résumer vos textes en français",
    description="Réduisez la longueur de vos textes et faites-en ressortir le sens en quelques secondes. S’il y a une erreur dans le résultat, soit d’orthographe, de syntaxe, de grammaire ou de compréhension du texte, simplement cliquer sur le bouton Signaler une erreur. Nul besoin de spécifier l’erreur, le modèle va retravailler par lui-même.",
    allow_flagging="manual",
    examples=["En Gaspésie : un festin de petits fruits bios à cueillir! La Gaspésie regroupe plusieurs productions fruitières biologiques et les visiteurs peuvent donc s’adonner à l’autocueillette dans des décors enchanteurs. Parmi les entreprises qui offrent des activités d’autocueillette, on retrouve la ferme Bio Culture Lepage de Sainte-Anne-des-Monts, spécialisée dans la culture de petits fruits bios depuis plus de 35 ans. Les propriétaires de la ferme proposent aux visiteurs de cueillir des fraises, des framboises, des bleuets et des cassis, tout en profitant d’une vue somptueuse sur le fleuve. De plus, ces petits fruits bios frais, congelés ou transformés sont en vente au kiosque à la ferme."]
)
demo.launch()