import gradio as gr
from helper_functions import ai_assistant

title = 'Ai Assistant 🤖'
description = 'A cascade approach consisting of a text transcription model combined with an llm and a synthesizer to create an ai assistant'

demo = gr.Interface(
    fn=ai_assistant,
    inputs=[gr.Audio(label='Command Input', sources=['microphone', 'upload'], type='filepath'), gr.Textbox(label='Groq API Key')],
    outputs=[gr.Audio(label='Output', type='numpy'), gr.Textbox(label="Reponse")],
    flagging_mode='never',
    title=title,
    description=description
)

demo.launch()