Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| import os | |
| from helpers import make_header, upload_file, request_transcript, wait_for_completion, make_paragraphs_string | |
| title = """<h1 align="center">🔥AssemblyAI: Conformer-1 Demo🔥</h1>""" | |
| subtitle = """<h2 align="center">Automatic Speech Recognition using the AssemblyAI API</h2>""" | |
| link = """<p align="center"><a href="https://www.assemblyai.com/blog/conformer-1/">Click here to learn more about the Conformer-1 model</a></p>""" | |
| def submit_to_AAI(api_key, | |
| radio, | |
| audio_file, | |
| mic_recording): | |
| if radio == "Audio File": | |
| audio_data = audio_file | |
| elif radio == "Record Audio": | |
| audio_data = mic_recording | |
| header = make_header(api_key) | |
| # 1. Upload the audio | |
| upload_url = upload_file(audio_data, header, is_file=False) | |
| # 2. Request transcript | |
| transcript_response = request_transcript(upload_url, header) | |
| transcript_id = transcript_response['id'] | |
| # 3. Wait for the transcription to complete | |
| _, error = wait_for_completion(transcript_id, header) | |
| if error is not None: | |
| return error | |
| # 4. Fetch paragraphs of transcript | |
| return make_paragraphs_string(transcript_id, header) | |
| def change_audio_source(radio): | |
| if radio == "Audio File": | |
| return [gr.Audio.update(visible=True), | |
| gr.Audio.update(visible=False)] | |
| elif radio == "Record Audio": | |
| return [gr.Audio.update(visible=False), | |
| gr.Audio.update(visible=True)] | |
| with gr.Blocks(css = """#col_container {width: 1000px; margin-left: auto; margin-right: auto;} | |
| #chatbot {height: 520px; overflow: auto;}""") as demo: | |
| gr.HTML('<center><a href="https://www.assemblyai.com/"><img src="file/images/logo.png" width="180px"></a></center>') | |
| gr.HTML(title) | |
| gr.HTML(subtitle) | |
| gr.HTML(link) | |
| gr.HTML('''<center><a href="https://huggingface.co/spaces/assemblyai/Conformer1-Demo?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>Duplicate the Space and run securely with your AssemblyAI API Key</center>''') | |
| with gr.Column(elem_id="col_container"): | |
| api_key = gr.Textbox(type='password', label="Enter your AssemblyAI API key here") | |
| with gr.Box(): | |
| # Selector for audio source | |
| radio = gr.Radio(["Audio File", "Record Audio"], label="Audio Source", value="Audio File") | |
| # Audio object for both file and microphone data | |
| audio_file = gr.Audio() | |
| mic_recording = gr.Audio(source="microphone", visible=False) | |
| gr.Examples([os.path.join(os.path.dirname(__file__),"audio/audio.mp3")], audio_file) | |
| btn = gr.Button("Run") | |
| out = gr.Textbox(placeholder="Your formatted transcript will appear here ...", lines=10) | |
| # Changing audio source changes Audio input component | |
| radio.change(fn=change_audio_source, | |
| inputs=[radio], | |
| outputs=[audio_file, mic_recording]) | |
| # Clicking "submit" uploads selected audio to AssemblyAI, performs requested analyses, and displays results | |
| btn.click(fn=submit_to_AAI, | |
| inputs=[api_key,radio,audio_file,mic_recording], | |
| outputs=out) | |
| demo.launch(debug=True) |