import gradio as gr
import discourse as d
import memory as m

# set a custom theme
theme = gr.themes.Default().set(
    body_background_fill="#000000",
)

with gr.Blocks(theme=theme) as ui:
    with gr.Row():
        with gr.Column(scale=1):
            message = gr.Audio(source="microphone", type="filepath")
    with gr.Row():
        btn1 = gr.Button("Respond")
    with gr.Row():
        with gr.Column(scale=1):
            audio_response = gr.Audio()
    with gr.Row():
        text_response = gr.Textbox(label="Transcript", max_lines=10)
    with gr.Row():
        btn2 = gr.Button("Save Conversation")

    btn1.click(fn=d.respond, inputs=[message, m.context], outputs=[audio_response, text_response])
    btn2.click(fn=m.save_as_hf_dataset)

ui.launch()