Update app.py

app.py

@@ -3,37 +3,38 @@ import subprocess
 import os
 import tempfile
 
-def rasterize_with_imagemagick(input_pdf):
+def rasterize_with_pdftoppm(input_pdf):
     with tempfile.TemporaryDirectory() as tmpdir:
-        # PDF→PNG（72dpi）
-        png_pattern = os.path.join(tmpdir, "page_%03d.png")
-        cmd1 = ["convert", "-density", "72", input_pdf, png_pattern]
+        output_prefix = os.path.join(tmpdir, "page")
+        cmd1 = ["pdftoppm", "-png", "-r", "150", input_pdf, output_prefix]
         res1 = subprocess.run(cmd1, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         if res1.returncode != 0:
-            return f"ImageMagick変換エラー: {res1.stderr.decode()}", None, None
+            return f"pdftoppm変換エラー: {res1.stderr.decode()}", None, None
 
-        # PNG→PDFにまとめる
+        # PNGファイル一覧取得
+        png_files = sorted([os.path.join(tmpdir, f) for f in os.listdir(tmpdir) if f.endswith(".png")])
+        if not png_files:
+            return "PNG変換に失敗しました", None, None
+
+        # PNG→PDF変換（ImageMagickのconvertが使えなければpypdfで合成も可能）
         output_pdf = os.path.join(tmpdir, "output_rasterized.pdf")
-        cmd2 = ["convert", os.path.join(tmpdir, "page_*.png"), output_pdf]
+        cmd2 = ["convert"] + png_files + [output_pdf]
         res2 = subprocess.run(cmd2, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         if res2.returncode != 0:
             return f"PDF作成エラー: {res2.stderr.decode()}", None, None
 
-        if not os.path.exists(output_pdf):
-            return "PDF生成に失敗しました", None, None
-
         size_bytes = os.path.getsize(output_pdf)
         size_kb = size_bytes / 1024
         size_str = f"{size_bytes} bytes ({size_kb:.1f} KB)"
 
-        return "ImageMagickでラスタライズ完了", output_pdf, size_str
+        return "pdftoppmでラスタライズ完了", output_pdf, size_str
 
 demo = gr.Interface(
-    fn=rasterize_with_imagemagick,
+    fn=rasterize_with_pdftoppm,
     inputs=gr.File(file_types=[".pdf"]),
     outputs=[gr.Text(), gr.File(), gr.Text()],
-    title="PDFをImageMagickでラスタライズ",
-    description="PDFを画像に変換して再PDF化し、ベクターを除去します。"
+    title="PDFをpdftoppmでラスタライズ",
+    description="pdftoppmでPDFをPNGに変換し再度PDF化します。"
 )
 
 demo.launch()