update

app.py

@@ -35,6 +35,12 @@ with open('ja_sentences.txt') as f:
 ####################################
 # Constants
 ####################################
+
+SPK1 = os.getenv('KOTOBA_SPK1')
+SPK2 = os.getenv('KOTOBA_SPK2')
+SPK3 = os.getenv('KOTOBA_SPK3')
+SPK4 = os.getenv('KOTOBA_SPK4')
+
 AVAILABLE_MODELS = {
     # 'XTTSv2': 'xtts',
     # # 'WhisperSpeech': 'whisperspeech',
@@ -51,10 +57,10 @@ AVAILABLE_MODELS = {
     # 'Parler TTS': 'parler'
     'MOE-VITS': 'moe-vits',
     'BARK': 'bark',
-    'KOTOBA-SPEECH-AVA': 'kotoba-speech-ava',
-    'KOTOBA-SPEECH-BRIA': 'kotoba-speech-bria',
-    'KOTOBA-SPEECH-ALEX': 'kotoba-speech-alex',
-    'KOTOBA-SPEECH-JACOB': 'kotoba-speech-jacob',
+    f'KOTOBA-SPEECH-{SPK1.upper()}': f'kotoba-speech-{SPK1.lower()}',
+    f'KOTOBA-SPEECH-{SPK2.upper()}': f'kotoba-speech-{SPK2.lower()}',
+    f'KOTOBA-SPEECH-{SPK3.upper()}': f'kotoba-speech-{SPK3.lower()}',
+    f'KOTOBA-SPEECH-{SPK4.upper()}': f'kotoba-speech-{SPK4.lower()}',
     'BLANE-TTS': 'blane-tts',
     'AMITARO-VITS': 'amitaro-vits',
     'GOOGLE-TTS': 'google-tts',
@@ -124,56 +130,56 @@ def get_db():
 
 def get_tts_file(text: str, model: str):
     url = {
-        "kotoba-speech-ava": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
-        "kotoba-speech-bria": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
-        "kotoba-speech-alex": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
-        "kotoba-speech-jacob": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
+        f"kotoba-speech-{SPK1.lower()}": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
+        f"kotoba-speech-{SPK2.lower()}": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
+        f"kotoba-speech-{SPK3.lower()}": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
+        f"kotoba-speech-{SPK4.lower()}": "https://kotoba-tech-kotoba-speech.hf.space/call/tts",
         "blane-tts": "https://blane187-blane-tts.hf.space/call/get_audio_file"
     }
     headers = {
         "Content-Type": "application/json"
     }
     data = {
-        "kotoba-speech-ava": {
+        f"kotoba-speech-{SPK1.lower()}": {
             "data": [
                 text,
                 5,
                 5,
                 "Preset voices",
-                "Ava",
+                SPK1,
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"},
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"}
             ]
         },
-        "kotoba-speech-bria": {
+        f"kotoba-speech-{SPK2.lower()}": {
             "data": [
                 text,
                 5,
                 5,
                 "Preset voices",
-                "Bria",
+                SPK2,
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"},
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"}
             ]
         },
-        "kotoba-speech-alex": {
+        f"kotoba-speech-{SPK3.lower()}": {
             "data": [
                 text,
                 5,
                 5,
                 "Preset voices",
-                "Alex",
+                SPK3,
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"},
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"}
             ]
         },
-        "kotoba-speech-jacob": {
+        f"kotoba-speech-{SPK4.lower()}": {
             "data": [
                 text,
                 5,
                 5,
                 "Preset voices",
-                "Jacob",
+                SPK4,
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"},
                 {"path": "https://keikaku-hoso.com/sample_voice/voice01_A.mp3"}
             ]
@@ -392,10 +398,10 @@ model_names = {
     # 'metavoice': 'MetaVoice-1B',
     'BARK': 'BARK',
     'MOE-VITS': 'MOE-VITS',
-    'KOTOBA-SPEECH-AVA': 'KOTOBA-SPEECH-SPK1',
-    'KOTOBA-SPEECH-BRIA': 'KOTOBA-SPEECH-SPK2',
-    'KOTOBA-SPEECH-ALEX': 'KOTOBA-SPEECH-SPK3',
-    'KOTOBA-SPEECH-JACOB': 'KOTOBA-SPEECH-SPK4',
+    f'KOTOBA-SPEECH-{SPK1.upper()}': 'KOTOBA-SPEECH-SPK1',
+    f'KOTOBA-SPEECH-{SPK2.upper()}': 'KOTOBA-SPEECH-SPK2',
+    f'KOTOBA-SPEECH-{SPK3.upper()}': 'KOTOBA-SPEECH-SPK3',
+    f'KOTOBA-SPEECH-{SPK4.upper()}': 'KOTOBA-SPEECH-SPK4',
     'BLANE-TTS': 'BLANE-TTS',
     'AMITARO-VITS': 'AMITARO-VITS',
     'GOOGLE-TTS': 'GOOGLE-TTS',
@@ -450,10 +456,10 @@ model_links = {
     # 'metavoice': 'https://github.com/metavoiceio/metavoice-src',
     'bark': 'https://suno-bark.hf.space/',
     'moe-vits': 'skytnt/moe-tts',
-    'kotoba-speech-ava': 'https://kotoba-tech-kotoba-speech.hf.space/',
-    'kotoba-speech-bria': 'https://kotoba-tech-kotoba-speech.hf.space/',
-    'kotoba-speech-alex': 'https://kotoba-tech-kotoba-speech.hf.space/',
-    'kotoba-speech-jacob': 'https://kotoba-tech-kotoba-speech.hf.space/',
+    f'kotoba-speech-{SPK1.lower()}': 'https://kotoba-tech-kotoba-speech.hf.space/',
+    f'kotoba-speech-{SPK2.lower()}': 'https://kotoba-tech-kotoba-speech.hf.space/',
+    f'kotoba-speech-{SPK3.lower()}': 'https://kotoba-tech-kotoba-speech.hf.space/',
+    f'kotoba-speech-{SPK4.lower()}': 'https://kotoba-tech-kotoba-speech.hf.space/',
     'blane-tts': 'https://blane187-blane-tts.hf.space/',
     'amitaro-vits': 'https://lycoris53-vits-tts-japanese-only-amitaro.hf.space/'
 }