yomitoku 0.9.1__tar.gz → 0.9.3__tar.gz

{yomitoku-0.9.1 → yomitoku-0.9.3}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: yomitoku
-Version: 0.9.1
+Version: 0.9.3
 Summary: Yomitoku is an AI-powered document image analysis package designed specifically for the Japanese language.
 Author-email: Kotaro Kinoshita <kotaro.kinoshita@mlism.com>
 License: CC BY-NC-SA 4.0
 Keywords: Deep Learning,Japanese,OCR
 Requires-Python: <3.13,>=3.10
 Requires-Dist: huggingface-hub>=0.26.1
+Requires-Dist: jaconv>=0.4.0
 Requires-Dist: lxml>=5.3.0
 Requires-Dist: omegaconf>=2.3.0
 Requires-Dist: onnx>=1.17.0
@@ -15,6 +16,7 @@ Requires-Dist: opencv-python>=4.10.0.84
 Requires-Dist: pyclipper>=1.3.0.post6
 Requires-Dist: pydantic>=2.9.2
 Requires-Dist: pypdfium2>=4.30.0
+Requires-Dist: reportlab>=4.4.1
 Requires-Dist: shapely>=2.0.6
 Requires-Dist: timm>=1.0.11
 Requires-Dist: torch>=2.5.0
@@ -41,7 +43,7 @@ YomiToku は日本語に特化した AI 文章画像解析エンジン(Document
 - 🤖 日本語データセットで学習した 4 種類(文字位置の検知、文字列認識、レイアウト解析、表の構造認識)の AI モデルを搭載しています。4 種類のモデルはすべて独自に学習されたモデルで日本語文書に対して、高精度に推論可能です。
 - 🇯🇵 各モデルは日本語の文書画像に特化して学習されており、7000 文字を超える日本語文字の認識をサーポート、手書き文字、縦書きなど日本語特有のレイアウト構造の文書画像の解析も可能です。（日本語以外にも英語の文書に対しても対応しています）。
 - 📈 レイアウト解析、表の構造解析, 読み順推定機能により、文書画像のレイアウトの意味的構造を壊さずに情報を抽出することが可能です。
-- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。
+- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。文書画像をサーチャブルPDFに変換する処理もサポートしています。
 - ⚡ GPU 環境で高速に動作し、効率的に文書の文字起こし解析が可能です。また、VRAM も 8GB 以内で動作し、ハイエンドな GPU を用意する必要はありません。
 
 ## 🖼️ デモ
@@ -85,7 +87,7 @@ yomitoku ${path_data} -f md -o results -v --figure --lite
 ```
 
 - `${path_data}` 解析対象の画像が含まれたディレクトリか画像ファイルのパスを直接して指定してください。ディレクトリを対象とした場合はディレクトリのサブディレクトリ内の画像も含めて処理を実行します。
-- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md をサポート)
+- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md, pdf(searchable-pdf) をサポート)
 - `-o`, `--outdir` 出力先のディレクトリ名を指定します。存在しない場合は新規で作成されます。
 - `-v`, `--vis` を指定すると解析結果を可視化した画像を出力します。
 - `-l`, `--lite` を指定すると軽量モデルで推論を実行します。通常より高速に推論できますが、若干、精度が低下する可能性があります。

{yomitoku-0.9.1 → yomitoku-0.9.3}/README.md

@@ -16,7 +16,7 @@ YomiToku は日本語に特化した AI 文章画像解析エンジン(Document
 - 🤖 日本語データセットで学習した 4 種類(文字位置の検知、文字列認識、レイアウト解析、表の構造認識)の AI モデルを搭載しています。4 種類のモデルはすべて独自に学習されたモデルで日本語文書に対して、高精度に推論可能です。
 - 🇯🇵 各モデルは日本語の文書画像に特化して学習されており、7000 文字を超える日本語文字の認識をサーポート、手書き文字、縦書きなど日本語特有のレイアウト構造の文書画像の解析も可能です。（日本語以外にも英語の文書に対しても対応しています）。
 - 📈 レイアウト解析、表の構造解析, 読み順推定機能により、文書画像のレイアウトの意味的構造を壊さずに情報を抽出することが可能です。
-- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。
+- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。文書画像をサーチャブルPDFに変換する処理もサポートしています。
 - ⚡ GPU 環境で高速に動作し、効率的に文書の文字起こし解析が可能です。また、VRAM も 8GB 以内で動作し、ハイエンドな GPU を用意する必要はありません。
 
 ## 🖼️ デモ
@@ -60,7 +60,7 @@ yomitoku ${path_data} -f md -o results -v --figure --lite
 ```
 
 - `${path_data}` 解析対象の画像が含まれたディレクトリか画像ファイルのパスを直接して指定してください。ディレクトリを対象とした場合はディレクトリのサブディレクトリ内の画像も含めて処理を実行します。
-- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md をサポート)
+- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md, pdf(searchable-pdf) をサポート)
 - `-o`, `--outdir` 出力先のディレクトリ名を指定します。存在しない場合は新規で作成されます。
 - `-v`, `--vis` を指定すると解析結果を可視化した画像を出力します。
 - `-l`, `--lite` を指定すると軽量モデルで推論を実行します。通常より高速に推論できますが、若干、精度が低下する可能性があります。

