import os
import sys

sys.path.append(os.getcwd())

try:
    argv = sys.argv[1]
except IndexError:
    argv = None

argv_is_allows = ["--audio_effects", "--audioldm2", "--convert", "--create_dataset", "--create_index", "--extract", "--preprocess", "--separator_music", "--train", "--help_audio_effects", "--help_audioldm2", "--help_convert", "--help_create_dataset", "--help_create_index", "--help_extract", "--help_preprocess", "--help_separator_music", "--help_train", "--help"]

if argv not in argv_is_allows:
    print("Invalid syntax! Use --help for more information")
    quit()

if argv_is_allows[0] in argv: from main.inference.audio_effects import main
elif argv_is_allows[1] in argv: from main.inference.audioldm2 import main
elif argv_is_allows[2] in argv: from main.inference.convert import main
elif argv_is_allows[3] in argv: from main.inference.create_dataset import main
elif argv_is_allows[4] in argv: from main.inference.create_index import main
elif argv_is_allows[5] in argv: from main.inference.extract import main
elif argv_is_allows[6] in argv: from main.inference.preprocess import main
elif argv_is_allows[7] in argv: from main.inference.separator_music import main
elif argv_is_allows[8] in argv: from main.inference.train import main
elif argv_is_allows[9] in argv:
    print("""Parameters for `--audio_effects`:
        1. File paths:
            - `--input_path` (required): Path to the input audio file.
            - `--output_path` (default: `./audios/apply_effects.wav`): Path to save the output file.
            - `--export_format` (default: `wav`): Output file format (`wav`, `mp3`, ...).

        2. Resampling:
            - `--resample` (default: `False`): Whether to resample or not.
            - `--resample_sr` (default: `0`): New sampling frequency (Hz).

        3. Chorus effect:
            - `--chorus`: Enable/disable chorus.
            - `--chorus_depth`, `--chorus_rate`, `--chorus_mix`, `--chorus_delay`, `--chorus_feedback`: Parameters to adjust chorus.

        4. Distortion effect:
            - `--distortion`: Enable/disable distortion.
            - `--drive_db`: Degree of audio distortion.

        5. Reverb effect:
            - `--reverb`: Enable/disable reverb.
            - `--reverb_room_size`, `--reverb_damping`, `--reverb_wet_level`, `--reverb_dry_level`, `--reverb_width`, `--reverb_freeze_mode`: Adjust reverb.

        6. Pitch shift effect:
            - `--pitchshift`: Enable/disable pitch shift.
            - `--pitch_shift`: Pitch shift value.

        7. Delay effect:
            - `--delay`: Enable/disable delay.
            - `--delay_seconds`, `--delay_feedback`, `--delay_mix`: Adjust delay time, feedback, and mix.

        8. Compressor:
            - `--compressor`: Enable/disable compressor.
            - `--compressor_threshold`, `--compressor_ratio`, `--compressor_attack_ms`, `--compressor_release_ms`: Compression parameters.

        9. Limiter:
            - `--limiter`: Enable/disable audio level limiter.
            - `--limiter_threshold`, `--limiter_release`: Limiter threshold and release time.

        10. Gain (Amplification):
            - `--gain`: Enable/disable gain.
            - `--gain_db`: Gain level (dB).

        11. Bitcrush:
            - `--bitcrush`: Enable/disable bit resolution reduction effect.
            - `--bitcrush_bit_depth`: Bit depth for bitcrush.

        12. Clipping:
            - `--clipping`: Enable/disable audio clipping.
            - `--clipping_threshold`: Clipping threshold.

        13. Phaser:
            - `--phaser`: Enable/disable phaser effect.
            - `--phaser_rate_hz`, `--phaser_depth`, `--phaser_centre_frequency_hz`, `--phaser_feedback`, `--phaser_mix`: Adjust phaser effect.

        14. Boost bass & treble:
            - `--treble_bass_boost`: Enable/disable bass and treble boost.
            - `--bass_boost_db`, `--bass_boost_frequency`, `--treble_boost_db`, `--treble_boost_frequency`: Bass and treble boost parameters.

        15. Fade in & fade out:
            - `--fade_in_out`: Enable/disable fade effect.
            - `--fade_in_duration`, `--fade_out_duration`: Fade in/out duration.

        16. Audio combination:
            - `--audio_combination`: Enable/disable combining multiple audio files.
            - `--audio_combination_input`: Path to additional audio files.
    """)
    quit()
