Update app.py

app.py

@@ -1,4 +1,3 @@
-#This was made with ChatGPT because my dumbass can't actually code
 import os
 import subprocess
 import uuid
@@ -35,15 +34,14 @@ def convert_video(use_youtube, youtube_url, video_file, downscale, faster, use_m
     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)
@@ -53,47 +51,51 @@ def convert_video(use_youtube, youtube_url, video_file, downscale, faster, use_m
         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.
+    # Handle optional downscale first
+    if downscale and not audio_only:
+        downscaled_path = os.path.join(UPLOAD_FOLDER, f"downscaled_{uuid.uuid4()}.mp4")
+        downscale_cmd = ['./ffmpeg', '-hwaccel', accel, '-y', '-i', input_path,
+                         '-vf', 'scale=144:-2', '-c:v', 'libx264', '-preset', 'ultrafast', '-crf', '40',
+                         '-an', downscaled_path]
+        try:
+            subprocess.run(downscale_cmd, check=True)
+        except subprocess.CalledProcessError as e:
+            return f"An error occurred during downscaling: {e}", None
+        input_path = downscaled_path  # Use this for next pass
+
+    # Now do the final encode
     ffmpeg_cmd = ['./ffmpeg', '-hwaccel', accel, '-y', '-i', input_path] + extra.split()
-    
+
     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)
-    
+    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.
@@ -106,6 +108,9 @@ with gr.Blocks() as demo:
         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)