File size: 4,831 Bytes
d83b57c 6567ae9 d83b57c 76e6c8b d83b57c f699201 d83b57c 6567ae9 c841491 6567ae9 f699201 6567ae9 f699201 6567ae9 f699201 6567ae9 f699201 6567ae9 f699201 6567ae9 f699201 6567ae9 f699201 d83b57c 6567ae9 c841491 6567ae9 c841491 6567ae9 1dc23c5 6567ae9 d83b57c 6567ae9 d83b57c 6567ae9 55f9dd1 6567ae9 d83b57c 6567ae9 d83b57c |
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 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 |
import os
import asyncio
import subprocess
import uuid
import glob
import shutil
import gradio as gr
accel = 'auto'
video_base_opts = ['-crf', '63', '-c:v', 'libx264', '-tune', 'zerolatency']
UPLOAD_FOLDER = 'uploads'
CONVERTED_FOLDER = 'converted'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(CONVERTED_FOLDER, exist_ok=True)
async def run_subprocess(cmd):
process = await asyncio.create_subprocess_exec(
*cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
)
stdout, stderr = await process.communicate()
if process.returncode != 0:
raise subprocess.CalledProcessError(process.returncode, cmd, stderr.decode())
return stdout.decode(), stderr.decode()
async def convert_video_task(input_path, downscale, faster, use_mp3):
if use_mp3:
# Only extract MP3 audio, no splitting
output_path = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp3")
ffmpeg_cmd = [
'ffmpeg', '-y', '-i', input_path, '-vn',
'-c:a', 'libmp3lame', '-b:a', '8k', '-ar', '24000', '-ac', '1', output_path
]
await run_subprocess(ffmpeg_cmd)
return output_path, None
# Non-MP3 path: split audio and video
audio_wav = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.wav")
audio_output = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.m4a")
video_output = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp4")
# Extract mono WAV
await run_subprocess([
'ffmpeg', '-y', '-i', input_path, '-ac', '1', '-ar', '8000', audio_wav
])
# Compress WAV β AAC with local fdkaac
await run_subprocess([
'./fdkaac', '-b', '1k', '-C', '-f', '2', '-G', '1', '-w', '8000',
'-o', audio_output, audio_wav
])
# Compress video with extreme low quality
video_cmd = [
'ffmpeg', '-y', '-hwaccel', accel, '-i', input_path,
*video_base_opts, '-an', video_output
]
if faster:
video_cmd.extend(['-preset', 'ultrafast'])
await run_subprocess(video_cmd)
return audio_output, video_output
async def process_conversion(use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only):
try:
if use_youtube:
yt_uuid = str(uuid.uuid4())
yt_out = os.path.join(UPLOAD_FOLDER, yt_uuid + ".%(ext)s")
yt_cmd = ['yt-dlp', '-o', yt_out, '-f', 'b', youtube_url]
await run_subprocess(yt_cmd)
pattern = os.path.join(UPLOAD_FOLDER, yt_uuid + ".*")
files = glob.glob(pattern)
if not files:
return "Download failed.", None
input_path = files[0]
else:
if not video_file:
return "No video provided.", None
ext = os.path.splitext(video_file)[1]
input_path = os.path.join(UPLOAD_FOLDER, f"{uuid.uuid4()}{ext}")
shutil.copy2(video_file, input_path)
audio_out, video_out = await convert_video_task(input_path, downscale, faster, use_mp3)
if audio_only:
return audio_out, audio_out
return video_out or audio_out, video_out or audio_out
except Exception as e:
return str(e), None
def convert_video(*args):
return asyncio.run(process_conversion(*args))
# Gradio Interface
with gr.Blocks(theme=gr.themes.Default(primary_hue="rose")) as demo:
gr.Markdown("""
# π₯ **Low Quality Video Inator**
Welcome to your personal bad-video-making machine.
Upload a video or paste a YouTube URL below, then tweak the settings to ruin your video (on purpose).
""")
with gr.Group():
with gr.Row():
use_youtube = gr.Checkbox(label="π Use YouTube URL (experimental)", value=False)
youtube_url = gr.Textbox(label="YouTube URL", placeholder="Paste YouTube URL here")
video_file = gr.File(label="π Upload Video File")
gr.Markdown("### βοΈ **Conversion Settings**")
with gr.Group():
with gr.Row():
downscale = gr.Checkbox(label="Downscale Video to 144p", value=False)
faster = gr.Checkbox(label="Faster Video Compression", value=False)
with gr.Row():
use_mp3 = gr.Checkbox(label="Use MP3 Audio", value=False)
audio_only = gr.Checkbox(label="Audio Only", value=False)
convert_button = gr.Button("Convert Now", variant="primary")
gr.Markdown("### **Conversion Preview**")
video_preview = gr.Video(label="Preview Output (if applicable)")
gr.Markdown("### **Download Your Masterpiece**")
file_download = gr.File(label="Download Result")
convert_button.click(
convert_video,
inputs=[use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only],
outputs=[video_preview, file_download]
)
demo.launch() |