elif argv_is_allows[10] in argv:
    print("""Parameters for `--audioldm2`:
        1. File paths:
            - `--input_path` (required): Path to the input audio file.
            - `--output_path` (default: `./output.wav`): Path to save the output file.
            - `--export_format` (default: `wav`): Output file format.

        2. Audio configuration:
            - `--sample_rate` (default: `44100`): Sampling frequency (Hz).

        3. AudioLDM model configuration:
            - `--audioldm_model` (default: `audioldm2-music`): Select AudioLDM model for processing.

        4. Model guidance prompt:
            - `--source_prompt` (default: ``): Description of source audio.
            - `--target_prompt` (default: ``): Description of target audio.

        5. Processing algorithm configuration:
            - `--steps` (default: `200`): Number of steps in audio synthesis process.
            - `--cfg_scale_src` (default: `3.5`): Guidance scale for source audio.
            - `--cfg_scale_tar` (default: `12`): Guidance scale for target audio.
            - `--t_start` (default: `45`): Editing level.

        6. Computation optimization:
            - `--save_compute` (default: `False`): Whether to enable compute optimization mode.
    """)
    quit()
elif argv_is_allows[11] in argv:
    print("""Parameters for `--convert`:
        1. Voice processing configuration:
            - `--pitch` (default: `0`): Adjust pitch.
            - `--filter_radius` (default: `3`): F0 curve smoothness.
            - `--index_rate` (default: `0.5`): Voice index usage rate.
            - `--volume_envelope` (default: `1`): Volume amplitude adjustment factor.
            - `--protect` (default: `0.33`): Consonant protection.

        2. Frame hop configuration:
            - `--hop_length` (default: `64`): Hop length during audio processing.

        3. F0 configuration:
            - `--f0_method` (default: `rmvpe`): F0 prediction method (`pm`, `dio`, `mangio-crepe-tiny`, `mangio-crepe-small`, `mangio-crepe-medium`, `mangio-crepe-large`, `mangio-crepe-full`, `crepe-tiny`, `crepe-small`, `crepe-medium`, `crepe-large`, `crepe-full`, `fcpe`, `fcpe-legacy`, `rmvpe`, `rmvpe-legacy`, `harvest`, `yin`, `pyin`, `swipe`).
            - `--f0_autotune` (default: `False`): Whether to auto-tune F0.
            - `--f0_autotune_strength` (default: `1`): Strength of F0 auto-tuning.
            - `--f0_file` (default: ``): Path to existing F0 file.
            - `--f0_onnx` (default: `False`): Whether to use ONNX version of F0.

        4. Embedding model:
            - `--embedder_model` (default: `contentvec_base`): Embedding model used.
            - `--embedders_mode` (default: `fairseq`): Embedding mode (`fairseq`, `transformers`, `onnx`).

        5. File paths:
            - `--input_path` (required): Path to input audio file.
            - `--output_path` (default: `./audios/output.wav`): Path to save output file.
            - `--export_format` (default: `wav`): Output file format.
            - `--pth_path` (required): Path to `.pth` model file.
            - `--index_path` (default: `None`): Path to index file (if any).

        6. Audio cleaning:
            - `--clean_audio` (default: `False`): Whether to apply audio cleaning.
            - `--clean_strength` (default: `0.7`): Cleaning strength.

        7. Resampling & audio splitting:
            - `--resample_sr` (default: `0`): New sampling frequency (0 means keep original).
            - `--split_audio` (default: `False`): Whether to split audio before processing.

        8. Testing & optimization:
            - `--checkpointing` (default: `False`): Enable/disable checkpointing to save RAM.

        9. Formant shifting:
            - `--formant_shifting` (default: `False`): Whether to enable formant shifting effect.
            - `--formant_qfrency` (default: `0.8`): Formant shift frequency factor.
            - `--formant_timbre` (default: `0.8`): Voice timbre change factor.
    """)
    quit()
