import typer
from typing_extensions import Annotated

# NEW
import os 
import calendar
 
from pathlib import Path

from programs import (
    bad_contrast,
    bad_predator,
    blur_pix,
    broken_scroll,
    convert,
    double_cup,
    download_yt,
    extract_frames, 
    extract_sound, 
    fever,
    frame_delay_pro1,
    frame_delay_pro2,
    light_snow,
    looper_pro,
    lsd_feedback,
    mirror_delay,
    nostalgic_stutter,
    num_edits,
    overexposed_stutter,
    overlay_img_pro,
    pickle_juice,
    rb_blur,
    resize, 
    reverse,
    scrolling_pro,
    silence_xtraction,    
    slight_smear,
    speed,
    stack_2x, 
    steel_wash,   
    stutter_pro,
    transcraibe)

from utils import load_config
from datetime import datetime

from pyfiglet import Figlet
a = Figlet(font='ogre')
print(a.renderText("videobeaux"))
print("Your friendly multilateral video toolkit built for artists by artists.")
print("It's your best friend.")
print('-' * 50)

config = load_config.load_config()

proj_mgmt_config = config['proj_mgmt']
v_ext = proj_mgmt_config['default_video_file_ext']
a_ext = proj_mgmt_config['default_audio_file_ext']

now = datetime.now()
ct = now.strftime("%Y-%m-%d_%H-%M-%S")

app = typer.Typer()