{yomitoku-0.9.1 → yomitoku-0.9.3}/README_EN.md

@@ -59,7 +59,7 @@ yomitoku ${path_data} -f md -o results -v --figure --lite
 ```
 
 - `${path_data}`: Specify the path to a directory containing images to be analyzed or directly provide the path to an image file. If a directory is specified, images in its subdirectories will also be processed.
-- `-f`, `--format`: Specify the output file format. Supported formats are json, csv, html, and md.
+- `-f`, `--format`: Specify the output file format. Supported formats are json, csv, html, md , and pdf(searchable-pdf).
 - `-o`, `--outdir`: Specify the name of the output directory. If it does not exist, it will be created.
 - `-v`, `--vis`: If specified, outputs visualized images of the analysis results.
 - `-l`, `--lite`: inference is performed using a lightweight model. This enables fast inference even on a CPU.

{yomitoku-0.9.1 → yomitoku-0.9.3}/docs/cli.en.md

@@ -7,7 +7,7 @@ yomitoku ${path_data} -v -o results
 ```
 
 - `${path_data}`: Specify the path to a directory containing images to be analyzed or directly provide the path to an image file. If a directory is specified, images in its subdirectories will also be processed.
-- `-f`, `--format`: Specify the output file format. Supported formats are json, csv, html, and md.
+- `-f`, `--format`: Specify the output file format. Supported formats are json, csv, html, md , and pdf(searchable-pdf).
 - `-o`, `--outdir`: Specify the name of the output directory. If it does not exist, it will be created.
 - `-v`, `--vis`: If specified, outputs visualized images of the analysis results.
 
@@ -41,6 +41,8 @@ You can specify the output format of the analysis results using the --format or
 yomitoku ${path_data} -f md
 ```
 
+- `pdf`: Detect the text in the image and embed it into the PDF as invisible text, converting the file into a searchable PDF.
+
 ## Specifying the Output Device
 
 You can specify the device for running the model using the -d or --device option. Supported options are cuda, cpu, and mps. If a GPU is not available, inference will be performed on the CPU. (Default: cuda)

{yomitoku-0.9.1 → yomitoku-0.9.3}/docs/cli.ja.md

@@ -31,12 +31,14 @@ yomitoku ${path_data} --lite -v
 
 ## 出力フォーマットの指定
 
-`--format`, `-f` オプションを使用することで解析結果の出力形式のフォーマットを指定できます。出力フォーマットの種類は json, csv, html, md(markdown)をサポートしています。
+- `-f`, `--format` 出力形式のファイルフォーマットを指定します。(json, csv, html, md, pdf(searchable-pdf) をサポート)
 
 ```
 yomitoku ${path_data} -f md
 ```
 
+- pdf: 画像内の文字情報を認識し、文字情報を透明テキストとして、PDFに埋め込むことで、サーチャブルPDFに変換します。
+
 ## 出力デバイスの指定
 
 - `-d`, `--device` オプションを使用することで、モデルを実行するためのデバイスを指定します。(cuda | cpu | mps)。gpu が利用できない場合は cpu で推論が実行されます。(デフォルト: cuda)
@@ -117,5 +119,4 @@ yomitoku ${path_data} --reading_order left2right
 
 - `left2right`: 左から右方向に優先的に読み取り順を推定します。レシートや保険証などキーに対して、値を示すテキストが段組みになっているようなレイアウトに有効です。
 
-- `right2left:` 右から左方向に優先的に読み取り順を推定します。縦書きのドキュメントに対して有効です。
-
+- `right2left:` 右から左方向に優先的に読み取り順を推定します。縦書きのドキュメントに対して有効です。

{yomitoku-0.9.1 → yomitoku-0.9.3}/docs/index.en.md

@@ -5,7 +5,7 @@ YomiToku is a Document AI engine specialized in Japanese document image analysis
 - 🤖 Equipped with four AI models trained on Japanese datasets: text detection, text recognition, layout analysis, and table structure recognition. All models are independently trained and optimized for Japanese documents, delivering high-precision inference.
 - 🇯🇵 Each model is specifically trained for Japanese document images, supporting the recognition of over 7,000 Japanese characters, including vertical text and other layout structures unique to Japanese documents. (It also supports English documents.)
 - 📈 By leveraging layout analysis, table structure parsing, and reading order estimation, it extracts information while preserving the semantic structure of the document layout.
-- 📄 Supports a variety of output formats, including HTML, Markdown, JSON, and CSV. It also allows for the extraction of diagrams and images contained within the documents.
+- 📄 Supports a variety of output formats, including HTML, Markdown, JSON, and CSV. It also allows for the extraction of diagrams and images contained within the documents.It also supports converting document images into fully text-searchable PDFs.
 - ⚡ Operates efficiently in GPU environments, enabling fast document transcription and analysis. It requires less than 8GB of VRAM, eliminating the need for high-end GPUs.。
 
 ## 🙋 FAQ

{yomitoku-0.9.1 → yomitoku-0.9.3}/docs/index.ja.md

@@ -5,7 +5,7 @@ YomiToku は日本語に特化した AI 文章画像解析エンジン(Document
 - 🤖 日本語データセットで学習した 4 種類(文字位置の検知、文字列認識、レイアウト解析、表の構造認識)の AI モデルを搭載しています。4 種類のモデルはすべて独自に学習されたモデルで日本語文書に対して、高精度に推論可能です。
 - 🇯🇵 各モデルは日本語の文書画像に特化して学習されており、7000 文字を超える日本語文字の認識をサーポート、手書き文字、縦書きなど日本語特有のレイアウト構造の文書画像の解析も可能です。（日本語以外にも英語の文書に対しても対応しています）。
 - 📈 レイアウト解析、表の構造解析, 読み順推定機能により、文書画像のレイアウトの意味的構造を壊さずに情報を抽出することが可能です。
-- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。
+- 📄 多様な出力形式をサポートしています。html やマークダウン、json、csv のいずれかのフォーマットに変換可能です。また、文書内に含まれる図表、画像の抽出の出力も可能です。文書画像を全文検索可能なサーチャブルPDFに変換する処理もサポートしています。
 - ⚡ GPU 環境で高速に動作し、効率的に文書の文字起こし解析が可能です。また、VRAM も 8GB 以内で動作し、ハイエンドな GPU を用意する必要はありません。
 
 ## 🙋 FAQ

{yomitoku-0.9.1 → yomitoku-0.9.3}/docs/mcp.en.md

@@ -46,3 +46,19 @@ For example, if you use `yomitoku/demo/sample.pdf` as a sample, instruct as foll
 ```txt
 Analyze sample.pdf using OCR and translate it into English.
 ```
+
+## Starting the SSE Server
+
+Set the path to the folder containing the images to be processed by OCR in the resource directory.
+
+```
+export RESOURCE_DIR="path of dataset"
+```
+
+Start the SSE server using the following command:
+
+```
+uv run yomitoku_mcp -t sse
+```
+
+The SSE server endpoint will be available at `http://127.0.0.1:8000/sse`.

{yomitoku-0.9.1 → yomitoku-0.9.3}/docs/mcp.ja.md

@@ -13,6 +13,7 @@
 uv sync --extra mcp
 ```
 
+
 ## Claude Desktopの設定
 
 次にClaude Desktopの設定ファイルの`mcpServers`に以下ように設定を追加します。(設定ファイルの開き方は[こちら](https://modelcontextprotocol.io/quickstart/user)を参照してください)
@@ -48,3 +49,16 @@ uv sync --extra mcp
 ```txt
 sample.pdfをOCRで解析して要約してください。
 ```
+
+## SSEサーバーの起動
+環境変数の`RESOURCE_DIR`にOCRの対象画像が含まれたフォルダのパスを設定してください。
+```
+export RESOURCE_DIR="path of dataset"
+```
+
+以下のコマンドでSSEサーバーを起動します。
+```
+uv run yomitoku_mcp -t sse
+```
+
+` http://127.0.0.1:8000/sse`がSSEサーバーのエンドポイントになります。

