import gradio as gr
from inference import predict

title = "Demo para la clasificación de comandos de texto para asistente virtual"
description = "Este demo clasifica comandos en 11 clases: \n'messaging', 'calling', 'event', 'timer', 'music',
              'weather', 'alarm', 'people', 'reminder', 'recipes', 'news'"

demo = gr.Interface(fn=predict, 
                    inputs='text', 
                    outputs='text',
                    title=title,
                    description=description,
                    examples=["Recuérdame el cumpleaños de Juan para el 25 de Agosto", "Pon la alarma a las 6 am", 
                              "¿Algo divertido este lunes?", "¿Cuál es la receta del pastel de zanahoria?", 
                              "Envía el mensaje de WhatsApp", "¿Quién trabaja en el Starbucks local?",
                              "Haz una videollamada a Karina y a Simon.", "Noticias principales en la ciudad", 
                              "Reproduce mi última canción", "Dime la temperatura actual"]
                   )

demo.launch()