Spaces:
Runtime error
Runtime error
| import re | |
| from pathlib import Path | |
| import gradio as gr | |
| from elevenlabslib import ElevenLabsUser | |
| from elevenlabslib.helpers import save_bytes_to_path | |
| import modules.shared as shared | |
| params = { | |
| 'activate': True, | |
| 'api_key': '12345', | |
| 'selected_voice': 'None', | |
| } | |
| initial_voice = ['None'] | |
| wav_idx = 0 | |
| user = ElevenLabsUser(params['api_key']) | |
| user_info = None | |
| if not shared.args.no_stream: | |
| print("Please add --no-stream. This extension is not meant to be used with streaming.") | |
| raise ValueError | |
| # Check if the API is valid and refresh the UI accordingly. | |
| def check_valid_api(): | |
| global user, user_info, params | |
| user = ElevenLabsUser(params['api_key']) | |
| user_info = user._get_subscription_data() | |
| print('checking api') | |
| if not params['activate']: | |
| return gr.update(value='Disconnected') | |
| elif user_info is None: | |
| print('Incorrect API Key') | |
| return gr.update(value='Disconnected') | |
| else: | |
| print('Got an API Key!') | |
| return gr.update(value='Connected') | |
| # Once the API is verified, get the available voices and update the dropdown list | |
| def refresh_voices(): | |
| global user, user_info | |
| your_voices = [None] | |
| if user_info is not None: | |
| for voice in user.get_available_voices(): | |
| your_voices.append(voice.initialName) | |
| return gr.Dropdown.update(choices=your_voices) | |
| else: | |
| return | |
| def remove_surrounded_chars(string): | |
| # this expression matches to 'as few symbols as possible (0 upwards) between any asterisks' OR | |
| # 'as few symbols as possible (0 upwards) between an asterisk and the end of the string' | |
| return re.sub('\*[^\*]*?(\*|$)', '', string) | |
| def input_modifier(string): | |
| """ | |
| This function is applied to your text inputs before | |
| they are fed into the model. | |
| """ | |
| return string | |
| def output_modifier(string): | |
| """ | |
| This function is applied to the model outputs. | |
| """ | |
| global params, wav_idx, user, user_info | |
| if not params['activate']: | |
| return string | |
| elif user_info is None: | |
| return string | |
| string = remove_surrounded_chars(string) | |
| string = string.replace('"', '') | |
| string = string.replace('“', '') | |
| string = string.replace('\n', ' ') | |
| string = string.strip() | |
| if string == '': | |
| string = 'empty reply, try regenerating' | |
| output_file = Path(f'extensions/elevenlabs_tts/outputs/{wav_idx:06d}.wav'.format(wav_idx)) | |
| voice = user.get_voices_by_name(params['selected_voice'])[0] | |
| audio_data = voice.generate_audio_bytes(string) | |
| save_bytes_to_path(Path(f'extensions/elevenlabs_tts/outputs/{wav_idx:06d}.wav'), audio_data) | |
| string = f'<audio src="file/{output_file.as_posix()}" controls></audio>' | |
| wav_idx += 1 | |
| return string | |
| def ui(): | |
| # Gradio elements | |
| with gr.Row(): | |
| activate = gr.Checkbox(value=params['activate'], label='Activate TTS') | |
| connection_status = gr.Textbox(value='Disconnected', label='Connection Status') | |
| voice = gr.Dropdown(value=params['selected_voice'], choices=initial_voice, label='TTS Voice') | |
| with gr.Row(): | |
| api_key = gr.Textbox(placeholder="Enter your API key.", label='API Key') | |
| connect = gr.Button(value='Connect') | |
| # Event functions to update the parameters in the backend | |
| activate.change(lambda x: params.update({'activate': x}), activate, None) | |
| voice.change(lambda x: params.update({'selected_voice': x}), voice, None) | |
| api_key.change(lambda x: params.update({'api_key': x}), api_key, None) | |
| connect.click(check_valid_api, [], connection_status) | |
| connect.click(refresh_voices, [], voice) | |