Update app.py

app.py

@@ -6,6 +6,15 @@ from pathlib import Path
 import shutil
 import json
 
+def adjust_dimensions(width, height, codec):
+    """Adjust dimensions to be compatible with the codec's requirements"""
+    if codec in ["libx265", "libvpx-vp9"]:
+        # These codecs require dimensions to be multiples of 8
+        new_width = (width + 7) // 8 * 8
+        new_height = (height + 7) // 8 * 8
+        return new_width, new_height
+    return width, height
+
 def compress_video(
     input_video,
     resolution,
@@ -38,6 +47,23 @@ def compress_video(
         output_ext = f".{output_format.lower()}"
         output_path = input_path.replace(Path(input_video).suffix, f"_compressed{output_ext}")
         
+        # Get video info to check dimensions
+        info_cmd = ["ffprobe", "-v", "quiet", "-print_format", "json", "-show_format", "-show_streams", input_path]
+        probe_result = subprocess.run(info_cmd, capture_output=True, text=True)
+        video_info = json.loads(probe_result.stdout)
+        
+        # Extract video stream info
+        video_stream = next((s for s in video_info["streams"] if s["codec_type"] == "video"), None)
+        if video_stream:
+            width = video_stream.get("width", 0)
+            height = video_stream.get("height", 0)
+            
+            # Adjust dimensions if needed
+            new_width, new_height = adjust_dimensions(width, height, video_codec)
+            needs_resize = (new_width != width or new_height != height)
+        else:
+            needs_resize = False
+        
         # Build FFmpeg command
         cmd = ["ffmpeg"]
         
@@ -65,8 +91,12 @@ def compress_video(
             "1440p": "scale=-1:1440",
             "4K": "scale=-1:2160"
         }
+        
         if resolution != "Original":
             vf_filters.append(scale_map[resolution])
+        elif needs_resize:
+            # If we need to adjust dimensions for codec compatibility
+            vf_filters.append(f"scale={new_width}:{new_height}")
         
         # Rotation
         if rotate != "None":
@@ -98,6 +128,11 @@ def compress_video(
             crf_map = {"Low": 32, "Medium": 28, "High": 24, "Ultra High": 20}
             crf = crf_map[quality]
             cmd.extend(["-c:v", video_codec, "-crf", str(crf)])
+        elif video_codec == "libvpx-vp9":
+            # Map quality to CRF for VP9
+            crf_map = {"Low": 31, "Medium": 26, "High": 21, "Ultra High": 15}
+            crf = crf_map[quality]
+            cmd.extend(["-c:v", video_codec, "-crf", str(crf), "-b:v", "0"])
         else:
             cmd.extend(["-c:v", video_codec])
         
@@ -123,13 +158,16 @@ def compress_video(
             cmd.extend(["-pass", "2"])
         
         # Audio settings
-        cmd.extend(["-c:a", audio_codec])
-        
-        if audio_bitrate != "Auto":
-            cmd.extend(["-b:a", audio_bitrate])
-        
-        if audio_sample_rate != "Original":
-            cmd.extend(["-ar", audio_sample_rate])
+        if audio_codec == "copy":
+            cmd.extend(["-c:a", "copy"])
+        else:
+            cmd.extend(["-c:a", audio_codec])
+            
+            if audio_bitrate != "Auto":
+                cmd.extend(["-b:a", audio_bitrate])
+            
+            if audio_sample_rate != "Original":
+                cmd.extend(["-ar", audio_sample_rate])
         
         # Remove metadata
         if remove_metadata:
@@ -204,7 +242,17 @@ def compress_video(
         return output_path, info_text
         
     except subprocess.CalledProcessError as e:
-        return None, f"❌ Compression failed: {e.stderr}"
+        error_message = e.stderr
+        
+        # Check for specific error conditions
+        if "Picture width must be an integer multiple" in error_message:
+            error_message = "The video dimensions are not compatible with the selected codec. The app will try to adjust the dimensions automatically."
+        elif "Cannot open libx265 encoder" in error_message:
+            error_message = "Error with H.265 encoding. Try using H.264 instead."
+        elif "Unsupported codec" in error_message:
+            error_message = "The selected codec is not supported. Try a different codec."
+        
+        return None, f"❌ Compression failed: {error_message}"
     except Exception as e:
         return None, f"❌ An error occurred: {str(e)}"
     finally: