From e69983ee85786e7725512fce618f1053a4f64f19 Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Mon, 6 Jul 2026 11:16:27 +0800
Subject: [PATCH] fix: add merge_result_button function for combining OCR
 results

---
 module/ocr/utils.py | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/module/ocr/utils.py b/module/ocr/utils.py
index ed6701e..f05cc18 100644
--- a/module/ocr/utils.py
+++ b/module/ocr/utils.py
@@ -51,6 +51,47 @@ def _merge_boxed_result(left: BoxedResult, right: BoxedResult) -> BoxedResult:
     return left
 
 
+def merge_result_button(
+        results: list[BoxedResult],
+        left_func: callable,
+        right_func: callable,
+        text_func: callable
+) -> list[BoxedResult]:
+    """
+    Args:
+        results:
+        left_func: Function that inputs ocr_text (str) and outputs bool
+            True means mark as left text
+        right_func:
+        text_func: Function that inputs left_text (str) right_text (str) and outputs text (str)
+    """
+    left = None
+    right = None
+    for result in results:
+        if left_func(result.ocr_text):
+            left = result
+        elif right_func(result.ocr_text):
+            right = result
+
+    text = text_func(
+        left.ocr_text if left is not None else '',
+        right.ocr_text if right is not None else ''
+    )
+    if left is not None:
+        if right is not None:
+            results.remove(right)
+            left.box = _merge_area(left.box, right.box)
+            left.ocr_text = text
+        else:
+            left.ocr_text = text
+    else:
+        if right is not None:
+            right.ocr_text = text
+        else:
+            pass
+    return results
+
+
 def merge_buttons(buttons: list[BoxedResult], thres_x=20, thres_y=20) -> list[BoxedResult]:
     """
     Args: