Update app.py

app.py

@@ -5,65 +5,41 @@ import img2pdf
 import tempfile
 import os
 
-def downscale_pdf(pdf_file, max_width=3000, input_dpi=150):
+def resize_pdf(pdf_file, width_percent, height_percent):
+    width_percent = float(width_percent)
+    height_percent = float(height_percent)
+    
     with tempfile.TemporaryDirectory() as tmpdir:
-        # 1. dpiを下げて読み込み（元が300なら150に半減）
-        images = convert_from_path(pdf_file.name, dpi=input_dpi, fmt='jpeg', output_folder=tmpdir)
-
-        downscaled_images = []
+        # PDF → 画像（全ページ）
+        images = convert_from_path(pdf_file.name, dpi=200, output_folder=tmpdir)
+        
+        resized_images = []
         for i, img in enumerate(images):
-            w, h = img.size
-            print(f"Original Page {i+1}: {w}x{h} = {w * h} px")
-
-            # サイズ制限
-            if w > max_width:
-                ratio = max_width / w
-                new_w = int(w * ratio)
-                new_h = int(h * ratio)
-                img = img.resize((new_w, new_h), Image.LANCZOS)
-                print(f"Downscaled Page {i+1}: {new_w}x{new_h} = {new_w * new_h} px")
-            else:
-                new_w, new_h = w, h
-                print(f"Page {i+1}: サイズ変更なし")
-
-            tmp_image_path = os.path.join(tmpdir, f"page_{i+1}.jpg")
-            img.save(tmp_image_path, "JPEG", quality=90)
-            downscaled_images.append((tmp_image_path, (new_w, new_h)))
-
-        # 2. img2pdfのdpiを画像サイズから計算する
-        #    dpi = (元画像のdpi) * (縮小率)
-        #    今回は画像のピクセル数が縮小されているので、それに対応するdpiを設定する
-        #    ここでは単純に input_dpi をそのまま使うことも可
-
-        pdf_bytes = b""
-        layout_fun = img2pdf.get_layout_fun(pagesize=None)  # デフォルトページサイズ
-
-        with open(os.path.join(tmpdir, "downscaled.pdf"), "wb") as f:
-            f.write(
-                img2pdf.convert(
-                    [path for path, size in downscaled_images],
-                    dpi=input_dpi,
-                )
-            )
-
-        final_pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-        with open(os.path.join(tmpdir, "downscaled.pdf"), "rb") as src, open(final_pdf_file.name, "wb") as dst:
-            dst.write(src.read())
-
-        return final_pdf_file.name
-
-
+            # リサイズ幅・高さ計算
+            new_width = int(img.width * width_percent / 100)
+            new_height = int(img.height * height_percent / 100)
+            resized_img = img.resize((new_width, new_height), Image.LANCZOS)
+            
+            # 一時保存
+            img_path = os.path.join(tmpdir, f"page_{i}.png")
+            resized_img.save(img_path, format='PNG')
+            resized_images.append(img_path)
+        
+        # 画像 → PDF
+        output_pdf_path = os.path.join(tmpdir, "resized_output.pdf")
+        with open(output_pdf_path, "wb") as f_out:
+            f_out.write(img2pdf.convert(resized_images))
+        
+        return output_pdf_path
 
 with gr.Blocks() as demo:
-    gr.Markdown("## PDFダウンサイザー for Audiveris")
-    gr.Markdown(
-        "PDFをアップロードすると、ページ画像を縮小してAudiveris向けに最適化したPDFを生成します。"
-    )
-    with gr.Row():
-        pdf_input = gr.File(label="PDFファイルをアップロード")
-        pdf_output = gr.File(label="変換後PDF")
-
-    convert_button = gr.Button("変換してダウンロード")
-    convert_button.click(fn=downscale_pdf, inputs=pdf_input, outputs=pdf_output)
+    gr.Markdown("# PDFサイズを％でリサイズ")
+    pdf_input = gr.File(label="PDFファイルを選択")
+    width_input = gr.Number(label="幅の倍率(%)", value=100)
+    height_input = gr.Number(label="高さの倍率(%)", value=100)
+    output_pdf = gr.File(label="リサイズ後のPDF")
+
+    btn = gr.Button("リサイズ実行")
+    btn.click(resize_pdf, inputs=[pdf_input, width_input, height_input], outputs=output_pdf)
 
 demo.launch()