Update app.py

app.py

@@ -31,17 +31,17 @@ def compress_pdf(input_file, url, strength):
                 pdf_content.seek(0)
 
         if strength == "Low":
-            compress_level = 1
+            compress_level = pikepdf.compress.CompressSettings(level=1)
         elif strength == "Medium":
-            compress_level = 2
+            compress_level = pikepdf.compress.CompressSettings(level=2)
         else:  # High
-            compress_level = 3
+            compress_level = pikepdf.compress.CompressSettings(level=3)
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as 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)
+        pdf.save(temp_file_path, compress_streams=True, compress=compress_level)
 
         # Check the compression ratio achieved
         compressed_size = os.path.getsize(temp_file_path)
@@ -57,30 +57,4 @@ def compress_pdf(input_file, url, strength):
     except Exception as e:
         return None, f"Error compressing PDF: {str(e)}"
 
-# The rest of the code remains the same
-
-def process_and_compress(input_file, url, strength):
-    output_file, message = compress_pdf(input_file, url, strength)
-    if output_file:
-        return output_file, message
-    else:
-        return None, message
-
-with gr.Blocks() as demo:
-    gr.Markdown("# PDF Compressor")
-    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],
-        outputs=[output_file, message]
-    )
-
-if __name__ == "__main__":
-    demo.launch(share=True)
+# The rest of the code remains the same