| import gradio as gr | |
| import numpy as np | |
| import runpod | |
| import os | |
| RUNPOD_API = os.environ.get("RUNPOD_API") | |
| RUNPOD_ENDPOINT = os.environ.get("RUNPOD_ENDPOINT") | |
| auth = [(os.environ.get("ADMIN"), os.environ.get("ADMIN_PW")), (os.environ.get("USER"), os.environ.get("USER_PW"))] | |
| def welcome_user(request: gr.Request): | |
| return f"Bienvenue {request.username}" | |
| def doDebug(request: gr.Request): | |
| if request.username == os.environ.get("ADMIN"): | |
| return gr.Textbox(label="Debug Transcription", visible=True) | |
| return gr.Textbox(label="Debug Transcription", visible=False) | |
| def process_transcribe(file): | |
| audio_nparray = file[1] | |
| audio_list = audio_nparray.tolist() | |
| runpod.api_key = RUNPOD_API | |
| endpoint = runpod.Endpoint(RUNPOD_ENDPOINT) | |
| run_request = endpoint.run_sync( | |
| {"audio_list": audio_list} | |
| ) | |
| raw_text = run_request | |
| text_corrected = raw_text["text_corrected"] | |
| return raw_text,text_corrected | |
| with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as iface: | |
| m = gr.Markdown(value="bienvenue") | |
| iface.load(welcome_user, None, m) | |
| with gr.Tab("Speech2text prescription"): | |
| with gr.Row(): | |
| with gr.Column(): | |
| audio_file = gr.Audio(sources=["upload"], type="numpy", label="Audio") | |
| submit_btn = gr.Button("Submit", variant="primary") | |
| with gr.Column(): | |
| debug_transcript = gr.Textbox() | |
| iface.load(doDebug, None, debug_transcript) | |
| transcript = gr.Textbox(label="Transcription") | |
| submit_btn.click(process_transcribe, inputs=[audio_file], outputs=[debug_transcript,transcript]) | |
| logout_button = gr.Button("Logout", link="/logout") | |
| iface.launch(auth=auth) |