import gradio as gr
import tempfile
from pdf2image import convert_from_path
from PIL import Image
import img2pdf
import os

def downscale_pdf(pdf_file, dpi=150, max_width=1500):
    import shutil

    with tempfile.TemporaryDirectory() as tmpdir:
        # ステップ 1: PDF → 画像
        images = convert_from_path(pdf_file.name, dpi=dpi, output_folder=tmpdir)

        downscaled_paths = []
        for i, img in enumerate(images):
            w, h = img.size
            if w > max_width:
                new_h = int(h * max_width / w)
                img = img.resize((max_width, new_h), Image.LANCZOS)

            img_path = os.path.join(tmpdir, f"page_{i}.jpg")
            img.save(img_path, "JPEG", quality=85)
            downscaled_paths.append(img_path)

        # ステップ 2: 画像 → PDF
        tmp_output_path = os.path.join(tmpdir, "downscaled.pdf")
        with open(tmp_output_path, "wb") as f_out:
            f_out.write(img2pdf.convert(downscaled_paths))

    # ここでtmpdirは消えるので、ファイルを別の場所にコピー
    result_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
    with open(tmp_output_path, "rb") as src:
        shutil.copyfileobj(src, result_file)
    result_file.close()

    return result_file.name


with gr.Blocks() as demo:
    gr.Markdown("# PDF 解像度ダウンサイザー\nPDFのページ画像を縮小して容量を減らします。")
    with gr.Row():
        with gr.Column():
            pdf_input = gr.File(label="入力PDF", file_types=[".pdf"])
            dpi_input = gr.Slider(72, 300, value=150, step=1, label="変換DPI")
            width_input = gr.Slider(500, 3000, value=1500, step=50, label="最大幅(px)")
            convert_button = gr.Button("変換")

        with gr.Column():
            pdf_output = gr.File(label="出力PDF")

    convert_button.click(
        fn=downscale_pdf,
        inputs=[pdf_input, dpi_input, width_input],
        outputs=pdf_output
    )

demo.launch()