Update app.py

app.py

@@ -57,4 +57,28 @@ 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)