screenshot-vision-algorithm 0.3.5__tar.gz → 0.3.8__tar.gz

{screenshot_vision_algorithm-0.3.5 → screenshot_vision_algorithm-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: screenshot_vision_algorithm
-Version: 0.3.5
+Version: 0.3.8
 Summary: Multi-platform screenshot vision algorithms: WeChat (card bbox, speaker bands, avatar detection, OCR helpers), XHS & Zhihu (planned)
 License: MIT
 Requires-Python: >=3.10

{screenshot_vision_algorithm-0.3.5 → screenshot_vision_algorithm-0.3.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "screenshot_vision_algorithm"
-version = "0.3.5"
+version = "0.3.8"
 description = "Multi-platform screenshot vision algorithms: WeChat (card bbox, speaker bands, avatar detection, OCR helpers), XHS & Zhihu (planned)"
 readme = "README.md"
 license = { text = "MIT" }

{screenshot_vision_algorithm-0.3.5 → screenshot_vision_algorithm-0.3.8}/src/screenshot_vision_algorithm/android/wechat/algorithms/speaker_band.py

@@ -26,7 +26,7 @@ from screenshot_vision_algorithm.android.wechat.algorithms.template_matching imp
 CHAT_BAND_FIRST_AVATAR_TOP_GAP_REJECT_BASELINE = 30
 
 #: 首个发言人筒上移量（一个头像高度，防止几何误差），见 PRD §5 补充2。
-CHAT_FIRST_BAND_TOP_EXTEND_BASELINE = 100
+CHAT_FIRST_BAND_TOP_EXTEND_BASELINE = 20
 
 #: 底部输入区预留（与主列表 scanner 一脉，baseline）。
 CHAT_COMPOSER_RESERVE_BOTTOM_BASELINE = 160
@@ -173,7 +173,7 @@ def detect_chat_side_avatar_ytops(
     y1 = max(0, bounds.y_top)
     y2 = min(h, bounds.y_bottom_excl)
     try:
-        from processor.left_avatar_column import (
+        from screenshot_vision_algorithm.android.wechat.algorithms.avatar_column import (
             detect_left_avatar_column_layout,
             left_avatar_ytops,
         )
@@ -254,7 +254,7 @@ def build_prd_speaker_vertical_bands(
         if not merged_starts or y - merged_starts[-1] >= min_sep:
             merged_starts.append(y)
 
-    # 首个筒上界上移（PRD §5 补充2）：上移 100px @1080，防几何误差与系统变化。
+    # 首个筒上界上移（PRD §5 补充2）：上移 20px @1080，防几何误差与系统变化。
     first_band_extend = int(round(int(first_band_top_extend_baseline) * sw))
 
     bands: list[tuple[int, int]] = []
@@ -275,6 +275,13 @@ def build_prd_speaker_vertical_bands(
         elif first_band_start < orphan_top[1]:
             orphan_top = (orphan_top[0], first_band_start)
 
+    # 无 orphan 时，首个 band 上界对齐 y_ct，保证纵向连续全覆盖
+    # （extend 较小且头像紧贴 y_ct 时可能产生小间隙）
+    if orphan_top is None and bands:
+        first_y0, first_y1 = bands[0]
+        if first_y0 > y_ct:
+            bands[0] = (y_ct, first_y1)
+
     return PrdChatVerticalLayout(
         content_bounds=bounds,
         orphan_top_band=orphan_top,

{screenshot_vision_algorithm-0.3.5 → screenshot_vision_algorithm-0.3.8}/src/screenshot_vision_algorithm/android/wechat/ocr/avatar_guard.py

@@ -19,7 +19,7 @@ except ImportError:
     cv2 = None  # type: ignore[assignment]
 
 try:
-    from processor.left_avatar_column import AvatarCentroid
+    from screenshot_vision_algorithm.android.wechat.algorithms.avatar_column import AvatarCentroid
 except ImportError:
     AvatarCentroid = None  # type: ignore[misc, assignment]
 

{screenshot_vision_algorithm-0.3.5 → screenshot_vision_algorithm-0.3.8}/src/screenshot_vision_algorithm/android/wechat/ocr/nickname_binding.py

@@ -351,6 +351,9 @@ _PURE_TIME_RE = re.compile(r"^\d{1,2}:\d{2}(:\d{2})?$")
 _WEEKDAY_TIME_RE = re.compile(
     r"^周[一二三四五六日天].*?(?:上午|下午|晚上|凌晨)?\s*\d{0,2}[:：]?\d{0,2}"
 )
+_YESTERDAY_TODAY_TIME_RE = re.compile(
+    r"^(?:昨天|今天)\s*\d{0,2}:?\d{0,2}$"
+)
 _MOBILE_PHONE_SUFFIX_RE = re.compile(r"1\d{10}\s*$")
 
 
@@ -388,6 +391,8 @@ def _is_chat_timestamp_or_divider(text: str) -> bool:
         return True
     if _WEEKDAY_TIME_RE.match(t):
         return True
+    if _YESTERDAY_TODAY_TIME_RE.match(t):
+        return True
     return False
 
 
@@ -411,7 +416,7 @@ def _is_avatar_column_nickname_row(
     avatar_layout: Any = None,
 ) -> bool:
     try:
-        from processor.left_avatar_column import nickname_bbox_in_avatar_column
+        from screenshot_vision_algorithm.android.wechat.algorithms.avatar_column import nickname_bbox_in_avatar_column
     except ImportError:
         nickname_bbox_in_avatar_column = None  # type: ignore[assignment,misc]
     if avatar_layout is not None and nickname_bbox_in_avatar_column is not None:
@@ -446,7 +451,7 @@ def _avatar_anchor_for_block(
     if avatar_layout is None or getattr(avatar_layout, "empty", True):
         return None
     try:
-        from processor.left_avatar_column import find_avatar_anchor_for_nickname_bbox
+        from screenshot_vision_algorithm.android.wechat.algorithms.avatar_column import find_avatar_anchor_for_nickname_bbox
     except ImportError:
         return None
     return find_avatar_anchor_for_nickname_bbox(
@@ -463,9 +468,9 @@ def _load_left_avatar_layout(
     if png_path is None:
         return None
     try:
-        from processor.chat_bubble_geometry_prd import compute_chat_content_vertical_bounds
-        from processor.left_avatar_column import detect_left_avatar_column_layout
-        from processor.nickname_avatar_guard import load_png_bgr
+        from screenshot_vision_algorithm.android.wechat.algorithms.speaker_band import compute_chat_content_vertical_bounds
+        from screenshot_vision_algorithm.android.wechat.algorithms.avatar_column import detect_left_avatar_column_layout
+        from screenshot_vision_algorithm.android.wechat.ocr.avatar_guard import load_png_bgr
     except ImportError:
         return None
     arr = load_png_bgr(str(png_path))
@@ -595,7 +600,7 @@ def _nickname_passes_avatar_guard(
     if bgr_guard is None:
         return not config.require_avatar_guard_for_nickname
     try:
-        from processor.nickname_avatar_guard import avatar_roi_pass
+        from screenshot_vision_algorithm.android.wechat.ocr.avatar_guard import avatar_roi_pass
     except ImportError:
         return not config.require_avatar_guard_for_nickname
 
@@ -687,13 +692,13 @@ def _load_prd_layout_optional(
     if png_path is None or not config.use_prd_chat_vertical_bands:
         return None
     try:
-        from processor.chat_bubble_geometry_prd import (
+        from screenshot_vision_algorithm.android.wechat.algorithms.speaker_band import (
             build_prd_speaker_vertical_bands,
             compute_chat_content_vertical_bounds,
             detect_chat_side_avatar_ytops,
             merge_avatar_ytops_time_order,
         )
-        from processor.nickname_avatar_guard import load_png_bgr
+        from screenshot_vision_algorithm.android.wechat.ocr.avatar_guard import load_png_bgr
     except ImportError:
         return None
     arr = load_png_bgr(str(png_path))
@@ -962,7 +967,17 @@ def _resume_thumb_bindings_and_orphans(
             _cy1, _cy2 = float(tbb[1]), float(tbb[3])
             for _bi, (_b0, _b1) in enumerate(_sb_list):
                 if _cy2 > float(_b0) and _cy1 < float(_b1):
-                    band_idx = _bi  # 相交 → 候选；持续覆盖 → 最后一个（下方）胜出
+                        band_idx = _bi  # 相交 → 候选；持续覆盖 → 最后一个（下方）胜出
+
+        # ── 如果卡片与 orphan_top_band 相交 → 直接记为找不到归属 ──
+        if layout_opt is not None:
+            ot = getattr(layout_opt, "orphan_top_band", None)
+            if ot is not None:
+                oy0, oy1 = float(ot[0]), float(ot[1])
+                _cy1, _cy2 = float(tbb[1]), float(tbb[3])
+                if _cy2 > oy0 and _cy1 < oy1:
+                    orphan += 1
+                    continue
 
         matched_nickname: Optional[str] = None
 
@@ -1177,7 +1192,7 @@ def first_attrib_verbatim_display_line_and_bbox(
     scale_w = float(original_width) / 1080.0
     if png_path is not None:
         try:
-            from processor.nickname_avatar_guard import load_png_bgr
+            from screenshot_vision_algorithm.android.wechat.ocr.avatar_guard import load_png_bgr
 
             bgr_guard = load_png_bgr(str(png_path))
         except ImportError:
@@ -1748,7 +1763,7 @@ def extract_nicknames(
     bgr_guard = None
     if png_path is not None:
         try:
-            from processor.nickname_avatar_guard import load_png_bgr
+            from screenshot_vision_algorithm.android.wechat.ocr.avatar_guard import load_png_bgr
 
             bgr_guard = load_png_bgr(str(png_path))
         except ImportError:

{screenshot_vision_algorithm-0.3.5 → screenshot_vision_algorithm-0.3.8}/src/screenshot_vision_algorithm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: screenshot_vision_algorithm
-Version: 0.3.5
+Version: 0.3.8
 Summary: Multi-platform screenshot vision algorithms: WeChat (card bbox, speaker bands, avatar detection, OCR helpers), XHS & Zhihu (planned)
 License: MIT
 Requires-Python: >=3.10