from tool.file_name import *
import ffmpeg


def convert_video_to_audio(input_file, start_time, end_time):
    print("CONVERTING VIDEO TO AUDIO")
    print(f"INPUT: {input_file}")
    print(f"OUTPUT: {dir_cut_audio_file}")

    (
        ffmpeg
        .input(input_file)
        .output(dir_cut_audio_file, ss=start_time, to=end_time, format="wav", acodec='pcm_s16le')
        .run(overwrite_output=True)
    )


def cut_video(input_file, start_time, end_time):
    print("CUTTING VIDEO")
    print(f"INPUT: {input_file}")
    print(f"OUTPUT: {dir_cut_video_file}")

    (
        ffmpeg
        .input(input_file)
        .output(dir_cut_video_file, ss=start_time, to=end_time, acodec='copy', vcodec='copy')
        .run(overwrite_output=True)
    )


def add_subtitle_to_video():
    print("ADDING SUBTITLE")
    print(f"SUBTITLE: {dir_adjusted_subtitle_file}")
    print(f"OUTPUT: {dir_video_subtitle_file}")

    (
        ffmpeg
        .input(dir_cut_video_file)
        .output(dir_video_subtitle_file, vf=f'subtitles={dir_adjusted_subtitle_file}', acodec='copy')
        .run(overwrite_output=True)
    )