import pickle
import gradio as gr
from classifier import Classifier

def classify(txt):
    with open('classifier.pickle', 'rb') as f:
        classifier = pickle.load(f)
    return classifier.predict(txt)


title = 'Detector de Quechua y Español'
description =(  'Bolivia lucha para que no desaparezcan los idiomas indígenas, sin embargo,' + 
                'es aún muy complicado acceder a recursos que ayuden a su asimilación y aprendizaje.' +
                'Presentamos una herramienta de clasificación de idiomas, que si bien es una tarea sencilla, ' +
                'resulta esencial para realizar tareas más complejas como la traducción automática.'
            )
article = 'Demo del proyecto para Saturdays.\nAutores del modelo: Cota V. Andreina, Cusi L. Evelyn, Nina M. Juan Dilan'

iface = gr.Interface(
    fn=classify,
    inputs= gr.inputs.Textbox(lines=3, label='TEXTO', placeholder='Introduzca un texto'),
    outputs= gr.outputs.Textbox(label='IDIOMA'),
    examples = ['¿Maytaq ashkallanchikega?', 'Entonces el Inka dijo ¡Mach\'a!', '¡Aragan kanki wamraqa!', 'Señora, ¿yanapariwayta atiwaqchu?', '¿A dónde vas?'],
    description = description,
    title = title,
    article = article,
    theme = 'peach'
    )

iface.launch()