from transformers import pipeline
import gradio as gr

from helper import identify_locations

title = "Countries/Cities Identification Demo"
description = "This is a simple demo just for demonstration purposes for the Serco team, to validate the first results of the Natural Language module concerning the identification of countries and cities"

examples = [
    ["I want an earthquake that is located in Rome, Italy in 01/01/23 with magnitude greater than 6.2"],
    ["I want an earthquake that is located in Buenos Aires, Argentina in 01/01/23 with magnitude greater than 6.2"],
    ["I want an earthquake that is located in United Arab Emirates in 01/01/23 with magnitude greater than 6.2"],
    ["I want an earthquake that is located in Ishkashim, Tajikistan, in 01/01/23 with magnitude greater than 6.2"],
    ["I want an earthquake that is located in Burkina Faso in 01/01/23 with magnitude greater than 6.2."],
    ["I want an earthquake that is located in Dibba Al-Fujairah, United Arab Emirates in 01/01/23 with magnitude greater than 6.2"],
]

gr.Interface(
    fn=identify_locations, 
    inputs="text", 
    outputs="text",
    title=title,
    description=description,
    examples=examples,
    enable_queue=True,
    ).launch()