Update app.py

app.py

@@ -1,50 +1,42 @@
 # verify_highlight_annotation.py
-# 実行: python verify_highlight_annotation.py
-# 出力: output_highlighted.pdf (カレントディレクトリ)
-# 確認: Adobe Acrobat等で開き、ハイライトが「注釈」として認識されるか確認
+# Hugging Face Spaces上で実行する検証用アプリ
+# フォントファイル NotoSansJP-Regular.ttf を同じフォルダに配置すること
 
 import io
-import os
+import gradio as gr
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas as rl_canvas
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 import fitz  # PyMuPDF
+import tempfile
+import os
 
-# =====================
-# 設定
-# =====================
 FONT_NAME = "NotoSansJP"
-FONT_FILE = "NotoSansJP-Regular.ttf"  # カレントディレクトリに配置すること
-OUTPUT_FILE = "output_highlighted.pdf"
+FONT_FILE = "NotoSansJP-Regular.ttf"
 
-# ハイライト色 (RGB 0.0-1.0)
 COLORS = {
-    "color1": (1.0, 0.6, 0.6),   # 赤系
-    "color2": (0.6, 0.8, 1.0),   # 青系
-    "mono":   (0.5, 0.5, 0.5),   # グレー（モノクロ代替）
+    "color1": (1.0, 0.6, 0.6),
+    "color2": (0.6, 0.8, 1.0),
+    "mono":   (0.5, 0.5, 0.5),
 }
 
 # =====================
 # Step1: reportlabでベースPDF生成
 # =====================
-def build_base_pdf(font_file: str) -> bytes:
-    """テスト用テキストを含むPDFをreportlabで生成してbytesで返す"""
-
-    pdfmetrics.registerFont(TTFont(FONT_NAME, font_file))
-
+def build_base_pdf() -> bytes:
+    pdfmetrics.registerFont(TTFont(FONT_NAME, FONT_FILE))
     buffer = io.BytesIO()
     PAGE_W, PAGE_H = A4
     MARGIN = 20
     c = rl_canvas.Canvas(buffer, pagesize=A4)
     c.setFont(FONT_NAME, 12)
 
-    # テスト文字列（差分あり想定）
     lines = [
-        "吾輩は猫である。名前はまだない。",       # 行1: 差分なし想定
-        "吾輩は犬である。名前はもうある。",        # 行2: 差分あり想定（「犬」「もうある」）
-        "abcXYZdef",                               # 行3: 差分あり（「XYZ」）
-        "テスト　スペース　あり",                  # 行4: 差分なし想定
+        "吾輩は猫である。名前はまだない。",
+        "吾輩は犬である。名前はもうある。",
+        "abcXYZdef",
+        "テスト　スペース　あり",
     ]
 
     y = PAGE_H - MARGIN - 20
@@ -55,120 +47,72 @@ def build_base_pdf(font_file: str) -> bytes:
     c.save()
     return buffer.getvalue()
 
-
 # =====================
-# Step2: PyMuPDFでHighlight Annotation付与
+# Step2: テキスト検索によるHighlight Annotation
 # =====================
 def add_highlight_annotations(pdf_bytes: bytes, color_mode: str = "color1") -> bytes:
-    """
-    PyMuPDFを使い、指定テキストにHighlight Annotationを付与する。
-    テキスト検索 → quads取得 → add_highlight_annot の流れを検証。
-    """
     hl_color = COLORS.get(color_mode, COLORS["color1"])
-
     doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+    page = doc[0]
 
-    # 検証用: ハイライトしたいテキストのリスト
-    highlight_targets = [
-        "犬",
-        "もうある",
-        "XYZ",
-    ]
-
-    page = doc[0]  # 1ページ目
+    highlight_targets = ["犬", "もうある", "XYZ"]
 
     for target in highlight_targets:
-        # テキスト検索でquadsを取得
         areas = page.search_for(target, quads=True)
         if not areas:
-            print(f"[WARNING] テキストが見つかりません: '{target}'")
             continue
-
         for quad in areas:
             annot = page.add_highlight_annot(quad)
             annot.set_colors(stroke=hl_color)
             annot.update()
-            print(f"[OK] ハイライト注釈追加: '{target}' at {quad}")
 
     result = io.BytesIO()
     doc.save(result)
     doc.close()
     return result.getvalue()
 
-
 # =====================
-# Step3: 文字座標から直接Annotationを付与する方式の検証
+# メイン処理
 # =====================
