Added Identification Overlay Node

Nodes/General Purpose/identification_overlay.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python3
+"""
+Identification Overlay Node:
+- Creates an OCR region in data_collector with a blue overlay.
+- Detects instances of a specified word and draws adjustable overlays.
+- Users can configure offset and margin dynamically.
+"""
+
+import re
+from OdenGraphQt import BaseNode
+from PyQt5.QtWidgets import QMessageBox
+from PyQt5.QtCore import QTimer
+from Modules import data_manager, data_collector
+
+
+class IdentificationOverlayNode(BaseNode):
+    __identifier__ = "bunny-lab.io.identification_overlay_node"
+    NODE_NAME = "Identification Overlay"
+
+    def __init__(self):
+        super(IdentificationOverlayNode, self).__init__()
+
+        # User-configurable options
+        self.add_text_input("search_term", "Search Term", text="Aibatt")
+        self.add_text_input("offset_value", "Offset Value (X,Y)", text="0,0")  # New input
+        self.add_text_input("margin", "Margin", text="5")  # New input
+
+        self.region_id = "identification_overlay"
+        data_collector.create_ocr_region(self.region_id, x=250, y=50, w=300, h=200, color=(0, 0, 255))
+
+        data_collector.start_collector()
+        self.set_name("Identification Overlay")
+
+        # Timer for updating overlays
+        self.timer = QTimer()
+        self.timer.timeout.connect(self.update_overlay)
+        self.timer.start(500)  # Update every 500ms
+
+    def update_overlay(self):
+        """
+        Updates the overlay with detected word positions.
+        """
+        search_term = self.get_property("search_term")
+        offset_text = self.get_property("offset_value")
+        margin_text = self.get_property("margin")
+
+        # Parse user-defined offset
+        try:
+            offset_x, offset_y = map(int, offset_text.split(","))
+        except ValueError:
+            offset_x, offset_y = 0, 0  # Default to no offset if invalid input
+
+        # Parse user-defined margin
+        try:
+            margin = int(margin_text)
+        except ValueError:
+            margin = 5  # Default margin if invalid input
+
+        if not search_term:
+            return
+
+        # Get detected word positions
+        detected_positions = data_collector.find_word_positions(self.region_id, search_term, offset_x, offset_y, margin)
+
+        # Draw detected word boxes
+        data_collector.draw_identification_boxes(self.region_id, detected_positions, color=(0, 0, 255))