Update app.py

app.py

@@ -1,53 +1,39 @@
 import gradio as gr
 import subprocess
 import os
+import tempfile
 
-def rasterize_and_report(input_pdf):
-    output_pdf = "/tmp/output.pdf"
+def rasterize_with_imagemagick(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]
+        res1 = subprocess.run(cmd1, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        if res1.returncode != 0:
+            return f"ImageMagick変換エラー: {res1.stderr.decode()}", None, None
 
-    # Ghostscriptコマンドで強制ラスタライズ
-    cmd = [
-        "gs",
-        "-o", output_pdf,
-        "-sDEVICE=pdfwrite",
-        "-dColorImageResolution=150",
-        "-dGrayImageResolution=150",
-        "-dMonoImageResolution=150",
-        "-dFILTERVECTOR",
-        "-dCompatibilityLevel=1.4",
-        "-dNOPAUSE",
-        "-dBATCH",
-        "-dQUIET",
-        input_pdf
-    ]
-    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    if result.returncode != 0:
-        return f"Ghostscriptエラー: {result.stderr.decode()}", None, None
+        # PNG→PDFにまとめる
+        output_pdf = os.path.join(tmpdir, "output_rasterized.pdf")
+        cmd2 = ["convert", os.path.join(tmpdir, "page_*.png"), 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
+        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 f"変換完了 / ファイルサイズ: {size_str}", output_pdf, size_str
+        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
 
 demo = gr.Interface(
-    fn=rasterize_and_report,
+    fn=rasterize_with_imagemagick,
     inputs=gr.File(file_types=[".pdf"]),
-    outputs=[
-        gr.Text(label="結果メッセージ"),
-        gr.File(label="変換後PDF"),
-        gr.Text(label="変換後ファイルサイズ")
-    ],
-    title="PDFラスタライズ（Ghostscript）",
-    description=(
-        "SVGなどのベクターPDFを強制的に画像化（ラスタライズ）してPDF出力します。"
-        "Audiveris用にサイズを落とすのに便利です。"
-    )
+    outputs=[gr.Text(), gr.File(), gr.Text()],
+    title="PDFをImageMagickでラスタライズ",
+    description="PDFを画像に変換して再PDF化し、ベクターを除去します。"
 )
 
 demo.launch()