import gradio as gr
import subprocess
import os
import datetime

def combine_video_subtitle(video_file, subtitle_file):
      # Output video file name
    try:
      current_datetime = datetime.datetime.now()
      formatted_datetime = current_datetime.strftime("%Y%m%d_%H%M%S")

      output_file = f'output_combined_{formatted_datetime}.mp4'

      cmd1 = [
          "ffmpeg",
          '-i',
          subtitle_file.name,
          'subtites.ass','&&' ,"ffmpeg",
          "-i",
            video_file.name, '-vf' ,'ass=subtitle.ass',

          output_file
      ]
      # Run ffmpeg command to combine video and subtitle
      cmd2 = [
          "ffmpeg", "-i",
            video_file.name, '-vf' ,'ass=subtitle.ass',

          output_file
      ]

      subprocess.run(cmd1, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

      
      return output_file
    except subprocess.CalledProcessError as e:
          error_message = f"An error occurred: {e.returncode}\n{e.stderr.decode()}"
          print(error_message)
          return None, error_message

# Create Gradio interface
inputs = [
    gr.inputs.File(label="Video File"),
    gr.inputs.File(label="Subtitle File")
]

outputs = gr.outputs.File(label="Combined Video with Subtitle")

title = "Video Subtitle Combiner"
description = "Combine a video file and a subtitle file using ffmpeg."

iface = gr.Interface(fn=combine_video_subtitle, inputs=inputs, outputs=outputs, title=title, description=description)

# Launch the Gradio interface
iface.launch()