app.py

import gradio as gr
import PyPDF2
import requests
import io
import os
import tempfile

def compress_pdf(input_file, url, quality):
    if input_file is None and url is None:
        return None, "Please provide either a file or a URL."
    
    if input_file is not None and url is not None:
        return None, "Please provide either a file or a URL, not both."
    
    if url:
        try:
            response = requests.get(url)
            response.raise_for_status()
            pdf_content = io.BytesIO(response.content)
        except requests.RequestException as e:
            return None, f"Error downloading PDF: {str(e)}"
    else:
        pdf_content = input_file

    try:
        reader = PyPDF2.PdfReader(pdf_content)
        writer = PyPDF2.PdfWriter()

        for page in reader.pages:
            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)

        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file:
            writer.write(temp_file)
            temp_file_path = temp_file.name

        return temp_file_path, "PDF compressed successfully!"
    except Exception as e:
        return None, f"Error compressing PDF: {str(e)}"

def process_and_compress(input_file, url, quality):
    output_file, message = compress_pdf(input_file, url, quality)
    if output_file:
        return gr.File.update(value=output_file, visible=True), message
    else:
        return gr.File.update(visible=False), 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")
    quality = gr.Radio(["High", "Medium", "Low"], label="Compression Quality", value="Medium")
    compress_btn = gr.Button("Compress")
    output_file = gr.File(label="Download Compressed PDF", visible=False)
    message = gr.Textbox(label="Message")

    compress_btn.click(
        process_and_compress,
        inputs=[input_file, url_input, quality],
        outputs=[output_file, message]
    )

if __name__ == "__main__":
    demo.launch()