import requests
import gradio as gr
import os
import urllib.request

from gradio_client import Client

hf_link = os.environ['link']
client = Client(hf_link)
local_link = os.environ['local_link']
local_api = os.environ['local_api']

local_code = requests.head(local_link).status_code
print(local_code)

def s2t_hf(source, mic, fi, lang):
    if source == 'file':
        in_file = fi
    else:
        in_file = mic

    result = client.predict(
				in_file,	# str (filepath or URL to file) in 'inputs' Audio component
				"transcribe",	# str in 'Task' Radio component
				True,	# bool in 'Return timestamps' Checkbox component
				api_name="/predict"
            )
    print(result)
    return result

def s2t_local(source, mic, fi, lang):
    if source == 'file':
        in_file = fi
    else:
        in_file = mic

    form_data = {"language": lang}
    files = {'input_file': open(in_file, 'rb')}
    print(files)
    
    response = requests.post(local_api, files=files, data=form_data, timeout=10000)

    return response.json()

def s2t(source, mic, fi, lang):
    if local_code==200:
        return s2t_local(source, mic, fi, lang)
    else:
        return s2t_hf(source, mic, fi, lang)

def update_audio_ui(audio_source: str, input_audio_mic, input_audio_file):
    mic = audio_source == "microphone"
    # input_audio_mic.visible = mic
    # print(type(gr.update(visible=mic, value=None)))
    return gr.update(visible=mic, value=None), gr.update(visible=not mic, value=None)

with gr.Blocks() as demo:
    gr.Markdown('<h1 style="text-align: center;">Speech to Text</h1>')
    with gr.Group():
        with gr.Row() as audio_box:
            with gr.Column():
                input_lang = gr.Dropdown(['auto','vi', 'ja', 'en', 'cn', 'ko'], label='Language?', value='en', interactive=True)
                audio_source = gr.Radio(
                    label="Audio source",
                    choices=["file", "microphone"],
                    value="file",
                    interactive=True
                )
            input_audio_mic = gr.Audio(
                label="Input speech",
                type="filepath",
                sources="microphone",
                visible=False,
            )
            input_audio_file = gr.Audio(
                label="Input speech",
                type="filepath",
                sources="upload",
                visible=True,
            )
        js = gr.JSON(label="json")
        with gr.Row():
            btn = gr.Button("Run")
            btn_clean = gr.ClearButton([input_audio_mic, input_audio_file, js])

    audio_source.change(
        fn=update_audio_ui,
        inputs=[audio_source, input_audio_mic, input_audio_file],
        outputs=[
            input_audio_mic,
            input_audio_file,
        ],
        queue=False,
        api_name=False,
    )


    btn.click(fn=s2t, inputs=[audio_source, input_audio_mic, input_audio_file, input_lang], outputs=[js])

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