elif argv_is_allows[12] in argv:
    print("""Parameters for `--create_dataset`:
        1. Dataset paths & configuration:
            - `--input_audio` (required): Path to audio link (YouTube link, can use `,` for multiple links).
            - `--output_dataset` (default: `./dataset`): Output data directory.
            - `--sample_rate` (default: `44100`): Audio sampling frequency.

        2. Data cleaning:
            - `--clean_dataset` (default: `False`): Whether to apply data cleaning.
            - `--clean_strength` (default: `0.7`): Data cleaning strength.

        3. Voice separation & effects:
            - `--separator_reverb` (default: `False`): Whether to separate voice reverb.
            - `--kim_vocal_version` (default: `2`): Kim Vocal model version for separation (`1`, `2`).

        4. Audio segmentation configuration:
            - `--overlap` (default: `0.25`): Overlap level between segments during separation.
            - `--segments_size` (default: `256`): Size of each segment.

        5. MDX (Music Demixing) configuration:
            - `--mdx_hop_length` (default: `1024`): MDX hop length during processing.
            - `--mdx_batch_size` (default: `1`): Batch size during MDX processing.
            - `--denoise_mdx` (default: `False`): Whether to apply denoising during MDX separation.

        6. Skip audio sections:
            - `--skip` (default: `False`): Whether to skip any audio seconds.
            - `--skip_start_audios` (default: `0`): Time (seconds) to skip at the start of audio.
            - `--skip_end_audios` (default: `0`): Time (seconds) to skip at the end of audio.
    """)
    quit()
elif argv_is_allows[13] in argv:
    print("""Parameters for `--create_index`:
        1. Model information:
            - `--model_name` (required): Model name.
            - `--rvc_version` (default: `v2`): Version (`v1`, `v2`).
            - `--index_algorithm` (default: `Auto`): Index algorithm used (`Auto`, `Faiss`, `KMeans`).
    """)
    quit()
elif argv_is_allows[14] in argv:
    print("""Parameters for `--extract`:
        1. Model information:
            - `--model_name` (required): Model name.
            - `--rvc_version` (default: `v2`): RVC version (`v1`, `v2`).

        2. F0 configuration:
            - `--f0_method` (default: `rmvpe`): F0 prediction method (`pm`, `dio`, `mangio-crepe-tiny`, `mangio-crepe-small`, `mangio-crepe-medium`, `mangio-crepe-large`, `mangio-crepe-full`, `crepe-tiny`, `crepe-small`, `crepe-medium`, `crepe-large`, `crepe-full`, `fcpe`, `fcpe-legacy`, `rmvpe`, `rmvpe-legacy`, `harvest`, `yin`, `pyin`, `swipe`).
            - `--pitch_guidance` (default: `True`): Whether to use pitch guidance.

        3. Processing configuration:
            - `--hop_length` (default: `128`): Hop length during processing.
            - `--cpu_cores` (default: `2`): Number of CPU threads used.
            - `--gpu` (default: `-`): Specify GPU to use (e.g., `0` for first GPU, `-` to disable GPU).
            - `--sample_rate` (required): Input audio sampling frequency.

        4. Embedding configuration:
            - `--embedder_model` (default: `contentvec_base`): Embedding model name.
            - `--f0_onnx` (default: `False`): Whether to use ONNX version of F0.
            - `--embedders_mode` (default: `fairseq`): Embedding mode (`fairseq`, `transformers`, `onnx`).
    """)
    quit()
elif argv_is_allows[15] in argv:
    print("""Parameters for `--preprocess`:
        1. Model information:
            - `--model_name` (required): Model name.

        2. Data configuration:
            - `--dataset_path` (default: `./dataset`): Path to directory containing data files.
            - `--sample_rate` (required): Audio data sampling frequency.

        3. Processing configuration:
            - `--cpu_cores` (default: `2`): Number of CPU threads used.
            - `--cut_preprocess` (default: `True`): Whether to cut data files.
            - `--process_effects` (default: `False`): Whether to apply preprocessing.
            - `--clean_dataset` (default: `False`): Whether to clean data files.
            - `--clean_strength` (default: `0.7`): Data cleaning strength.
    """)
    quit()
