Spaces:
Sleeping
Sleeping
| from streamlit_mic_recorder import mic_recorder | |
| import streamlit as st | |
| import io | |
| from openai import OpenAI | |
| import os | |
| def whisper_stt(openai_api_key, start_prompt="Start recording", stop_prompt="Stop recording", just_once=False, | |
| use_container_width=False, language=None, callback=None, args=(), kwargs=None, key=None): | |
| if not '_last_speech_to_text_transcript_id' in st.session_state: | |
| st.session_state._last_speech_to_text_transcript_id = 0 | |
| if not '_last_speech_to_text_transcript' in st.session_state: | |
| st.session_state._last_speech_to_text_transcript = None | |
| if key and not key + '_output' in st.session_state: | |
| st.session_state[key + '_output'] = None | |
| audio = mic_recorder(start_prompt=start_prompt, stop_prompt=stop_prompt, just_once=just_once, | |
| use_container_width=use_container_width,format="webm", key=key) | |
| new_output = False | |
| if audio is None: | |
| output = None | |
| else: | |
| if openai_api_key: | |
| if not 'openai_client' in st.session_state: | |
| #assert openai_api_key, openai_api_key | |
| st.session_state.openai_client = OpenAI(api_key=openai_api_key) | |
| id = audio['id'] | |
| new_output = (id > st.session_state._last_speech_to_text_transcript_id) | |
| if new_output: | |
| output = None | |
| st.session_state._last_speech_to_text_transcript_id = id | |
| audio_bio = io.BytesIO(audio['bytes']) | |
| audio_bio.name = 'audio.webm' | |
| success = False | |
| err = 0 | |
| while not success and err < 3: # Retry up to 3 times in case of OpenAI server error. | |
| try: | |
| transcript = st.session_state.openai_client.audio.transcriptions.create( | |
| model="whisper-1", | |
| file=audio_bio, | |
| language=language | |
| ) | |
| except Exception as e: | |
| print(str(e)) # log the exception in the terminal | |
| err += 1 | |
| else: | |
| success = True | |
| output = transcript.text | |
| st.session_state._last_speech_to_text_transcript = output | |
| elif not just_once: | |
| output = st.session_state._last_speech_to_text_transcript | |
| else: | |
| output = None | |
| else: | |
| output = None | |
| if key: | |
| st.session_state[key + '_output'] = output | |
| if new_output and callback: | |
| callback(*args, **(kwargs or {})) | |
| return output |