Spaces:
Running
Running
| from TTS.api import TTS | |
| from bs4 import BeautifulSoup | |
| import requests | |
| import streamlit as st | |
| import tempfile | |
| import os | |
| import json | |
| with open('config.json', 'r') as f: | |
| config = json.load(f) | |
| APP_NAME = config['APP_NAME'] | |
| APP_LOGO = config['APP_LOGO'] | |
| APP_DESCRIPTION = config['APP_DESCRIPTION'] | |
| LANGUAGES_URL = config['LANGUAGES_URL'] | |
| OUTPUT_FILENAME = config['OUTPUT_FILENAME'] | |
| def contains_only_ascii(input_string): | |
| return all(ord(char) < 128 for char in input_string) | |
| def get_iso_languages(): | |
| response = requests.get(LANGUAGES_URL) | |
| soup = BeautifulSoup(response.text, 'html.parser') | |
| p_tags = soup.find_all('p') | |
| iso_language_dict = {} | |
| for p_tag in p_tags[1:]: # Skipping the first <p> which contains the header | |
| parts = p_tag.get_text().split() | |
| if len(parts) == 2: | |
| iso_code, language_name = parts | |
| if contains_only_ascii(language_name): | |
| iso_language_dict[language_name] = iso_code | |
| return iso_language_dict | |
| def create_temp_file(input_wav): | |
| temp_file = tempfile.NamedTemporaryFile(delete=False) | |
| temp_file.write(input_wav.read()) | |
| return temp_file | |
| def remove_temp_file(temp_file): | |
| temp_file.close() | |
| os.remove(temp_file.name) | |
| def update_progress(percent, text): | |
| progress_bar.progress(percent) | |
| status_text.text(text) | |
| iso_languages = get_iso_languages() | |
| languages = list(iso_languages.keys()) | |
| st.set_page_config(page_title=APP_NAME) | |
| st.title(APP_NAME) | |
| st.image(APP_LOGO, use_column_width=True) | |
| st.markdown(APP_DESCRIPTION) | |
| language = st.selectbox('Select a language', languages) | |
| prompt = st.text_input('Enter your prompt') | |
| input_wav = st.file_uploader("Upload a WAV file", type=["wav"]) | |
| if input_wav: | |
| if not input_wav or input_wav is None: | |
| st.error('Please upload wav input audio') | |
| elif not prompt: | |
| st.error('Please write prompt') | |
| else: | |
| progress_bar = st.progress(0) | |
| status_text = st.empty() | |
| temp_file = create_temp_file(input_wav) | |
| iso_code = iso_languages[language] | |
| print(f'Language: {language}, prompt: {prompt}') | |
| update_progress(0, 'Loading TTS model...') | |
| api = TTS(f"tts_models/{iso_code}/fairseq/vits") | |
| update_progress(50, 'Generating audio...') | |
| api.tts_with_vc_to_file( | |
| prompt, | |
| speaker_wav=temp_file.name, | |
| file_path=OUTPUT_FILENAME | |
| ) | |
| remove_temp_file(temp_file) | |
| audio_file = open(OUTPUT_FILENAME, 'rb') | |
| audio_bytes = audio_file.read() | |
| update_progress(100, 'Audio generated successfully!') | |
| st.audio(audio_bytes, format='audio/wav') | |
| st.download_button('Download WAV', data=audio_bytes, file_name='file.wav') |