import gradio as gr
from inference import query
from script import *


def interface(Language, Audio):

    if Language == 'Hausa':
        command =  query(Audio, 'ha')
        state = activate_hausa(command)
        return state
   
    # elif Language == 'English':
    #      command =  query(Audio, lang ='en')
    #      state = activate_english(command)
    #      return state
    
    elif Language == 'Yoruba':
        command =  query(Audio, 'yo')
        state = activate_yoruba(command)
        return state

    else:
        pass


demo = gr.Interface(
    fn=interface,
    inputs=[gr.Dropdown(['Hausa', 'English', 'Yoruba'], value = 'Hausa', label='Select Your Prefered Language'), gr.Audio(source ='microphone', type='numpy')],
    outputs="text",
    live=True
)


if __name__ == '__main__':
    demo.launch()