{yomitoku-0.9.1 → yomitoku-0.9.3}/pyproject.toml

@@ -30,6 +30,8 @@ dependencies = [
     "torch>=2.5.0",
     "torchvision>=0.20.0",
     "onnxruntime>=1.20.1",
+    "reportlab>=4.4.1",
+    "jaconv>=0.4.0",
 ]
 
 [tool.uv-dynamic-versioning]
@@ -70,7 +72,7 @@ explicit = true
 
 [project.scripts]
 yomitoku = "yomitoku.cli.main:main"
-yomitoku_mcp = "yomitoku.cli.mcp:run_mcp_server"
+yomitoku_mcp = "yomitoku.cli.mcp_server:main"
 
 [project.optional-dependencies]
 mcp = [

{yomitoku-0.9.1 → yomitoku-0.9.3}/src/yomitoku/cli/main.py

@@ -9,6 +9,7 @@ from ..constants import SUPPORT_OUTPUT_FORMAT
 from ..data.functions import load_image, load_pdf
 from ..document_analyzer import DocumentAnalyzer
 from ..utils.logger import set_logger
+from ..utils.searchable_pdf import create_searchable_pdf
 
 from ..export import save_csv, save_html, save_json, save_markdown
 from ..export import convert_json, convert_csv, convert_html, convert_markdown
@@ -48,10 +49,15 @@ def merge_all_pages(results):
             else:
                 out += "\n" + data
 
+        elif format == "pdf":
+            if out is None:
+                out = [data]
+            else:
+                out.append(data)
     return out
 
 
-def save_merged_file(out_path, args, out):
+def save_merged_file(out_path, args, out, imgs):
     if args.format == "json":
         save_json(out, out_path, args.encoding)
     elif args.format == "csv":
@@ -60,6 +66,13 @@ def save_merged_file(out_path, args, out):
         save_html(out, out_path, args.encoding)
     elif args.format == "md":
         save_markdown(out, out_path, args.encoding)
+    elif args.format == "pdf":
+        create_searchable_pdf(
+            imgs,
+            out,
+            output_path=out_path,
+            font_path=args.font_path,
+        )
 
 
 def validate_encoding(encoding):
@@ -80,7 +93,7 @@ def process_single_file(args, analyzer, path, format):
     else:
         imgs = load_image(path)
 
-    results = []
+    format_results = []
     for page, img in enumerate(imgs):
         result, ocr, layout = analyzer(img)
         dirname = path.parent.name
@@ -130,7 +143,7 @@ def process_single_file(args, analyzer, path, format):
                     figure_dir=args.figure_dir,
                 )
 
