Spaces:
Paused
Paused
File size: 3,301 Bytes
485608a faa2bdb c8942d9 79683ca faa2bdb 7a80d3a 5296733 40446f2 79683ca 40446f2 ec46e62 701ed9a b7d9c54 131761b 79683ca 06412df 79683ca 6f1f7b0 79683ca 6f1f7b0 79683ca 6f1f7b0 79683ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
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: dict, 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"
# Write the uploaded file to the temporary directory
with open(os.path.join(temp_dir, video_file['name']), 'wb') as f:
f.write(video_file['content'])
input_path = os.path.join(temp_dir, video_file['name'])
# If a URL was provided
elif video_url is not None:
# Get the file extension
file_extension = video_url.split(".")[-1]
# Define output file name
output_file_name = video_url.split("/")[-1][:-len(file_extension)] + ".m3u8"
# Download the file to the temporary directory
response = requests.get(video_url)
if response.status_code == 200:
with open(os.path.join(temp_dir, output_file_name), 'wb') as f:
f.write(response.content)
input_path = os.path.join(temp_dir, output_file_name)
else:
print("Failed to download file from URL.")
return None
else:
print("No input was provided.")
return None
# 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 gr.outputs.File(output_path, label="Download")
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 None
except FileNotFoundError:
print("ffmpeg is not installed.")
return None
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 gr.outputs.File(output_path, label="Download")
gr.Interface(convert_video, inputs=[video_file, quality, aspect_ratio, video_url], outputs='file').launch()
|