-def add_highlight_by_position(pdf_bytes: bytes, color_mode: str = "color1") -> bytes:
-    """
-    reportlabで描画した文字位置を座標で指定してHighlight Annotationを付与。
-    (テキスト検索不要の方式 - 本番実装候補)
-    reportlabのy座標(下原点) → PyMuPDFのy座標(上原点)変換を検証。
-    """
-    hl_color = COLORS.get(color_mode, COLORS["color1"])
-    PAGE_W, PAGE_H = A4  # (595.27, 841.89)
-    MARGIN = 20
-    FONT_SIZE = 12
-    LINE_HEIGHT = 20
+def run_verify():
+    log = []
+
+    try:
+        # Step1
+        base_pdf = build_base_pdf()
+        log.append(f"[OK] Step1: ベースPDF生成完了 ({len(base_pdf)} bytes)")
+    except Exception as e:
+        return f"[ERROR] Step1失敗: {e}", None
+
+    try:
+        # Step2
+        highlighted_pdf = add_highlight_annotations(base_pdf, color_mode="color1")
+        log.append(f"[OK] Step2: Highlight Annotation付与完了 ({len(highlighted_pdf)} bytes)")
+    except Exception as e:
+        return "\n".join(log) + f"\n[ERROR] Step2失敗: {e}", None
+
+    try:
+        # 一時ファイルに保存してGradioのFileで返す
+        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+        tmp.write(highlighted_pdf)
+        tmp.close()
+        log.append(f"[OK] PDF出力完了: {tmp.name}")
+    except Exception as e:
+        return "\n".join(log) + f"\n[ERROR] ファイル出力失敗: {e}", None
+
+    return "\n".join(log), tmp.name
 
-    doc = fitz.open(stream=pdf_bytes, filetype="pdf")
-    page = doc[0]
-
-    # reportlabで描画した行2の「犬」の座標を手動計算
-    # reportlab: y = PAGE_H - MARGIN - 20 - LINE_HEIGHT * 1 = 841.89 - 20 - 20 - 20 = 781.89
-    # PyMuPDF: y_mupdf = PAGE_H - y_rl = 841.89 - 781.89 = 60.0  ← ベースライン
-    # Highlight用rect: x0, y0(上端), x1, y1(下端)
-
-    # 行インデックス1（0始まり）= 「吾輩は犬である...」行
-    line_idx = 1
-    y_rl = PAGE_H - MARGIN - 20 - LINE_HEIGHT * line_idx  # reportlab y（ベースライン）
-    y_mupdf_base = PAGE_H - y_rl                           # PyMuPDF y（上原点換算）
-
-    # 「犬」は「吾輩は」(3文字)の後 → x方向はフォント幅が必要だが近似で指定
-    # 実際はreportlab側でstringWidthを使って正確に計算する
-    x0 = MARGIN + 3 * FONT_SIZE * 0.9   # 近似（全角文字）
-    x1 = x0 + FONT_SIZE
-
-    rect = fitz.Rect(x0, y_mupdf_base - FONT_SIZE, x1, y_mupdf_base + 2)
-    quad = rect.quad
-
-    annot = page.add_highlight_annot(quad)
-    annot.set_colors(stroke=hl_color)
-    annot.update()
-    print(f"[OK] 座標指定ハイライト注釈追加: rect={rect}")
+# =====================
+# Gradio UI
+# =====================
+with gr.Blocks(title="Highlight Annotation 検証") as demo:
+    gr.Markdown("## Highlight Annotation 検証ツール")
+    gr.Markdown("ボタンを押すとPDFを生成します。ダウンロードして注釈が認識されるか確認してください。")
 
-    result = io.BytesIO()
-    doc.save(result)
-    doc.close()
-    return result.getvalue()
+    run_btn = gr.Button("検証実行", variant="primary")
+    log_out = gr.Textbox(label="実行ログ", lines=8)
+    pdf_out = gr.File(label="PDFダウンロード")
 
+    run_btn.click(fn=run_verify, inputs=[], outputs=[log_out, pdf_out])
 
-# =====================
-# メイン
-# =====================
-if __name__ == "__main__":
-    # フォントファイル確認
-    if not os.path.isfile(FONT_FILE):
-        print(f"[ERROR] フォントファイルが見つかりません: {FONT_FILE}")
-        print("カレントディレクトリに NotoSansJP-Regular.ttf を配置してください。")
-        exit(1)
-
-    print("=== Step1: reportlabでベースPDF生成 ===")
-    base_pdf = build_base_pdf(FONT_FILE)
-    print(f"[OK] ベースPDF生成完了 ({len(base_pdf)} bytes)")
-
-    print("\n=== Step2: テキスト検索によるHighlight Annotation ===")
-    highlighted_pdf = add_highlight_annotations(base_pdf, color_mode="color1")
-    print(f"[OK] Annotation付与完了 ({len(highlighted_pdf)} bytes)")
-
-    print("\n=== Step3: 座標指定によるHighlight Annotation ===")
-    highlighted_pdf2 = add_highlight_by_position(highlighted_pdf, color_mode="color1")
-    print(f"[OK] 座標指定Annotation付与完了 ({len(highlighted_pdf2)} bytes)")
-
-    with open(OUTPUT_FILE, "wb") as f:
-        f.write(highlighted_pdf2)
-    print(f"\n=== 出力完了: {OUTPUT_FILE} ===")
-    print("Adobe Acrobat等で開き、以下を確認してください：")
-    print("  1. 「犬」「もうある」「XYZ」がハイライト注釈として表示される")
-    print("  2. 注釈パネルでHighlight Annotationとして認識されている")
-    print("  3. 座標指定のハイライトも正しい位置に表示される")
+demo.launch()