Jeffgold's picture
Update app.py
2024db3
raw
history blame
1.03 kB
import gradio as gr
import ffmpeg
import subprocess
from gradio import components
import tempfile
from pathlib import Path
# Define File object
File = Path
video_file = gr.inputs.File(label="Video File", accept="video/*", required=False)
quality = gr.inputs.Dropdown(choices=["18", "23", "28", "32"], label="Quality", default="23")
aspect_ratio = gr.inputs.Dropdown(choices=["16:9", "4:3", "3:2", "2:1"], label="Aspect Ratio", default="16:9")
def convert_video(video_file: File, quality, aspect_ratio):
if video_file is None:
output_path = gr.inputs.Textbox(label="Video URL")
else:
output_path = f"{video_file.name.split('.')[0]}.m3u8"
temp_dir = tempfile.gettempdir()
ffmpeg_command = f"ffmpeg -i {video_file or output_path} -c:v libx264 -crf {quality} -f hls -aspect {aspect_ratio} {temp_dir}/{output_path}"
subprocess.run(ffmpeg_command, shell=True)
return components.Video(output_path)
from gradio import outputs
gr.Interface(convert_video, inputs="video", outputs=[outputs.Video()]).launch()