File size: 5,069 Bytes
0383ad0 d83b57c 76e6c8b d83b57c 1dc23c5 d83b57c 76e6c8b 1dc23c5 76e6c8b d83b57c 76e6c8b 1dc23c5 d83b57c 1dc23c5 d83b57c 76e6c8b d83b57c 1dc23c5 d83b57c c1e1ec2 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 |
#This was made with ChatGPT because my dumbass can't actually code
import os
import subprocess
import uuid
import glob
import shutil
import gradio as gr
# Set default acceleration and extra encoding options.
accel = 'auto'
extra = '-crf 63 -c:v libx264' # default extra settings
# Fixed upload and conversion folders.
UPLOAD_FOLDER = 'uploads'
CONVERTED_FOLDER = 'converted'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(CONVERTED_FOLDER, exist_ok=True)
# Helper function to configure audio options.
def configure_audio(cmd_list, use_mp3):
if use_mp3:
cmd_list.extend(['-c:a', 'libmp3lame', '-b:a', '8k', '-ar', '24k', '-ac', '1'])
else:
cmd_list.extend(['-c:a', 'aac', '-b:a', '1k', '-ar', '8k', '-ac', '1'])
def convert_video(use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only):
"""
Converts a video either from a YouTube URL or an uploaded file.
Returns a tuple (preview_file, download_file) with the output path.
"""
# Determine output filename based on whether only audio is desired.
if audio_only:
output_filename = f"{uuid.uuid4()}.mp3" if use_mp3 else f"{uuid.uuid4()}.aac"
else:
output_filename = f"{uuid.uuid4()}.mp4"
output_path = os.path.join(CONVERTED_FOLDER, output_filename)
if use_youtube:
if not youtube_url:
return "Error: YouTube URL is required.", None
# Determine quality: if downscale is ticked use low quality; otherwise, use "-f b" for best quality.
quality = 'worstaudio' if audio_only and downscale else (
'worstvideo+worstaudio' if downscale else 'b')
yt_uuid = str(uuid.uuid4())
yt_input_filename = yt_uuid + ".%(ext)s"
yt_input_path = os.path.join(UPLOAD_FOLDER, yt_input_filename)
yt_dlp_cmd = ['yt-dlp', '-o', yt_input_path, '-f', quality, youtube_url]
try:
subprocess.run(yt_dlp_cmd, check=True)
except subprocess.CalledProcessError as e:
return f"An error occurred during YouTube download: {e}", None
# Locate the downloaded file (yt-dlp replaces %(ext)s with the actual extension).
pattern = os.path.join(UPLOAD_FOLDER, yt_uuid + ".*")
downloaded_files = glob.glob(pattern)
if not downloaded_files:
return "Failed to download YouTube video.", None
input_path = downloaded_files[0]
else:
# File upload branch.
if not video_file:
return "Error: No video file provided.", None
# Check if the provided file path exists.
if os.path.exists(video_file):
ext = os.path.splitext(video_file)[1]
input_filename = f"{uuid.uuid4()}{ext}"
input_path = os.path.join(UPLOAD_FOLDER, input_filename)
# Copy the file to our uploads folder.
shutil.copy2(video_file, input_path)
else:
# If the file doesn't exist (perhaps already in a persistent location), reuse it.
input_path = video_file
# Build the FFmpeg command.
ffmpeg_cmd = ['ffmpeg', '-hwaccel', accel, '-y', '-i', input_path] + extra.split()
if downscale:
ffmpeg_cmd.extend(['-vf', 'scale=144:-2'])
if faster:
ffmpeg_cmd.extend(['-preset', 'ultrafast'])
if audio_only:
ffmpeg_cmd.extend(['-vn'])
configure_audio(ffmpeg_cmd, use_mp3)
else:
configure_audio(ffmpeg_cmd, use_mp3)
ffmpeg_cmd.append(output_path)
try:
subprocess.run(ffmpeg_cmd, check=True)
except subprocess.CalledProcessError as e:
return f"An error occurred during conversion: {e}", None
# Return the same output for preview and download.
return output_path, output_path
# Create the Gradio interface.
with gr.Blocks() as demo:
gr.Markdown("## Low Quality Video Inator")
gr.Markdown("Upload a video or use a YouTube URL and adjust the options below.")
with gr.Row():
use_youtube = gr.Checkbox(label="Use YouTube URL (almost never works)", value=False)
youtube_url = gr.Textbox(label="YouTube URL", placeholder="Enter YouTube URL here")
video_file = gr.File(label="Upload Video")
with gr.Row():
downscale = gr.Checkbox(label="Downscale Video (144p)", value=False)
faster = gr.Checkbox(label="Faster Conversion (more pixelated)", value=False)
with gr.Row():
use_mp3 = gr.Checkbox(label="Use MP3 (video audio compression)", value=False)
audio_only = gr.Checkbox(label="Only Audio (no video)", value=False)
convert_button = gr.Button("Convert Video")
gr.Markdown("### Preview Converted Video")
video_preview = gr.Video(label="Converted Video")
gr.Markdown("### Download Converted Video")
file_download = gr.File(label="Download Video")
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() |