-            results.append(
+            format_results.append(
                 {
                     "format": format,
                     "data": json.model_dump(),
@@ -157,7 +170,7 @@ def process_single_file(args, analyzer, path, format):
                     figure_dir=args.figure_dir,
                 )
 
-            results.append(
+            format_results.append(
                 {
                     "format": format,
                     "data": csv,
@@ -188,7 +201,7 @@ def process_single_file(args, analyzer, path, format):
                     encoding=args.encoding,
                 )
 
-            results.append(
+            format_results.append(
                 {
                     "format": format,
                     "data": html,
@@ -219,20 +232,36 @@ def process_single_file(args, analyzer, path, format):
                     encoding=args.encoding,
                 )
 
-            results.append(
+            format_results.append(
                 {
                     "format": format,
                     "data": md,
                 }
             )
+        elif format == "pdf":
+            if not args.combine:
+                create_searchable_pdf(
+                    [img],
+                    [result],
+                    output_path=out_path,
+                    font_path=args.font_path,
+                )
+
+            format_results.append(
+                {
+                    "format": format,
+                    "data": result,
+                }
+            )
 
-    out = merge_all_pages(results)
+    out = merge_all_pages(format_results)
     if args.combine:
         out_path = os.path.join(args.outdir, f"{dirname}_{filename}.{format}")
         save_merged_file(
             out_path,
             args,
             out,
+            imgs,
         )
 
 
@@ -349,7 +378,12 @@ def main():
         type=str,
         choices=["auto", "left2right", "top2bottom", "right2left"],
     )
-
+    parser.add_argument(
+        "--font_path",
+        default=None,
+        type=str,
+        help="Path to the font file(.ttf) for PDF output",
+    )
     args = parser.parse_args()
 
     path = Path(args.arg1)
@@ -362,6 +396,13 @@ def main():
             f"Invalid output format: {args.format}. Supported formats are {SUPPORT_OUTPUT_FORMAT}"
         )
 
+    if (
+        args.font_path is not None
+        and not os.path.exists(args.font_path)
+        and format == "pdf"
+    ):
+        raise FileNotFoundError(f"Font file not found: {args.font_path}")
+
     validate_encoding(args.encoding)
 
     if format == "markdown":

yomitoku-0.9.1/src/yomitoku/cli/mcp.py → yomitoku-0.9.3/src/yomitoku/cli/mcp_server.py

@@ -1,14 +1,20 @@
-import json
-import io
 import csv
+import io
+import json
 import os
+from argparse import ArgumentParser
 from pathlib import Path
 
 from mcp.server.fastmcp import Context, FastMCP
 
 from yomitoku import DocumentAnalyzer
 from yomitoku.data.functions import load_image, load_pdf
-from yomitoku.export import convert_json, convert_markdown, convert_csv, convert_html
+from yomitoku.export import (
+    convert_csv,
+    convert_html,
+    convert_json,
+    convert_markdown,
+)
 
 try:
     RESOURCE_DIR = os.environ["RESOURCE_DIR"]
@@ -154,12 +160,37 @@ async def get_file_list() -> list[str]:
     return os.listdir(RESOURCE_DIR)
 
 
-def run_mcp_server():
+def run_mcp_server(transport="stdio", mount_path=None):
     """
     Run the MCP server.
     """
-    mcp.run(transport="stdio")
+
+    if transport == "stdio":
+        mcp.run()
+    elif transport == "sse":
+        mcp.run(transport=transport, mount_path=mount_path)
+
+
+def main():
+    parser = ArgumentParser(description="Run the MCP server.")
+    parser.add_argument(
+        "--transport",
+        "-t",
+        type=str,
+        default="stdio",
+        choices=["stdio", "sse"],
+        help="Transport method for the MCP server.",
+    )
+    parser.add_argument(
+        "--mount_path",
+        "-m",
+        type=str,
+        default=None,
+        help="Mount path for the MCP server (only used with SSE transport).",
+    )
+    args = parser.parse_args()
+    run_mcp_server(transport=args.transport, mount_path=args.mount_path)
 
 
 if __name__ == "__main__":
-    run_mcp_server()
+    main()

{yomitoku-0.9.1 → yomitoku-0.9.3}/src/yomitoku/constants.py

@@ -1,7 +1,7 @@
 import os
 
 ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
-SUPPORT_OUTPUT_FORMAT = ["json", "csv", "html", "markdown", "md"]
+SUPPORT_OUTPUT_FORMAT = ["json", "csv", "html", "markdown", "md", "pdf"]
 SUPPORT_INPUT_FORMAT = ["jpg", "jpeg", "png", "bmp", "tiff", "tif", "pdf"]
 MIN_IMAGE_SIZE = 32
 WARNING_IMAGE_SIZE = 720

yomitoku-0.9.3/src/yomitoku/utils/searchable_pdf.py

@@ -0,0 +1,119 @@
+import os
+
+from PIL import Image
+from io import BytesIO
+
+from reportlab.pdfgen import canvas
+from reportlab.pdfbase.ttfonts import TTFont
+from reportlab.pdfbase import pdfmetrics
+from reportlab.pdfbase.pdfmetrics import stringWidth
+
+import numpy as np
+import jaconv
+
+from ..constants import ROOT_DIR
+
+FONT_PATH = ROOT_DIR + "/resource/MPLUS1p-Medium.ttf"
+
+
+def _poly2rect(points):
+    """
+    Convert a polygon defined by its corner points to a rectangle.
+    The points should be in the format [[x1, y1], [x2, y2], [x3, y3], [x4, y4]].
+    """
+    points = np.array(points, dtype=int)
+    x_min = points[:, 0].min()
+    x_max = points[:, 0].max()
+    y_min = points[:, 1].min()
+    y_max = points[:, 1].max()
+
+    return [x_min, y_min, x_max, y_max]
+
+
+def _calc_font_size(content, bbox_height, bbox_width):
+    rates = np.arange(0.5, 1.0, 0.01)
+
+    min_diff = np.inf
+    best_font_size = None
+    for rate in rates:
+        font_size = bbox_height * rate
+        text_w = stringWidth(content, "MPLUS1p-Medium", font_size)
+        diff = abs(text_w - bbox_width)
+        if diff < min_diff:
+            min_diff = diff
+            best_font_size = font_size
+
+    return best_font_size
+
+
+def to_full_width(text):
+    fw_map = {
+        "\u00a5": "\uffe5",  # ¥ → ￥
+        "\u00b7": "\u30fb",  # · → ・
+        " ": "\u3000",  # 半角スペース→全角スペース
+    }
+
+    TO_FULLWIDTH = str.maketrans(fw_map)
+
+    jaconv_text = jaconv.h2z(text, kana=True, ascii=True, digit=True)
+    jaconv_text = jaconv_text.translate(TO_FULLWIDTH)
+
+    return jaconv_text
+
+
+def create_searchable_pdf(images, ocr_results, output_path, font_path=None):
+    if font_path is None:
+        font_path = FONT_PATH
+
+    pdfmetrics.registerFont(TTFont("MPLUS1p-Medium", font_path))
+
+    packet = BytesIO()
+    c = canvas.Canvas(packet)
+
+    for i, (image, ocr_result) in enumerate(zip(images, ocr_results)):
+        image = Image.fromarray(image[:, :, ::-1])  # Convert BGR to RGB
+        pdfmetrics.registerFont(TTFont("MPLUS1p-Medium", FONT_PATH))
+
+        image_path = f"tmp_{i}.png"
+        image.save(image_path)
+        w, h = image.size
+
+        c.setPageSize((w, h))
+        c.drawImage(image_path, 0, 0, width=w, height=h)
+        os.remove(image_path)  # Clean up temporary image file
+
+        for word in ocr_result.words:
+            text = word.content
+            bbox = _poly2rect(word.points)
+            direction = word.direction
+
+            x1, y1, x2, y2 = bbox
+            bbox_height = y2 - y1
+            bbox_width = x2 - x1
+
+            if direction == "vertical":
+                text = to_full_width(text)
+
+            if direction == "horizontal":
+                font_size = _calc_font_size(text, bbox_height, bbox_width)
+            else:
+                font_size = _calc_font_size(text, bbox_width, bbox_height)
+
+            c.setFont("MPLUS1p-Medium", font_size)
+            c.setFillColorRGB(1, 1, 1, alpha=0)  # 透明
+            if direction == "vertical":
+                base_y = h - y2 + (bbox_height - font_size)
+                for j, ch in enumerate(text):
+                    c.saveState()
+                    c.translate(x1 + font_size * 0.5, base_y - (j - 1) * font_size)
+                    c.rotate(-90)
+                    c.drawString(0, 0, ch)
+                    c.restoreState()
+            else:
+                base_y = h - y2 + (bbox_height - font_size) * 0.5
+                c.drawString(x1, base_y, text)
+        c.showPage()
+    c.save()
+
+    with open(output_path, "wb") as f:
+        f.write(packet.getvalue())

{yomitoku-0.9.1 → yomitoku-0.9.3}/tests/test_cli.py

@@ -152,6 +152,31 @@ def test_run_tiff_csv(monkeypatch, tmp_path):
     assert os.path.exists(out_path)
 
 
+def test_run_tiff_pdf(monkeypatch, tmp_path):
+    path_img = "tests/data/test.tiff"
+    monkeypatch.setattr(
+        "sys.argv",
+        [
+            "main.py",
+            path_img,
+            "-o",
+            str(tmp_path),
+            "-f",
+            "pdf",
+            "--tsr_cfg",
+            "tests/yaml/table_structure_recognizer.yaml",
+            "--lite",
+            "--figure",
+        ],
+    )
+    main.main()
+    path = Path(path_img)
+    dirname = path.parent.name
+    filename = path.stem
+    out_path = os.path.join(str(tmp_path), f"{dirname}_{filename}_p1.pdf")
+    assert os.path.exists(out_path)
+
+
 def test_run_pdf_md(monkeypatch, tmp_path):
     path_img = "tests/data/test.pdf"
     monkeypatch.setattr(