#################
# bad_contrast
#################
@app.command('bad-contrast', help='Apply a bad constrast effect.')
def bad_contrast_vb(
    input_file: str = typer.Argument(None, help="Input video file"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['bad_contrast']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    bad_contrast.bad_contrast(**params)

############
# blur-pix
############
@app.command('blur-pix', help='Apply blur pix effect to video file.')
def blur_pix_video(
    input_file: str = typer.Argument(None, help="Input video file"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['blur_pix']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    blur_pix.blur_pix(**params)

############
# bad_predator
############
@app.command('bad_predator', help='Apply bad Predator heat vision effect to video file.')
def bad_predator_vb(
    input_file1: str = typer.Argument(None, help="Input video file 1"), 
    input_file2: str = typer.Argument(None, help="Input video file 2"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file1": input_file1,
        "input_file2": input_file2,
        "output_file": output_file
    }
    defaults = config['bad_predator']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    bad_predator.bad_predator(**params)

#############
# broken_scroll
#############
@app.command('broken-scroll', help='Apply broken_scroll effect to video file.')
def broken_scroll_vb(
    input_file: str = typer.Argument(None, help="Input video file"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['broken_scroll']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    broken_scroll.broken_scroll(**params)

###########
# convert
###########
@app.command('convert', help='Convert a video to a different format.')
def convert_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file"),
    format: str = typer.Argument(None, help="Format of the output video")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file, 
        "format": format
    }
    defaults = config['convert']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    convert.convert(**params)

###########
# fever
###########
@app.command('fever', help='Apply fever effect to video.')
def fever_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['fever']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    fever.fever(**params)   

###########
# double_cup
###########
@app.command('double-cup', help='Apply the effect of purple drank.')
def double_cup_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['double_cup']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    double_cup.double_cup(**params)    

##########
# download_yt : yt-dlp 
##########
@app.command('download-yt', help='Downloads the provided link with yt-dlp')
def yt_dlp_vb(
    yt_url: str = typer.Argument(None, help="URL of the YT video"),
    output_file: str = typer.Argument(None, help="Width of the output video"),
    format: str = typer.Argument(v_ext, help="Width of the output video"),
):
    params = { 
        "yt_url": yt_url,
        "output_file": output_file,  
        "format": format
    }
    defaults = config['download_yt']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    download_yt.download_yt(**params)

##################
# extract-frames
##################
@app.command('extract-frames', help='Extract frames from a video at the specified frame rate.')
def extract_frames_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output folder for frames"),
    frame_rate: int = typer.Argument(None, help="Frame rate for extracting frames")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file, 
        "frame_rate": frame_rate
    }
    defaults = config['extract_frames']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    extract_frames.extract_frames(**params)

#################
# extract-sound
#################
@app.command('extract-sound', help='Extract audio from video file.')
def extract_sound_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output audio file")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['extract_sound']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    print(params)
    extract_sound.extract_sound(**params)

####################
# frame-delay-pro1
####################
@app.command('frame-delay-pro1', help='Apply the pro1 frame delay to video file.')
def frame_delay_pro1_vb(
    input_file: str = typer.Argument(None, help="Input video file "),
    num_of_frames: int = typer.Argument(None, help="Input weight for frame delay"),    
    frame_weights: str = typer.Argument(None, help="Input weight for frame delay"),    
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "num_of_frames": num_of_frames,
        "frame_weights": frame_weights,
        "output_file": output_file
    }
    defaults = config['frame_delay_pro1']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    frame_delay_pro1.frame_delay_pro1(**params)

####################
# frame-delay-pro2
####################
@app.command('frame-delay-pro2', help='Apply the pro2 frame delay to video file.')
def frame_delay_pro2_vb(
    input_file: str = typer.Argument(None, help="Input video file "),
    decay: int = typer.Argument(None, help=""),    
    planes: str = typer.Argument(None, help="Input weight for frame delay"),    
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "decay": decay,
        "planes": planes,
        "output_file": output_file
    }
    defaults = config['frame_delay_pro2']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    frame_delay_pro2.frame_delay_pro2(**params)

###########
# light_snow
###########
@app.command('light_snow', help='Slightly smearing RGB color space.')
def light_snow_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['light_snow']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    light_snow.light_snow(**params)

#################
# looper-pro
#################
@app.command('looper-pro', help='Apply video looper effect base on frame size & start frame.')
def scrolling_pro_video(
    input_file: str = typer.Argument(None, help="Input video file"), 
    loop_count: str = typer.Argument(None, help="Number of video loops"), 
    size_in_frames: str = typer.Argument(None, help="Size of loop in frames"), 
    start_frame: str = typer.Argument(None, help="Starting frame of loop"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "loop_count": loop_count,
        "size_in_frames": size_in_frames,
        "start_frame": start_frame,
        "output_file": output_file
    }
    defaults = config['looper_pro']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    looper_pro.looper_pro(**params)

################
# lsd-feedback
################
@app.command('lsd-feedback', help='Apply LSD feedback effect to video file.')
def lsd_feedback_vb(
    input_file: str = typer.Argument(None, help="Input video file "),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['lsd_feedback']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    lsd_feedback.lsd_feedback(**params)

################
# mirror-delay
################
@app.command('mirror-delay', help='Apply mirrored delay effect to video file.')
def mirror_delay_vb(
    input_file: str = typer.Argument(None, help="Input video file"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['mirror_delay']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    print(params)
    mirror_delay.mirror_delay(**params)

#####################
# nostalgic-stutter
#####################
@app.command('nostalgic-stutter', help='Apply nostaglic stutter effect to video file.')
def nostalgic_stutter_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['nostalgic_stutter']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    nostalgic_stutter.nostalgic_stutter(**params)

################
# num-edits
################
@app.command('num-edits', help='Apply LSD feedback effect to video file.')
def num_edits_vb(
    input_file: str = typer.Argument(None, help="Input video file "),
    count: str = typer.Argument(None, help="Input video file "),
    output_file: str = typer.Argument(None, help="Input video file ")
):
    params = { 
        "input_file": input_file,
        "count": count,
        "output_file": output_file
    }
    defaults = config['num_edits']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    num_edits.num_edits(**params)

######################
# overexposed-stutter
######################
@app.command('overexposed-stutter', help='Apply overexposed stutter effect to video file.')
def overexposed_stutter_vb(
    input_file: str = typer.Argument(None, help="Input video file"), 
    output_file: str = typer.Argument(None, help="Output video file")
):

    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['overexposed_stutter']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    overexposed_stutter.overexposed_stutter(**params)

######################
# overlay_img_pro
######################
@app.command('overlay-img-pro', help='Overlay an image with location & dimension control.')
def overlay_img_pro_vb(
    input_file: str = typer.Argument(None, help="Input video file "),
    overlay_image: int = typer.Argument(None, help="Image file"),    
    x_position: str = typer.Argument(None, help="X position of the image file"),    
    y_position: str = typer.Argument(None, help="Y position of the image file"),
    overlay_width: str = typer.Argument(None, help="Overlay image width"),
    overlay_height: str = typer.Argument(None, help="Overlay image height"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "overlay_image": overlay_image,
        "x_position": x_position,
        "y_position": y_position,
        "overlay_width": overlay_width,
        "overlay_height": overlay_height,
        "output_file": output_file
    }
    defaults = config['overlay_img_pro']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    overlay_img_pro.overlay_img_pro(**params)



##########
# pickle_juice 
##########
@app.command('pickle_juice', help='Apply filter like the video was dipped in pickle juice.')
def pickle_juice_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file"),
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['pickle_juice']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    pickle_juice.pickle_juice(**params)

##########
# rb_blur 
##########
@app.command('rb-blur', help='Resize a video to the given width and height.')
def rb_blur_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file"),
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['rb_blur']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    rb_blur.rb_blur(**params)

##########
# resize 
##########
@app.command('resize', help='Resize a video to the given width and height.')
def resize_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file"),
    width: int = typer.Argument(None, help="Width of the output video"),
    height: int = typer.Argument(None, help="Height of the output video")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file, 
        "width": width, 
        "height": height 
    }
    defaults = config['resize']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    resize.resize(**params)

###########
# reverse
###########
@app.command('reverse', help='Reverse video file.')
def reverse_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['reverse']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    reverse.reverse(**params)

#################
# scrolling-pro
#################
@app.command('scrolling-pro', help='Apply scrolling pro effect to video file.')
def scrolling_pro_video(
    input_file: str = typer.Argument(None, help="Input video file"), 
    horizontal: str = typer.Argument(None, help="Horizontal scroll parameter"), 
    vertical: str = typer.Argument(None, help="Vertical scroll parameter"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "horizontal": horizontal,
        "vertical": vertical,
        "output_file": output_file
    }
    defaults = config['scrolling_pro']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    scrolling_pro.scrolling_pro(**params)


#####################
# silence-xtraction
#####################
@app.command('silence-xtraction', help="Stitches togehter video chunks that have no discernable words." +
              "This does NOT use audio analysis, but instead identifes the presence of a 'word' using the .srt transcription file")
def silence_xtraction_vb(
    min_d: int = typer.Argument(None, help="Minimum duration of a chunk of silence."),
    max_d: int = typer.Argument(None, help="Maximum duration of a chunk of silence."),
    adj: int = typer.Argument(None, help="Adjustment value"),
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "min_d": min_d,
        "max_d": max_d,
        "adj": adj,
        "input_file": input_file, 
        "output_file": output_file,
    }

    defaults = config['silence_x']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    silence_xtraction.silence_xtraction(**params)

###########
# slight_smear
###########
@app.command('slight_smear', help='Slightly smearing RGB color space.')
def slight_smear_vb(
    input_file: str = typer.Argument(None, help="Input video file"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file, 
        "output_file": output_file
    }
    defaults = config['slight_smear']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    slight_smear.slight_smear(**params)

#############
# speed
#############
@app.command('speed', help='Apply speed effect to video file.')
def speed_vb(
    input_file: str = typer.Argument(None, help="Input video file"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['speed']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    speed.speed(**params)

############
# stack-2x
############
@app.command('stack-2x', help='Stack 2 videos on top of each other keeping the original orientation.')
def stack_2x_vb(
    input_file1: str = typer.Argument(None, help="Input video file 1"),
    input_file2: str = typer.Argument(None, help="Input video file 2"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file1": input_file1,
        "input_file2": input_file2, 
        "output_file": output_file
    }
    defaults = config['stack_2x']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    stack_2x.stack_2x(**params)

############
# steel_wash
############
@app.command('steel_wash', help='Apply steel blue filter to video.')
def steel_wash_vb(
    input_file: str = typer.Argument(None, help="Input video file 1"),
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "output_file": output_file
    }
    defaults = config['steel_wash']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    steel_wash.steel_wash(**params)

###############
# stutter-pro
###############
@app.command('stutter-pro', help='Apply stutter pro effect to video file.')
def stutter_pro_vb(
    input_file: str = typer.Argument(None, help="Input video file"), 
    stutter: str = typer.Argument(None, help="Frame stutter parameter"), 
    output_file: str = typer.Argument(None, help="Output video file")
):
    params = { 
        "input_file": input_file,
        "stutter": stutter,
        "output_file": output_file
    }
    defaults = config['stutter_pro']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    stutter_pro.stutter_pro(**params)

###############
# transcraibe
###############
@app.command('transcraibe', help='Transcribes the video with ai (vosk) - transcrAIbe')
def transcraibe_vb(
    input_file: str = typer.Argument(None, help='Video file you would like to transcribe.'),
    stt_model: str = typer.Argument(None, help="URL of the YT video")
):
    params = { 
        "input_file": input_file,
        "stt_model": stt_model
    }
    defaults = config['transcraibe']
    params = {key: params.get(key) or defaults[key] for key in defaults}
    transcraibe.vosk_stt(**params)

if __name__ == "__main__":
    app()