Spaces:
Running
Running
| import gradio as gr | |
| import os | |
| import assemblyai as aai | |
| import io | |
| from scipy.io.wavfile import write | |
| title = """<h1 align="center">🔥AssemblyAI: Conformer-2 Demo🔥</h1>""" | |
| subtitle = ( | |
| """<p align="center">Automatic Speech Recognition using the AssemblyAI API</p>""" | |
| ) | |
| link = """<p align="center"><a href="https://www.assemblyai.com/blog/conformer-2/">Click here to learn more about the Conformer-2 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 | |
| if not api_key: | |
| return "Error! Did you use a valid API key?" | |
| aai.settings.api_key = api_key | |
| transcriber = aai.Transcriber() | |
| # Create temporary "file" and write data to it | |
| sr, aud = audio_data | |
| bytes_wav = bytes() | |
| temp_file = io.BytesIO(bytes_wav) | |
| write(temp_file, sr, aud) | |
| # Workaround to upload a file-like object before transcribing | |
| # This should be abstracted away in future SDK versions: | |
| try: | |
| upload_url = aai.api.upload_file(aai.Client.get_default().http_client, temp_file) | |
| except aai.types.TranscriptError as e: | |
| return str(e) | |
| # Now we can transcibe the url | |
| transcript = transcriber.transcribe(upload_url) | |
| if transcript.error is not None: | |
| return transcript.error | |
| paragraphs = transcript.get_paragraphs() | |
| return "\n\n".join(p.text for p in paragraphs) | |
| 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. <a href="https://www.assemblyai.com/dashboard/signup">Get a free key here.</a></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_sample1.flac"), | |
| os.path.join( | |
| os.path.dirname(__file__), "audio/assemblyai_company.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.queue(max_size=20, concurrency_count=10).launch(debug=True) | |