import time
import os
import gradio as gr
import requests
import subprocess
from gradio import components
import tempfile
from pathlib import Path
from gradio import inputs, outputs
from moviepy.editor import VideoFileClip

# Define File object
File = Path

# Define temp directory
temp_dir = tempfile.mkdtemp()

video_file = gr.inputs.File(label="Video File", type="file")
quality = gr.inputs.Dropdown(choices=["18", "23", "28", "32"], label="Quality", default="23")

# default will now be None, which will retain original aspect ratio
aspect_ratio = gr.inputs.Dropdown(choices=[
  None,  # Add None option to retain original aspect ratio
  "1:1",
  "4:3",
  "3:2",
  "5:4",
  "16:9",
  "21:9",
  "1.85:1",
  "2.35:1",
  "3:1",
  "360",
  "9:16",
  "16:9",
  "2:1",
  "1:2",
  "9:1",
], label="Aspect Ratio", default=None) 

video_url = gr.inputs.Textbox(label="Video URL")

def convert_video(video_file, quality, aspect_ratio, video_url):
    # If a file was uploaded
    if video_file is not None:
        # Get the file extension
        file_extension = video_file.name.split(".")[-1]

        # Define output file name
        output_file_name = video_file.name[:-len(file_extension) - 1] + ".m3u8"
        
        input_path = video_file.name
    
    # If a URL was provided
    elif video_url:
        # Get the file name from URL
        file_name = video_url.split("/")[-1]

        # Get the file extension
        file_extension = file_name.split(".")[-1]

        # Define output file name
        output_file_name = file_name[:-len(file_extension) - 1] + ".m3u8"
        
        # Download the file to the temporary directory
        response = requests.get(video_url)
        
        if response.status_code == 200:
            input_path = os.path.join(temp_dir, file_name)
            with open(input_path, 'wb') as f:
                f.write(response.content)
        else:
            print("Failed to download file from URL.")
            return "Failed to download file from URL."
    else:
        print("No input was provided.")
        return "No input was provided."

    # Define the output path
    output_path = os.path.join(temp_dir, output_file_name)

    # Check if the output file exists
    if os.path.exists(output_path):
        # The file already exists, so we can just display it in the output viewer
        return output_path
    else:
        # The file does not exist, so we need to convert it
        if aspect_ratio is None:
            video = VideoFileClip(input_path)
            aspect_ratio = f"{video.size[0]}:{video.size[1]}"
        
        ffmpeg_command = f"ffmpeg -i {input_path} -c:v libx264 -crf {quality} -vf scale=-1:720,setsar={aspect_ratio} -hls_time 6 -hls_playlist_type vod -f hls {output_path}"

        try:
            subprocess.run(ffmpeg_command, shell=True, timeout=600)
        except subprocess.TimeoutExpired:
            print("ffmpeg timed out")
            return "ffmpeg command timed out."
        except FileNotFoundError:
            print("ffmpeg is not installed.")
            return "ffmpeg is not installed."

        for i in range(10):
            if os.path.exists(output_path):
                break
            else:
                time.sleep(1)

        # The file has now been converted, so we can display it in the output viewer
        return output_path

gr.Interface(convert_video, inputs=[video_file, quality, aspect_ratio, video_url], outputs='file').launch()