Update app.py

app.py

@@ -42,15 +42,13 @@ def compress_pdf(input_file, url, strength):
         else:  # High
             target_ratio = 0.25  # 75% compression
 
-        # First pass: apply basic compression
+        # Apply compression to each page
         for page in reader.pages:
             page.compress_content_streams()  # Apply content stream compression
             writer.add_page(page)
 
         # Set compression parameters
         writer.compress = True
-        writer._compress_streams = True
-        writer._compress_pages = True
 
         # Write the compressed PDF to a temporary file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
@@ -59,30 +57,13 @@ def compress_pdf(input_file, url, strength):
 
         # Check the compression ratio achieved
         compressed_size = os.path.getsize(temp_file_path)
-        current_ratio = compressed_size / initial_size
+        compression_ratio = compressed_size / initial_size
+        compression_percentage = (1 - compression_ratio) * 100
 
-        # If we haven't reached the target ratio, apply additional compression
-        if current_ratio > target_ratio:
-            reader = PyPDF2.PdfReader(temp_file_path)
-            writer = PyPDF2.PdfWriter()
-
-            for page in reader.pages:
-                # Apply more aggressive compression
-                page.compress_content_streams()
-                writer.add_page(page)
-
-            writer.compress = True
-            writer._compress_streams = True
-            writer._compress_pages = True
-
-            # Overwrite the temporary file with the more compressed version
-            with open(temp_file_path, 'wb') as temp_file:
-                writer.write(temp_file)
-
-        # Final compression ratio
-        final_size = os.path.getsize(temp_file_path)
-        final_ratio = final_size / initial_size
-        compression_percentage = (1 - final_ratio) * 100
+        # If compression increased file size or didn't meet minimum threshold, return original file
+        if compression_ratio >= 1 or compression_percentage < 5:
+            os.remove(temp_file_path)
+            return input_file, f"Unable to compress the PDF effectively. Original file returned. (Attempted compression: {compression_percentage:.2f}%)"
 
         return temp_file_path, f"PDF compressed successfully! Compression achieved: {compression_percentage:.2f}%"
     except Exception as e: