Update app.py

app.py

@@ -2,11 +2,10 @@ import gradio as gr
 import PyPDF2
 import requests
 import io
-import os
 import tempfile
 import sys
 
-def compress_pdf(input_file, url, quality):
+def compress_pdf(input_file, url, strength):
     if input_file is None and (url is None or url.strip() == ""):
         return None, "Please provide either a file or a URL."
     
@@ -29,15 +28,15 @@ def compress_pdf(input_file, url, quality):
 
         for i, page in enumerate(reader.pages):
             try:
-                if quality == "High":
-                    page.compress_content_streams()  # This is the lowest compression
-                elif quality == "Medium":
-                    page.compress_content_streams(level=2)
-                else:  # Low quality
-                    page.compress_content_streams(level=3)
-                writer.add_page(page)
-            except RecursionError:
-                return None, f"Error: PDF structure too complex. Compression stopped at page {i+1}."
+                if strength == "High":
+                    page.compress_content_streams()  # Maximum compression
+                elif strength == "Medium":
+                    page.compress_content_streams()  # Medium compression (same as high for now)
+                else:  # Low strength
+                    writer.add_page(page)  # No compression
+                
+                if strength != "Low":
+                    writer.add_page(page)
             except Exception as e:
                 return None, f"Error compressing page {i+1}: {str(e)}"
 
@@ -49,11 +48,10 @@ def compress_pdf(input_file, url, quality):
     except Exception as e:
         return None, f"Error compressing PDF: {str(e)}"
 
-def process_and_compress(input_file, url, quality):
-    # Increase recursion limit
+def process_and_compress(input_file, url, strength):
     sys.setrecursionlimit(10000)
     
-    output_file, message = compress_pdf(input_file, url, quality)
+    output_file, message = compress_pdf(input_file, url, strength)
     if output_file:
         return output_file, message
     else:
@@ -64,14 +62,14 @@ with gr.Blocks() as demo:
     with gr.Row():
         input_file = gr.File(label="Upload PDF")
         url_input = gr.Textbox(label="Or enter PDF URL")
-    quality = gr.Radio(["High", "Medium", "Low"], label="Compression Quality", value="Medium")
+    strength = gr.Radio(["High", "Medium", "Low"], label="Compression Strength", value="Medium", info="High: Maximum compression, Low: Least compression")
     compress_btn = gr.Button("Compress")
     output_file = gr.File(label="Download Compressed PDF")
     message = gr.Textbox(label="Message")
 
     compress_btn.click(
         process_and_compress,
-        inputs=[input_file, url_input, quality],
+        inputs=[input_file, url_input, strength],
         outputs=[output_file, message]
     )