initial commit

app.py

@@ -0,0 +1,98 @@
+from TTS.api import TTS
+from bs4 import BeautifulSoup
+import requests
+import streamlit as st
+import tempfile
+import os
+import json
+import datetime
+
+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']
+
+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)
+input_wav = st.file_uploader("Upload a WAV file with your voice", type=["wav"])
+clone_wav = st.file_uploader("Upload a WAV file with voice to clone", type=["wav"])
+
+if input_wav and clone_wav:
+    progress_bar = st.progress(0)
+    status_text = st.empty()
+
+    current_datetime = datetime.datetime.now()
+    formatted_datetime = current_datetime.strftime("%Y-%m-%d_%H%M%S")
+    output_filename = f"recording_{formatted_datetime}.wav"
+
+    temp_input_file = create_temp_file(input_wav)
+    temp_clone_file = create_temp_file(clone_wav)
+
+    iso_code = iso_languages[language]
+
+    print(f'Language: {language}')
+
+    update_progress(0, 'Loading TTS model...')
+    api = TTS(f"tts_models/{iso_code}/fairseq/vits")
+
+    update_progress(50, 'Generating audio...')
+    api.voice_conversion_to_file(
+        source_wav=temp_input_file.name,
+        target_wav=temp_clone_file.name,
+        file_path=output_filename
+    )
+
+    remove_temp_file(temp_input_file)
+    remove_temp_file(temp_clone_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='output.wav')

config.json

@@ -0,0 +1,6 @@
+{
+    "APP_NAME": "TTS-Voice-Cloner",
+    "APP_LOGO": "https://raw.githubusercontent.com/coqui-ai/TTS/main/images/coqui-log-green-TTS.png",
+    "APP_DESCRIPTION": "Simple app that uses [Coqui TTS](https://github.com/coqui-ai/TTS) library to clone your voice!",
+    "LANGUAGES_URL": "https://dl.fbaipublicfiles.com/mms/tts/all-tts-languages.html"
+}

requirements.txt

@@ -0,0 +1,2 @@
+beautifulsoup4
+TTS