main

app.py

@@ -19,8 +19,35 @@ from underthesea import sent_tokenize
 from unidecode import unidecode
 from vinorm import TTSnorm
 
-from TTS.tts.configs.xtts_config import XttsConfig
-from TTS.tts.models.xtts import Xtts
+print("Setting up the environment...")
+import subprocess
+
+setup_command = [
+    "pip", "install", "--use-deprecated=legacy-resolver", "-e", ".", "-q",
+    "&&", "cd", "TTS ",
+    "&&", "git submodule update --init --recursive",
+    "&&", "echo", "Installing other requirements...",
+    "&&", "pip", "install", "-r", "requirements.txt", "-q",
+    "&&", "echo", "Downloading Japanese/Chinese tokenizer...",
+    "&&", "python", "-m", "unidic", "download",
+    "&&", "pip", "install", "--upgrade", "gradio"
+]
+
+try:
+    subprocess.run(setup_command, check=True, shell=True)
+    print("Environment setup completed successfully.")
+except subprocess.CalledProcessError as e:
+    print(f"Error during environment setup: {e}")
+    print("Please run the setup command manually if needed.")
+
+print("\nTo set up the environment manually, run the following command:")
+print("pip install --use-deprecated=legacy-resolver -e . -q && \\ ")
+print("cd .. && \\ ")
+print("echo \"Installing other requirements...\" && \\ ")
+print("pip install -r requirements.txt -q && \\ ")
+print("echo \"Downloading Japanese/Chinese tokenizer...\" && \\ ")
+print("python -m unidic download && \\ ")
+print("pip install --upgrade gradio")
 
 XTTS_MODEL = None
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -56,7 +83,9 @@ def load_model(checkpoint_dir="model/", repo_id="capleaf/viXTTS", use_deepspeed=
         )
         yield f"Model download finished..."
 
-    xtts_config = os.path.join(checkpoint_dir, "config.json")
+    xtts_config = os.path.join(checkpoint_dir, "config.json")\
+    from TTS.tts.configs.xtts_config import XttsConfig
+    from TTS.tts.models.xtts import Xtts
     config = XttsConfig()
     config.load_json(xtts_config)
     XTTS_MODEL = Xtts.init_from_config(config)
@@ -279,14 +308,6 @@ def read_logs():
         return f.read()
 
 
-def run_docker_setup():
-    """Run the Docker setup commands."""
-    subprocess.run("python -m venv .env && . .env/bin/activate && git submodule update --init --recursive && cd TTS && git fetch --tags && git checkout 0.1.1 && echo 'Installing TTS...' && pip install --use-deprecated=legacy-resolver -e . -q && cd .. && echo 'Installing other requirements...' && pip install -r requirements.txt -q && echo 'Downloading Japanese/Chinese tokenizer...' && python -m unidic download && pip install --upgrade gradio && touch .env/ok", shell=True)
-
-# Call the function at the appropriate place in your code
-run_docker_setup()
-
-
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         description="""viXTTS inference demo\n\n""",
@@ -417,3 +438,5 @@ if __name__ == "__main__":
         )
 
     demo.launch()
+
+