Update app.py

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 os.system("chmod +x fdkaac")  # Ensure fdkaac is executable
 
 accel = 'auto'
-video_base_opts = ['-crf', '63', '-c:v', 'libx264', '-tune', 'zerolatency', '-movflags', '+faststart']
+video_base_opts = ['-crf', '63', '-c:v', 'libx264', '-tune', 'zerolatency']
 
 UPLOAD_FOLDER = 'uploads'
 CONVERTED_FOLDER = 'converted'
@@ -34,7 +34,7 @@ async def run_subprocess(cmd, use_fdkaac=False):
     print(f"[DEBUG] Command succeeded:\n{stdout.decode()}\n")
     return stdout.decode(), stderr.decode()
 
-async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only):
+async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate):
     if use_mp3:
         output_audio = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp3")
         ffmpeg_audio_cmd = [
@@ -50,11 +50,16 @@ async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only)
             # Video+MP3, no splitting; encode video with MP3 audio embedded
             output_video = os.path.join(CONVERTED_FOLDER, f"{uuid.uuid4()}.mp4")
             ffmpeg_video_cmd = [
-                'ffmpeg', '-y', '-hwaccel', accel, '-i', input_path,
-                *video_base_opts
+                'ffmpeg', '-y', '-hwaccel', accel, '-i', input_path
             ]
-            if faster:
-                ffmpeg_video_cmd.extend(['-preset', 'ultrafast'])
+
+            if custom_bitrate:
+                ffmpeg_video_cmd += ['-b:v', f"{int(video_bitrate)}k"]
+            else:
+                ffmpeg_video_cmd += video_base_opts
+                if faster:
+                    ffmpeg_video_cmd.extend(['-preset', 'ultrafast'])
+
             ffmpeg_video_cmd += [
                 '-c:a', 'libmp3lame', '-b:a', '8k', '-ar', '24000', '-ac', '1',
                 output_video
@@ -80,11 +85,17 @@ async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only)
 
     # Compress video without audio
     video_cmd = [
-        'ffmpeg', '-y', '-hwaccel', accel, '-i', input_path,
-        *video_base_opts, '-an', video_output
+        'ffmpeg', '-y', '-hwaccel', accel, '-i', input_path
     ]
-    if faster:
-        video_cmd.extend(['-preset', 'ultrafast'])
+
+    if custom_bitrate:
+        video_cmd += ['-b:v', f"{int(video_bitrate)}k"]
+    else:
+        video_cmd += video_base_opts
+        if faster:
+            video_cmd.extend(['-preset', 'ultrafast'])
+
+    video_cmd += ['-an', video_output]
     await run_subprocess(video_cmd)
 
     if audio_only:
@@ -105,7 +116,7 @@ async def convert_video_task(input_path, downscale, faster, use_mp3, audio_only)
 
     return None, merged_output
 
-async def process_conversion(use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only):
+async def process_conversion(use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate):
     try:
         if use_youtube:
             if not youtube_url:
@@ -127,7 +138,7 @@ async def process_conversion(use_youtube, youtube_url, video_file, downscale, fa
             shutil.copy2(video_file.name, input_path)
 
         audio_out, video_out = await convert_video_task(
-            input_path, downscale, faster, use_mp3, audio_only
+            input_path, downscale, faster, use_mp3, audio_only, custom_bitrate, video_bitrate
         )
 
         if audio_only:
@@ -139,7 +150,7 @@ async def process_conversion(use_youtube, youtube_url, video_file, downscale, fa
 def convert_video(*args):
     return asyncio.run(process_conversion(*args))
 
-# Gradio Interface (unchanged, still pretty)
+# Gradio Interface (Now With Custom Bitrate)
 with gr.Blocks(theme=gr.themes.Default(primary_hue="rose")) as demo:
     gr.Markdown("""
     # 🎥 **Low Quality Video Inator**
@@ -162,6 +173,16 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="rose")) as demo:
         with gr.Row():
             use_mp3 = gr.Checkbox(label="Use MP3 Audio", value=False)
             audio_only = gr.Checkbox(label="Audio Only", value=False)
+        with gr.Row():
+            custom_bitrate = gr.Checkbox(label="Custom Video Bitrate", value=False)
+            video_bitrate = gr.Number(label="Bitrate (kbps)", visible=False)
+
+        # Show/hide bitrate field
+        custom_bitrate.change(
+            lambda checked: gr.update(visible=checked),
+            inputs=[custom_bitrate],
+            outputs=[video_bitrate]
+        )
 
     convert_button = gr.Button("Convert Now", variant="primary")
 
@@ -173,7 +194,11 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="rose")) as demo:
 
     convert_button.click(
         convert_video,
-        inputs=[use_youtube, youtube_url, video_file, downscale, faster, use_mp3, audio_only],
+        inputs=[
+            use_youtube, youtube_url, video_file,
+            downscale, faster, use_mp3, audio_only,
+            custom_bitrate, video_bitrate
+        ],
         outputs=[video_preview, file_download]
     )