Spaces:
Paused
Paused
import os | |
import gradio as gr | |
import ffmpeg | |
import subprocess | |
from gradio import components | |
import tempfile | |
from pathlib import Path | |
from gradio import inputs | |
# 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: File, quality, aspect_ratio, video_url): | |
if video_file is None: | |
output_path = f"{temp_dir}/{Path(video_url).name}.m3u8" | |
else: | |
output_path = f"{temp_dir}/{Path(video_file).name}.m3u8" | |
ffmpeg_command = f"ffmpeg -i {video_file or video_url} -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_file, quality, aspect_ratio, video_url], outputs=[outputs.Video()]).launch(share=False) | |