elif argv_is_allows[16] in argv:
    print("""Parameters for `--separator_music`:
        1. Data paths:
            - `--input_path` (required): Path to input audio file.
            - `--output_path` (default: `./audios`): Directory to save output files.
            - `--format` (default: `wav`): Output file format (`wav`, `mp3`, ...).

        2. Audio processing configuration:
            - `--shifts` (default: `2`): Number of predictions.
            - `--segments_size` (default: `256`): Audio segment size.
            - `--overlap` (default: `0.25`): Overlap level between segments.
            - `--mdx_hop_length` (default: `1024`): MDX hop length during processing.
            - `--mdx_batch_size` (default: `1`): Batch size.

        3. Cleaning processing:
            - `--clean_audio` (default: `False`): Whether to clean audio.
            - `--clean_strength` (default: `0.7`): Cleaning filter strength.

        4. Model configuration:
            - `--model_name` (default: `HT-Normal`): Music separation model (`Main_340`, `Main_390`, `Main_406`, `Main_427`, `Main_438`, `Inst_full_292`, `Inst_HQ_1`, `Inst_HQ_2`, `Inst_HQ_3`, `Inst_HQ_4`, `Inst_HQ_5`, `Kim_Vocal_1`, `Kim_Vocal_2`, `Kim_Inst`, `Inst_187_beta`, `Inst_82_beta`, `Inst_90_beta`, `Voc_FT`, `Crowd_HQ`, `Inst_1`, `Inst_2`, `Inst_3`, `MDXNET_1_9703`, `MDXNET_2_9682`, `MDXNET_3_9662`, `Inst_Main`, `MDXNET_Main`, `MDXNET_9482`, `HT-Normal`, `HT-Tuned`, `HD_MMI`, `HT_6S`).
            - `--kara_model` (default: `Version-1`): Backing track separation model version (`Version-1`, `Version-2`).

        5. Effects and post-processing:
            - `--backing` (default: `False`): Whether to separate backing vocals.
            - `--mdx_denoise` (default: `False`): Whether to use MDX denoising.
            - `--reverb` (default: `False`): Whether to separate reverb.
            - `--backing_reverb` (default: `False`): Whether to separate reverb for backing vocals.

        6. Sampling frequency:
            - `--sample_rate` (default: `44100`): Output audio sampling frequency.
    """)
    quit()
elif argv_is_allows[17] in argv:
    print("""Parameters for `--train`:
        1. Model configuration:
            - `--model_name` (required): Model name.
            - `--rvc_version` (default: `v2`): RVC version (`v1`, `v2`).
            - `--model_author` (optional): Model author.

        2. Save configuration:
            - `--save_every_epoch` (required): Number of epochs between saves.
            - `--save_only_latest` (default: `True`): Save only the latest checkpoint.
            - `--save_every_weights` (default: `True`): Save all model weights.

        3. Training configuration:
            - `--total_epoch` (default: `300`): Total number of training epochs.
            - `--batch_size` (default: `8`): Batch size during training.
            - `--sample_rate` (required): Audio sampling frequency.

        4. Device configuration:
            - `--gpu` (default: `0`): Specify GPU to use (agena: Specify GPU to use (number or `-` if not using GPU).
            - `--cache_data_in_gpu` (default: `False`): Cache data in GPU for faster processing.

        5. Advanced training configuration:
            - `--pitch_guidance` (default: `True`): Use pitch guidance.
            - `--g_pretrained_path` (default: ``): Path to pretrained G weights.
            - `--d_pretrained_path` (default: ``): Path to pretrained D weights.
            - `--vocoder` (default: `Default`): Vocoder used (`Default`, `MRF-HiFi-GAN`, `RefineGAN`).

        6. Overtraining detection:
            - `--overtraining_detector` (default: `False`): Enable/disable overtraining detection.
            - `--overtraining_threshold` (default: `50`): Threshold for detecting overtraining.

        7. Data processing:
            - `--cleanup` (default: `False`): Clean old training files to start training from scratch.

        8. Optimization:
            - `--checkpointing` (default: `False`): Enable/disable checkpointing to save RAM.
            - `--deterministic` (default: `False`): When enabled, uses deterministic algorithms to ensure consistent results for the same input data.
            - `--benchmark` (default: `False`): When enabled, tests and selects the optimal algorithm for the hardware and specific size.
    """)
    quit()
elif argv_is_allows[18] in argv:
    print("""Usage:
        1. `--help_audio_effects`: Help for adding audio effects.
        2. `--help_audioldm2`: Help for music editing.
        3. `--help_convert`: Help for audio conversion.
        4. `--help_create_dataset`: Help for creating training data.
        5. `--help_create_index`: Help for creating an index.
        6. `--help_extract`: Help for extracting training data.
        7. `--help_preprocess`: Help for preprocessing data.
        8. `--help_separator_music`: Help for music separation.
        9. `--help_train`: Help for model training.
    """)
    quit()

if __name__ == "__main__":
    if "--train" in argv:
        import torch.multiprocessing as mp
        mp.set_start_method("spawn")
        
    try:
        main()
    except:
        pass