| import os |
| import sys |
| import requests |
| import tarfile |
| import gdown |
| import shutil |
| import subprocess |
| from pathlib import Path |
| from config import BASE_DIR |
|
|
| def run_command(command): |
| print(f"Running: {command}") |
| try: |
| result = subprocess.run(command, shell=True, check=True, text=True, |
| stdout=subprocess.PIPE, stderr=subprocess.PIPE) |
| print(result.stdout) |
| if result.stderr: |
| print(f"STDERR: {result.stderr}") |
| return True |
| except subprocess.CalledProcessError as e: |
| print(f"Error executing command: {e}") |
| print(f"STDERR: {e.stderr}") |
| return False |
|
|
|
|
| def download_file(url, destination): |
| print(f"Downloading {url} to {destination}") |
| try: |
| response = requests.get(url, stream=True) |
| response.raise_for_status() |
|
|
| with open(destination, 'wb') as f: |
| for chunk in response.iter_content(chunk_size=8192): |
| f.write(chunk) |
|
|
| return True |
| except Exception as e: |
| print(f"Error downloading file: {e}") |
| return False |
|
|
|
|
| def install_dependencies(): |
| print("Checking runtime dependencies...") |
| try: |
| import parallel_wavegan |
| print("✓ parallel-wavegan is already installed.") |
| except ImportError: |
| print("Installing parallel-wavegan with --no-build-isolation...") |
| cmd = f"{sys.executable} -m pip install parallel-wavegan>=0.5.4 --no-build-isolation" |
| success = run_command(cmd) |
| if not success: |
| print("WARNING: Failed to install parallel-wavegan. TTS might fail.") |
|
|
|
|
| def setup_environment(): |
| print("Setting up the environment for Persian TTS...") |
|
|
| BASE_DIR_PATH = Path(BASE_DIR) |
| MODEL_DIR = BASE_DIR_PATH / "saved_models" / "final_models" |
| RESULTS_DIR = BASE_DIR_PATH / "results" |
|
|
| os.makedirs(MODEL_DIR, exist_ok=True) |
| os.makedirs(RESULTS_DIR, exist_ok=True) |
|
|
| tacotron_repo = BASE_DIR_PATH / "pmt2" |
|
|
| encoder_file = tacotron_repo / "saved_models" / "default" / "encoder.pt" |
| if not os.path.exists(encoder_file): |
| default_model_dir = tacotron_repo / "saved_models" / "default" |
| os.makedirs(default_model_dir, exist_ok=True) |
|
|
| encoder_url = "https://github.com/MahtaFetrat/Persian-MultiSpeaker-Tacotron2/raw/refs/heads/master/saved_models/default/encoder.pt" |
| success = download_file(encoder_url, encoder_file) |
| if not success: |
| print(f"Warning: Could not download encoder model from {encoder_url}") |
| return False |
|
|
| shutil.copy(encoder_file, MODEL_DIR / "encoder.pt") |
|
|
| if not os.path.exists(MODEL_DIR / "vocoder_HiFiGAN.pkl"): |
| hifigan_tar_gz = BASE_DIR_PATH / "train_nodev_all_vctk_hifigan.v1.tar.gz" |
| url = "https://drive.google.com/uc?id=1oVOC4Vf0DYLdDp4r7GChfgj7Xh5xd0ex" |
|
|
| try: |
| print(f"Downloading HiFiGAN model from Google Drive...") |
| gdown.download(url, str(hifigan_tar_gz), quiet=False) |
|
|
| print(f"Extracting {hifigan_tar_gz}...") |
| with tarfile.open(hifigan_tar_gz, 'r:gz') as tar: |
| tar.extractall(path=BASE_DIR_PATH) |
|
|
| extracted_dir = BASE_DIR_PATH / "train_nodev_all_vctk_hifigan.v1" |
|
|
| shutil.move(extracted_dir / "checkpoint-2500000steps.pkl", |
| MODEL_DIR / "vocoder_HiFiGAN.pkl") |
| shutil.move(extracted_dir / "config.yml", MODEL_DIR / "config.yml") |
|
|
| shutil.rmtree(extracted_dir, ignore_errors=True) |
| os.remove(hifigan_tar_gz) |
|
|
| except Exception as e: |
| print(f"Warning: Could not download vocoder") |
| return False |
|
|
| if not os.path.exists(MODEL_DIR / "synthesizer.pt"): |
| synth_url = "https://huggingface.co/MahtaFetrat/Persian-Tacotron2-on-ManaTTS/resolve/main/synthesizer.pt" |
| success = download_file(synth_url, MODEL_DIR / "synthesizer.pt") |
| if not success: |
| print(f"Warning: Could not download synthesizer model from {synth_url}") |
| return False |
|
|
| if not os.path.exists(BASE_DIR_PATH / "sample.wav"): |
| sample_url = "https://huggingface.co/MahtaFetrat/Persian-Tacotron2-on-ManaTTS/resolve/main/sample.wav" |
| success = download_file(sample_url, BASE_DIR_PATH / "sample.wav") |
| if not success: |
| print(f"Warning: Could not download sample audio from {sample_url}") |
| return False |
|
|
| print("Setup complete!") |
| return True |
|
|
