Update app.py

app.py

@@ -5,7 +5,7 @@ import io
 import tempfile
 import os
 
-def compress_pdf(input_file, url, strength):
+def compress_pdf(input_file, url):
     if input_file is None and (url is None or url.strip() == ""):
         return None, "Please provide either a file or a URL."
     
@@ -35,12 +35,8 @@ def compress_pdf(input_file, url, strength):
 
         pdf = pikepdf.Pdf.open(pdf_content)
         
-        if strength == "Low":
-            compression_params = dict(compress_streams=True, object_stream_mode=pikepdf.ObjectStreamMode.disable)
-        elif strength == "Medium":
-            compression_params = dict(compress_streams=True, object_stream_mode=pikepdf.ObjectStreamMode.preserve)
-        else:  # High
-            compression_params = dict(compress_streams=True, object_stream_mode=pikepdf.ObjectStreamMode.generate)
+        # Always use the highest compression setting
+        compression_params = dict(compress_streams=True, object_stream_mode=pikepdf.ObjectStreamMode.generate)
 
         pdf.save(temp_file_path, **compression_params)
 
@@ -57,9 +53,9 @@ def compress_pdf(input_file, url, strength):
         return temp_file_path, f"PDF compressed successfully! Compression achieved: {compression_percentage:.2f}%"
     except Exception as e:
         return None, f"Error compressing PDF: {str(e)}"
-        
-def process_and_compress(input_file, url, strength):
-    output_file, message = compress_pdf(input_file, url, strength)
+
+def process_and_compress(input_file, url):
+    output_file, message = compress_pdf(input_file, url)
     if output_file:
         return output_file, message
     else:
@@ -70,14 +66,13 @@ 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: 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")
 
     compress_btn.click(
         process_and_compress,
-        inputs=[input_file, url_input, strength],
+        inputs=[input_file, url_input],
         outputs=[output_file, message]
     )