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()