# Import the os module
import os
from utils.default_models import ensure_default_models
from pathlib import Path
Sagemaker = False
if Sagemaker :
    env='source activate python3 && conda activate VideoMessage &&'
else:
    env='' 
## Step 1. Setup of the dependencies
is_first_time = True

#Install dependency
# Download pretrained model
# Get the current working directory
parent_dir = os.getcwd()
print(parent_dir)
if is_first_time:
    # Directory 
    directory = "sample_data"
    # Path 
    path = os.path.join(parent_dir, directory) 
    print(path)
    try:
        os.mkdir(path)
        print("Directory '% s' created" % directory) 
    except Exception:
         print("Directory '% s'was already created" % directory)
if is_first_time:
    os.system('git clone https://github.com/Rudrabha/Wav2Lip')
    #os.system('cd Wav2Lip &&{} pip install  -r requirements.txt'.format(env))
    ## Load the models one by one.
    print("Preparing the models of Wav2Lip")
    ensure_default_models(Path("Wav2Lip"))    
    os.system('git clone https://github.com/Edresson/Coqui-TTS -b multilingual-torchaudio-SE TTS')
    os.system('{} pip install -q -e TTS/'.format(env))
    os.system('{} pip install -q torchaudio==0.9.0'.format(env))
    os.system('{} pip install -q youtube-dl'.format(env))
    os.system('{} pip install ffmpeg-python'.format(env))
    os.system('{} pip install gradio==3.0.4'.format(env))
    os.system('{} pip install pytube==12.1.0'.format(env))
    os.system('{} pip install torchaudio==0.9.0 TTS'.format(env))    
    os.system('{} pip install opencv-contrib-python-headless==4.1.2.30'.format(env))
    os.system('{} pip install IPython==7.32.0'.format(env))
print("Installation repositories DONE!!")
#tqdm>=4.62.1