Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
-import PyPDF2
+import pikepdf
 import requests
 import io
 import tempfile
-import sys
 import os
 
 def compress_pdf(input_file, url, strength):
@@ -22,8 +21,7 @@ def compress_pdf(input_file, url, strength):
         else:
             if hasattr(input_file, 'name'):
                 # If input_file is a file path
-                with open(input_file.name, 'rb') as file:
-                    pdf_content = io.BytesIO(file.read())
+                pdf_content = input_file.name
                 initial_size = os.path.getsize(input_file.name)
             else:
                 # If input_file is file-like object
@@ -32,29 +30,19 @@ def compress_pdf(input_file, url, strength):
                 initial_size = pdf_content.tell()
                 pdf_content.seek(0)
 
-        reader = PyPDF2.PdfReader(pdf_content)
-        writer = PyPDF2.PdfWriter()
-
         if strength == "Low":
-            target_ratio = 0.75  # 25% compression
+            compress_level = 1
         elif strength == "Medium":
-            target_ratio = 0.50  # 50% compression
+            compress_level = 2
         else:  # High
-            target_ratio = 0.25  # 75% 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
+            compress_level = 3
 
-        # Write the compressed PDF to a temporary file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
-            writer.write(temp_file)
             temp_file_path = temp_file.name
 
+        pdf = pikepdf.Pdf.open(pdf_content)
+        pdf.save(temp_file_path, compress_streams=True, compress_images=True, optimize_images=True)
+
         # Check the compression ratio achieved
         compressed_size = os.path.getsize(temp_file_path)
         compression_ratio = compressed_size / initial_size
@@ -72,8 +60,6 @@ def compress_pdf(input_file, url, strength):
 # The rest of the code remains the same
 
 def process_and_compress(input_file, url, strength):
-    sys.setrecursionlimit(10000)
-    
     output_file, message = compress_pdf(input_file, url, strength)
     if output_file:
         return output_file, message
@@ -85,7 +71,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         input_file = gr.File(label="Upload PDF")
         url_input = gr.Textbox(label="Or enter PDF URL")
-    strength = gr.Radio(["Low", "Medium", "High"], label="Compression Strength", value="Medium", info="Low: ~25% compression, Medium: ~50% compression, High: ~75% compression")
+    strength = gr.Radio(["Low", "Medium", "High"], label="Compression Strength", value="Medium", info="Low: Minimal compression, Medium: Moderate compression, High: Maximum compression")
     compress_btn = gr.Button("Compress")
     output_file = gr.File(label="Download Compressed PDF")
     message = gr.Textbox(label="Message")