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

# 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")
aspect_ratio = gr.inputs.Dropdown(choices=[
  "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="16:9")

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)] + ".m3u8"
        
        input_path = video_file.name
    
    # If a URL was provided
    elif video_url is not None:
        # 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)] + ".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, output_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
        ffmpeg_command = f"ffmpeg -i {input_path} -c:v libx264 -crf {quality} -vf scale=-1:720,setsar=1:1 -hls_time 6 -hls_playlist_type vod -f hls {output_path}"

        try:
            subprocess.run(ffmpeg_command, shell=True, timeout=10)
        except subprocess.TimeoutExpired:
            print("ffmpeg timed out")
            return "ffmpeg 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()