From 0515f8feebfe0d0d15511e60253285d2d8afa7d0 Mon Sep 17 00:00:00 2001
From: Nicole Rappe <nicole.rappe@bunny-lab.io>
Date: Wed, 26 Feb 2025 01:57:47 -0700
Subject: [PATCH] Enabled GPU Acceleration for Identification Node.

---
 Installation Requirements.txt                 |   5 ++
 .../data_collector.cpython-312.pyc            | Bin 13762 -> 14564 bytes
 Modules/data_collector.py                     |  66 +++++++++++-------
 .../identification_overlay.cpython-312.pyc    | Bin 3995 -> 4179 bytes
 .../General Purpose/identification_overlay.py |  18 +++--
 5 files changed, 56 insertions(+), 33 deletions(-)
 create mode 100644 Installation Requirements.txt

diff --git a/Installation Requirements.txt b/Installation Requirements.txt
new file mode 100644
index 0000000..94ef1d6
--- /dev/null
+++ b/Installation Requirements.txt	
@@ -0,0 +1,5 @@
+We need to run the following commands to install the prerequisites for the project.
+
+
+This command is used to install pytorch and torchvision for the purposes of GPU-accelerated vision tasks.
+pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
diff --git a/Modules/__pycache__/data_collector.cpython-312.pyc b/Modules/__pycache__/data_collector.cpython-312.pyc
index f99d78d41f91fdbad54aa5b13dcd4e5eefd81f24..107d0944c578b79ecfebb370ae1ad1b1545a2489 100644
GIT binary patch
delta 4331
zcmbVPeN0=|6@S-X*kEIWKfxFb1``vK5C~1Ogd{W&NR!f!BuitZc#hx0G5D+ZJrbbL
zBhsoaohmKSof@{b8ffcEnogplQ`4zRyMCx$yG?kcP4^m2?Ut=dr%p^s|LGsy&Up@B
zRr61;^v=2W+;h*p=XXEOd9M#$d#Ln1i=~)@lK9|KTne5o^^iZGZQM0ZlE$!+=XjDd
zg-yUU$>LN=xFlr`n^Tssg>WF#CQDP+ur*~1+tju$S(Yjfm#6Gud&&`Zq?}=A$`y90
zefp$3<q3OI72%3hWw<is4SNZv;RNI07My)`;SF7J!WTVnyr3y`gnhSHjT6B%xP7$&
zYz^*Y_Z3yTsD?MCR~z&5#b+<M$7>hI@Fn3o-W(3_mT)~^8g78OoEL6&hJy(%+{lyY
z8s7Q{7Y^|@zD%eS0(|*^p0__j!cDvb<ZF2+&}QBRbRF*ox}LA%J<z{__wg0LTQpn`
z$5%eW@m|4pLCbL}>fw!}Bvj4%$)grUBc%6)h@y>WhYzxA<Q~@CU~+W9I?i#@;!?R&
zm6DbvH5*LaD1o7xF^Lsx_uX~(uqgSD3I13#nG|^c0L`TQ(s99`iYC%fp+b~jU+jm{
z7t$%_Xf_*3q(v#3jtSHNwiT^7Dk{1tjh|pIYVW?Y8e~EEL?4h-+)XoQbj%fbuet-j
z-2H6tRPVfd-DBnjEoZEvRiJVGOC+s8LfY&914e6ShjeceQq5Wn$9-r-DT~1(Bq5TC
z(FhgdiA<W}fEM;!LzYyrV9_7;2XJHqkdTfxVtoyg5RxX8YG^YsMJEX(l87h6@29G5
zb6k)jG<q_EJ+vKqy4W^jADI1-@!|21Mlnd>njk3#J}RU#X~4i#040%*D~5wAK8h_e
zoaGmnn9O9diXI@M66^_)M1u1brn9QzNGt<b$D|CUJD{%*J`r&{#a%U)T`_v{M$ct`
z%S8uEnyjR1RyM^n#(L;KPiQ^+w4|mPJ&g~j<zG8PmSvX}T{0m`s^f}oNEj7q6Kgep
z?*NXb>yWHRvH?j85T#fIaD}vZJR>QlEX~9OQ51L;N)eA4ZhWDr4$y5l63GGfPxB2}
z++Qtwi5Kq!MOdj92zs4e^qO^8TS=N_tzN+NackC!u+u$@WQdj6gep7<F+5F+#RL^N
z=-iD7J?kvm#@@Dd5d(X*D0H9UsE*@uMLA>6lq;5tWFvg0F;InCG4cVOs!@_N=PbEW
z*&>$=RRX?xxl}e|joKu%t`=S+8`z^|ZQ~`f88k}=bkA!q;C4aDStV<s2DYtBQk!fY
zv@Ob)`BCsbSC%W6ZMW^U=NxjGY#%EDO{ZKgJJgyh&gI-@S<mZ+;LOr5Bs=S<uF}ai
zUeCT;-WD+6g~yWfNC>W4Bm_|{vWG8XGv#jmdC;g}e=M)D8)fn&#~Yv4VqXPl$rbFY
z@~tKhZ@SewZm${l@Wq9eN?l`mj$884E8ziC3)m=Gq32wsSG7vdyh3laJ)i6z#9O(d
zunk`QzDil6?wnWFs=P|p@fMsxZTVKV26T(7Bi!Q`be!xyz)5(&)iT(`YQ1*1ZBZvz
zE#ZAH^liXCU-eR7ja<{kJ@_k*I||6=3^{+U7IAYg<K~y}HWo(LK`5|qy|o7-d3>X6
z;7e6c4ZH<;Or!37T`r*V`dq^bRO(pL?mZHaSL3k~-+H-j3@(~{dt@8rfU2?DhKebA
zHvVl<oGzdk;3Rwr)KbsU%3N^8{jkcxC%LCJi`Wq^4*nkDcv~*`{onzW+rA$>NltPj
z+9TXa5-J;en_Y4|+;>P6ME^m2OlUMF`Pb~e_fW{6q5kfuIC{_S`~7QrP{0dWA<e@}
z>d#=T5&WWn?;?aDAstVog+kzgNACyD(4JkeL`4gRXeZR*Nxu^XC7EF<J1j+{(X2o>
zfn-c`^cXyY!hob`rA(GS05W<6$aVFBVxK$Lw^b~^n9D&z(PwEQEm3%1INFT_UnxZ=
zCdLHX!}?rzZ8kvog9R&^G`#RpN~5DnF$PG1Mj-SmMyz8XQB2(<F(E5Ks0uaGW;W|;
zs8fnDeii(p-}oN0|G0dH2XW#)R_5+?mY&Q|K9bFd2{Z&j7D7yvrQPc~5~*lhh`=I>
zs2~m}B~kTUArVDC5=kY}ihi`fpcNfaTfm4Q;w@yKxU1QM+oYjBMm!ZIKfxK{6Srtt
zgJ*)QXO}#O$3L#{o-X~k#5v*oOG))Ld-WB2bKc%OePnLuy_fqFSNgO0{_K_h;e7w_
zW&4TKUDuqIPqj@*lY4$D&wHCcaIQN;uG*dFww~QOshQj~8JyNk^iLn037$WCvG#oA
zm1D1_UQWGl-+#5HZgSJKVQTvfnc6vXI3L(PU(<1>_iD{J%Joy*rXQNxG1HR|Y@4sy
zj&c+iO?Awa<pXUq@qD0TzGlaw+z6_*`M{=`llj1o`I?<)dOvp5U2%l+j?na$ndAAj
zJ6<!svn9Xlz}%q+=c4hs!J&L~D1Rt9mp-w;aSxMD4X`dvx8}bbH~uhDd4}vk>DnI6
z7aZs8)ht*!*ZPH4&Q?8V^xgDvp2~A0XGbPY^X{h8y;oiSE3Q!96`Ec%?`k{U^N&E&
zS?d|c8EN9aZvf-P)|Wb7=y>1W`C<Fki?uHtzR16}@#XlOqxEdxnM30f-BW#&ho-x)
zfy?dFo2Pcp*R&NJUES)Q?w-1PC31C>UlV;R5CW?Aw*$e>)vsM}vF^&|&3NTrH|)0D
z73AJ>bbbfe+t$u1VDB_{ZUy#k(6rB_eRqR(pHBPk4g&S}boza!*`HNzBK9B*?hENx
z-bVI`*K?1Wv*`*sn`!9+(2aOfag2y}6w{JIR#m}c2AAN4w3HY~#G-gzBF8c#f=DMo
z@nzOq^#EiaQ&mUF+S#wFE@((Id#<K+`!^&+>YVCCiYXF_C8MGkiHOEq)4`0V3VF~|
z%;Wcwx7Z#2-#`*{-9M-)$0P`HF$*wSisV@)1pJjt?~h^>1!;FC30WT|7Id2ZJWxxX
zW4{kn*uK6J_F15k_?W(aEqR}9tp5yFU)Au&9n&~Y9sDfR6hk5F3~A|e*l`XCVkBa?
zT11CF!JNTH@^97=41t+=@Ne6emPFAprKGpBuWNsI<iOr=_oA((3~{xsS$3+ip4?#1
zG(w6zd$VyL*)fYIjIm5wf*%G^(ZY672(BD`1_-3%>e60B!4HwV1Z2hbv0SK~e92}*
zLtw9_NzsqHK;b&(VCo|ovK5RiFBzdyisH%4v1sxEHJx2-9idO-Fnr?*`9J0W%kMk=
z9X4@QiY}EoA%u)8*0$s?8)<0*<bTo9rqPnw4_lfv0GhU~Y41}w`XZ7ikf7W24J3a8
zqF8PnmY9%CMj|0X@h?!#fp--S%vw}M@+Ldc)-;Y8ui6BJ)VQRQU!(LGkPv~5SeC0t
ztCeoW?!8FTNHF_Q^y%SLc9e=(*Tzn?(jKf82@q!?-Y9zb$%q|SwD<1aPw`X~qMM$X
zHpN4R`_<636vAQ<!gR4=KB|}#VkCMZnn*^EB?UUnlABt`UqQR({hfDq?~n9$@9pV5
zSU?(+FzP{qNt60#rv68%zcz*c1&pIq4Kov1Q=!0IR{aAkTw&w(RE8f;3OnfEp#!&H
z{2a)FmJo8oz>zZMYqy%QX8(vQ{fIMv%39hT+ZGI5Y1Qe&Kkk35f8yZ8o@xDM%lcEr
zA6mT2vKjqL#V-_JY3a$g^jvP)dtUd>@MX)PQ^l;SeKW+mTzkmZMM(LLK8=z1Zn`3b
NG@OZl!68@I_kT7HG*<us

delta 3606
zcmZ`*Yit|G5xzSf$>aMaQKU$bw5S(l$+jfh@*{~YE0SqPwh+fn?1NL5=AC5Pe0lF=
zJ#w^M)CF9iN#tySN=Vbld9<=!q)LP~DO#jKn#3uBqGlCW?!|5L!)@TAKMXkeQJ_CM
zdlV(BO%}LsXJ%*Tc4v2I=Jx4#PuV}R*~}EQmp*!X_+Ip)-A8v`Y~DS|WVoz}H)YMd
zIcwo9G$l}iE@REwcw5%a+cloeII>ROnRW55tebadJ-jDd#aC%#`b>4!%X_mmd`;HJ
z`?7xC4|0QOJZUF?TiX(~6<#p%0fFO#lY)t_6U=;wp+wKs<xxJIruc|JC)Wv<^Auk%
zSOuFH5QBn!NG~|f(|lBLg1kX+0X7P5z$U>1*ev*jDj07O{6aPKF^1Yp3EuORP$Sx}
z>dNN%)^R#kdxtOwW0+<p@zczpnFW?8-1`7O!$tADlVi^vkZJsHW+TnwAGtz|%;DF~
zLA${GOm~$bvnlGN3zyiC0qn68kKhj7BZ-5DFEgq^PNHE^Rt-W@%;s}36Ec9Z%B6FX
zoXn*}RVR&0Dho<yRBN(ONURK)<P%6t3hCUiYB;Qss$LuuQwdcMQ<026&>cA30LJRz
zlllP8QD2xTu4>5`EE$7~#?W<R)2*85E4@FBUy3i*v|q5^;#^Cduf+LYu`WeAOOejE
zHh;|RUgG*oT>lL&epgR%&8QylVyo%qg>SI$(R2f@HJqr6k|n!RupC)T<Wnet#Nl*4
zhqi&d8NX;K(Dm49{QW=-c(fiM#-esYZy>Odz$PMPP$zUMD~n^YMEn+sAkDTFT$Dh`
z(F7SmJ7HueZsZPvS($rk5P8XD7Lkpj5P<4Tj1+{*Vlw%B0Y!)kt<thY1GQ|{p*bp%
z%4afSO3ov+3&#5J<ECc1e&MVs#c-`4xJ;u~e8&=QBP$;s(%L_E#cZovP}xjclC_Pf
z>?v_vLYwer>$j@OY}7?yGl4Axw&KgyK$Mu0$Qz5SG;OEoL83!oKmOc$50>$XZ66(I
zCQ{-Mv<`smqfprg-sR}56G1Au0nSl#9dpu^?%Ce!4cp(M0pPrzaTFZHjc9KLj$@ls
z4Auji(gReQ5eV&Kx3PGzZaco_?5FkE?utFCKL(e+Xeb(sT+yT$6eE1x1gHaLBK;7n
zX_$+aqP1vKtcv**JYY$$*c1z)kW)5Qej_l79$#>EO_~)8XxfHYcqZgnpcU=1qfCKq
z`-;?|*iSku^40zr_@wA8x)jIPj=GB;#i_U_ETCDXxD<~@Rm&cVDsLT^I;vU4A+UJJ
z-PNnFJpH0~m@3xDfy&Qha=Wyqcm;zWsLJ9h{lnCge}td-@J)BvV^rwpDZ%)HuA;5@
zu-UV{!z*xPuHw_MY68|!nP2qFjpaYUxK}Wh_3(fvJn0q8Upp(XW|rkXz3MgAgEF~A
zS`(~_UtuTU{#Bw=v2M+1?b^{D#gGzESiwdNlG84RmEamaqSPsjHnL6$DLRd+S87hi
zu*>3^T&L74VI?xeR<=r0-!HM0Qu7!UrDXC-wAMhO+G6y+dtz6j&r>fjtFwlw6mygk
z9L4C_=wr+MC_PGz>5fvPbj&&NJ2QDt{pp+_MGAQ-EkhWR@NVz${zIZXf^t$M86j^q
zk{^m3&yVCt_yl4slFH{1Fc34O2of_%IekWq$oYtTLW~Ud9wIs$G}{mhzVya=dSJz9
z4*&$MUg%Uyenc*e$O(D8Afg>0nP83`hrl2X$*N9<2y~Rp;0ch2d<P3P8@jx!F=RCi
zs`>&-=VSygjzawe$h%cpDLo;gIDWroe?XT_j;RJn2Vzc8O?_i2u|RejYe1d2-`D6x
zUmba#_SG4_Nem9+d0)Jmg%Fuk?W3^s#HzDw{`TEti7aHCbWY{QSE$hy>O_SyjwOb`
zXBpKvzD&@f)Br!g-}cwxDZi0HQ9SGSTb>138a`<sgWvN{(@prv+9xJ&J8G94EhR_G
zod5dHfsc7{iO-by%o2aP#Gn4yfiAG0`2!d1w_M&INYk5V4Koic*0lWD6}!-L%i(%<
z{Cnfmd#2^-q1iptp}DHLq07Pf<mK>h>VMn*tM(g?`0YSwTAuBnQRaGPCg*EP;cbh7
z?Ni(>P24y0)ZApp%-MNkDcrpn*jAD6ohi;eS_*HRZ!U$mF9vp0<nfuabIDS8(|mU+
zykjx2bBg=IKn26or)OK{9xv5*Ee1B1Ro4h^2BNcP=IKilr9kHtx0btW=%iryNU64c
z%6PN3ewq5r8Cr5SmYj{Vqw~hMeD7?#-ZXH-dEjm%<!-&(f)|5ru_kcjyM|tF?<VSf
z$DRmu*EX5r9^JL=j<{X-Ax8uKp<N&MENrdYOncfu{a{Rg|4qZikZ-+~aoKXlHLV;!
z>IQX0=1AoJk=F!g_)aLS$G~32w?juD>pmDhX6OJuWyJ5aRV^%ppJQmadG($A|7JT-
z(te@YRZWRRDwC9?L_*@0t-}w_l~dmjK{DZk$971o9-c@Gko7ktpyEl(iNu@uPxWs=
zKKyR<B;zKL3=*)7SSPh3dI{4_k=m6frgD-f_vSN@(n&T!FXNp}>*!bUlTH5hU$H{f
za3Yx#GNMFM6xp)YVBeRT*3;MUADaGY&~l{?mzXMCYQD1dWnxt`{Sr{Bp<Hgnbm&zw
z@Eif+NQq=qFjn5+Mf`e8GyPZmmzEg#P#62#&Xolt;ttgk?>YGJKwsjqeSBZVR=H^S
zKU=TiU~2>YcihtoRnWpz>m&3-uMrbmDxZ^+P$a54I8$^9WM~!u$|G%QKPQ6k5O@RN
z{xkNsZK3~#2is18y<fMf`pGI#xI@yi7ENN#va!`A6Gy7X;Y|K`GIK<$a4OtUG)<<h
zmUyJBTMcmN7lfS!P}yw$j2PqYTieQhaaC6<9mTu4x|lDi1-Yx00jIsVrF9><Cuo7d
zMFM2o=xqWY0;sm-E0z*7nM5K+qo06OE0%VbPl4Q9MBta$y|wiSIarMahm*iuhB=^O
z^cs<N=hl~t%>6_$N1zr!)#paCg>jY16;xg7Oeczi5MS9kh6lShOuk7ptOGrJ`UVp5
zzI}V+hs%!HNTvq~kcg?(%35Ws6`|!iki>oTECEbFTlqRdX^V~mrE(8s^TJ3*d<fkH
zmYkM!1>mlZrs;bIigw}ubUS>XQ0`Bu_7c_pIo0tgwc`f0<5OzOBDDn%ZR=+C(fG&P
cV!;P!_q~T1j;{N{o1p2&so^gv(rf?vFQjj6%m4rY

diff --git a/Modules/data_collector.py b/Modules/data_collector.py
index 5f9df55..0f598df 100644
--- a/Modules/data_collector.py
+++ b/Modules/data_collector.py
@@ -6,13 +6,22 @@ import re
 import sys
 import numpy as np
 import cv2
+
+# Vision-related Imports
 import pytesseract
+import easyocr
+import torch
+
 from PIL import Image, ImageGrab, ImageFilter
 
 from PyQt5.QtWidgets import QApplication, QWidget
 from PyQt5.QtCore import QRect, QPoint, Qt, QMutex, QTimer
 from PyQt5.QtGui import QPainter, QPen, QColor, QFont
 
+# Initialize EasyOCR with CUDA support
+reader_cpu = easyocr.Reader(['en'], gpu=False)
+reader_gpu = easyocr.Reader(['en'], gpu=True if torch.cuda.is_available() else False)
+
 pytesseract.pytesseract.tesseract_cmd = r"C:\Program Files\Tesseract-OCR\tesseract.exe"
 
 DEFAULT_WIDTH = 180
@@ -29,11 +38,11 @@ app_instance = None
 def _ensure_qapplication():
     """
     Ensures that QApplication is initialized before creating widgets.
+    Must be called from the main thread.
     """
     global app_instance
-    if QApplication.instance() is None:
-        app_instance = QApplication(sys.argv)
-        threading.Thread(target=app_instance.exec_, daemon=True).start()
+    if app_instance is None:
+        app_instance = QApplication(sys.argv)  # Start in main thread
 
 
 def create_ocr_region(region_id, x=250, y=50, w=DEFAULT_WIDTH, h=DEFAULT_HEIGHT, color=(255, 255, 0)):
@@ -102,11 +111,11 @@ def _preprocess_image(image):
     return thresh.filter(ImageFilter.MedianFilter(3))
 
 
-def find_word_positions(region_id, word, offset_x=0, offset_y=0, margin=5):
+def find_word_positions(region_id, word, offset_x=0, offset_y=0, margin=5, ocr_engine="CPU"):
     """
     Finds positions of a specific word within the OCR region.
     Applies user-defined offset and margin adjustments.
-    Returns a list of bounding box coordinates relative to the OCR box.
+    Uses Tesseract (CPU) or EasyOCR (GPU) depending on the selected engine.
     """
     collector_mutex.lock()
     if region_id not in regions:
@@ -136,28 +145,36 @@ def find_word_positions(region_id, word, offset_x=0, offset_y=0, margin=5):
         scale_x = orig_width / proc_width
         scale_y = orig_height / proc_height
 
-        data = pytesseract.image_to_data(processed, config='--psm 6 --oem 1', output_type=pytesseract.Output.DICT)
-
         word_positions = []
-        for i in range(len(data['text'])):
-            if re.search(rf"\b{word}\b", data['text'][i], re.IGNORECASE):
-                # Scale the detected coordinates back to region-relative positions
-                x_scaled = int(data['left'][i] * scale_x)
-                y_scaled = int(data['top'][i] * scale_y)
-                w_scaled = int(data['width'][i] * scale_x)
-                h_scaled = int(data['height'][i] * scale_y)
 
-                # Apply user-configured margin
-                x_margin = max(0, x_scaled - margin)
-                y_margin = max(0, y_scaled - margin)
-                w_margin = w_scaled + (margin * 2)
-                h_margin = h_scaled + (margin * 2)
+        if ocr_engine == "CPU":
+            # Use Tesseract (CPU)
+            data = pytesseract.image_to_data(processed, config='--psm 6 --oem 1', output_type=pytesseract.Output.DICT)
 
-                # Apply user-configured offset
-                x_final = x_margin + offset_x
-                y_final = y_margin + offset_y
+            for i in range(len(data['text'])):
+                if re.search(rf"\b{word}\b", data['text'][i], re.IGNORECASE):
+                    x_scaled = int(data['left'][i] * scale_x)
+                    y_scaled = int(data['top'][i] * scale_y)
+                    w_scaled = int(data['width'][i] * scale_x)
+                    h_scaled = int(data['height'][i] * scale_y)
 
-                word_positions.append((x_final, y_final, w_margin, h_margin))
+                    word_positions.append((x_scaled + offset_x, y_scaled + offset_y, w_scaled + (margin * 2), h_scaled + (margin * 2)))
+
+        else:
+            # Use EasyOCR (GPU) - Convert PIL image to NumPy array
+            image_np = np.array(processed)
+            results = reader_gpu.readtext(image_np)
+
+            for (bbox, text, _) in results:
+                if re.search(rf"\b{word}\b", text, re.IGNORECASE):
+                    (x_min, y_min), (x_max, y_max) = bbox[0], bbox[2]
+
+                    x_scaled = int(x_min * scale_x)
+                    y_scaled = int(y_min * scale_y)
+                    w_scaled = int((x_max - x_min) * scale_x)
+                    h_scaled = int((y_max - y_min) * scale_y)
+
+                    word_positions.append((x_scaled + offset_x, y_scaled + offset_y, w_scaled + (margin * 2), h_scaled + (margin * 2)))
 
         return word_positions
     except Exception as e:
@@ -165,9 +182,6 @@ def find_word_positions(region_id, word, offset_x=0, offset_y=0, margin=5):
         return []
 
 
-
-
-
 def draw_identification_boxes(region_id, positions, color=(0, 0, 255)):
     """
     Draws non-interactive rectangles at specified positions within the given OCR region.
diff --git a/Nodes/General Purpose/__pycache__/identification_overlay.cpython-312.pyc b/Nodes/General Purpose/__pycache__/identification_overlay.cpython-312.pyc
index 8fbc8dd0ce6dceebd89d397eede6ae2c9749d523..681be49ec0bcdd97f04baea83b80522c2d24443a 100644
GIT binary patch
delta 1432
zcmYjRO>7%Q6rS;V*R#9!#;)TePEy-7Z4$Rp>ZCtOqzWjO2mzWl{Xw)K7ULbK+uB|;
zyJ_plRtxH(NJu4`OAs7-Kq$yTwCaH#5ES)BFNsoUv{E@$NZeWyP$dp9v#ASncz*NV
z_r3S_y*K_Wxth}MhC&Kp2mk#2%n!kNtqZ<7-@mWd2PS|aT?R$~3r&Fo062wez6_e%
zn7{JVD^;shGAlEwBEj#^;!2@5n95g-l#NTc;G_=cj;3&B#;o8W<#$e<%0L&rl(+Ge
zWniXD)5py+CK*Vt!xi`@{m^&UUk7JodczluYet(9HX<{=r7f-05HPRGz5SXIZu)pk
z03a};b)ZX|HvpWrmUbpgA1N^y<c!XCBW83MokrJEy!jLh8I}&Jy;}1r!%Gn3@KARX
zZW<Y=lRg%rp<pX4E24`={88BL$*o?J1iis9&o~UTjahpAHIV6{SN-$b6{|q>&33r{
z<FzVwh1`)7t}t=r1Z#z8I=F1pq<E|iu{@cGE84SFOh|x!ASQNddBe~PR{1?kFXPIr
z8?>>bSBb?CPR*4Go-wV8ZW=_SO>qSJ>969K(smYTE=|L}cz8(H&5G&hdX(903DTR=
z*~}UW-bXzvsAu8z#lBmy+u|*48TH&l6P_c!g5nE<i{d>rbTbk797+*JCEfHk%C-$M
zl@;jZiIWk?F#QJ&N_)6FONZrAxQm+d4t3-Ud-1?(|JZ7BoPH`FRrz=E9a{el(r*HN
zv{#v+2bFgEwIcf(e)_I*XjE)9y*1mn*^C=vLz00LU{vqag%+w2s6!jp5PhhOM#Q>Q
z57hmw*)|Z}sqPy_hP<Tww?8%X9B{l%=b^R(g4p&*tt=<O$3^;y+7k&h0!%-99$XMk
zg7Yw=((lxl!<<J#TxeWGX($+<>}6^Ok~IHO$bRm~GT{p4e3guGO|+{e(;?&BC2~Ww
zXjTmUyhRMXYT2g4MrV@&Y6X)QWirfRJd>+h#UdZGgD*I)TFw*plDrGyLv+zo_l0L#
z&8y4ZVqk|oKm5Pww#*o*)*;@gb~Dr5z@&jIj#)Gdd0qxh_*8mWMj`7QCZ&9Blws^A
zv_CW&UW>)=$NE-ceb?%%vAtIm`c7zi|5`A7b?T$3h1V9lu8-Y%<Myd_!Iw~$QTK)d
z<j(sjyMnUU+rLfRNGu;bw)|G{7c}!^V+Va0+LP``=QQx+xSC6YJDQeD3U?Ba%e0!y
z3U_wHT$YY&xr=;<y@nEAuN%_!LMd<C1zalWI_vuguaLjG7dO9(ZwcXo{|VSznRs#;
zti@iqG_@8UxHP%p?+^$6$SUHlzq+=Gry<$f!e?YRpzDTJ&~?{Gu&W+sotdD2Yr%8D
zBeiK~-_W#^vj`@B#&83NEMvBWZEv7nzOTS$EZ$oe9VmPL=gA3n<>lD}Ogs`Ggpd6q
e^sj3GDi1*F0Z2Xo=?!TT!qe~?`V)AH7wKR9-)oWp

delta 1253
zcmYLIO>7%Q6rS<^?|S`%lQeOg*hxW&^W&6+PyrF37=c=7R7VKZP*^RQjl1sJ>&|Y7
zH*%ndp7;@pMx3aC69PH6a%rVjv^{VrNUm~;R*E>HN>l<G5I2~y<7N);eeb<*-n@D9
zW+(e)CU#HPRY2FxUv8ClwXIk`T;3X;n?ewYj$Bcem5REoLJ$G30VG`kNVd2p$2I`4
ztOb#rvhh$Q(~-Wd&#%2_R#CR*I*wf}WmmCvY0avx1X&bR&5FHZI!=&RAFz#!IT*?d
zre`g>$f8&)p0Pa7ELjU~otk3tO}k>@TqKMl(>INZS*0;dK8E+9OwLE{rBSQ_44$Q8
z%hx&@fO3><alI3A*FMxJjuI$|o<OPXo~GFGL%NfEr5z)AO?4WeoIIp=5$bKi$K+Jk
z+dKfYh<5LoN%!}3<A=K3cZj>FJiv;CX%g@m@;;wp$MXTm^^xoR7IB58I7^y~hl00O
zvoKajOBjkO*ecm>)vytkNWVA(1@gN1tvF1n8zbbFo{k%aUA29~U}boFfMlhM*M5+$
zLXp+|0Fk$p6Y)_BX<pyNevZSF<O^j|%TphopkiZ&<{2grl_VS||0z#tv)kNvZ|sac
zx089EOsh+(L`m%JAKf9Js8w=QJ56*=Ni-wWN7s3(H~CG0e5f6f#0K<WAd#!u$t2$p
z8e&uEur?)MC-=Ey<c4<SX$i^ON_U<h2&Ch{f85e10hS>^LO1U@@To;kM~^1OO_AdA
zI;eB!z&gyS<b&v|T9oF;aV8Sv=jgzx0Sa>vXP8p)3=6PZhFrz0;b#~UyqaVCc#ehG
z-NMwWT}8&ai;+=tJ=>=Z_waF&*E283c#_fBA))53u6mYl_*UHywTg*LcC};32a$T%
zMFFkmGrd2>W-@!QaJ%K><hnj3K1==HZ{&CVB$<sB(?9Y*#cs!1XWwbPYqd&l>(ctp
zV7(=$_eJt`EDeXrjo9pnFtRWYZjWh&1h_LFE9l(aC}c9B6^6LGgRn3};_<>oHXH3L
z<8fBCZWt?$>3J)b;}{0*%oIDCjrYr)f)vJQBEbTKRQ!7d?DoENd2u%}cKOsk-y@9u
zHK_^{|MU+D7a*SQ@^evgGoJk@el{rjbNQnGwv9@b?_u^s!)VuAaIuBi5QNeh7p*y#
zhuKGLcfNh>_n?!KbtjbPD{az?c!`><N{_wX9~^}6PhNohqZokd9?0&2(LIp)6`bhZ
R7hi$!0{lk)8?-??+yCM^BFz8*

diff --git a/Nodes/General Purpose/identification_overlay.py b/Nodes/General Purpose/identification_overlay.py
index 8bccf67..1f70b00 100644
--- a/Nodes/General Purpose/identification_overlay.py	
+++ b/Nodes/General Purpose/identification_overlay.py	
@@ -3,14 +3,13 @@
 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, margin, and polling frequency dynamically.
+- Users can configure offset, margin, polling frequency, and select OCR engine.
 """
 
 import re
 from OdenGraphQt import BaseNode
-from PyQt5.QtWidgets import QMessageBox
 from PyQt5.QtCore import QTimer
-from Modules import data_manager, data_collector
+from Modules import data_collector
 
 
 class IdentificationOverlayNode(BaseNode):
@@ -24,7 +23,9 @@ class IdentificationOverlayNode(BaseNode):
         self.add_text_input("search_term", "Search Term", text="Aibatt")
         self.add_text_input("offset_value", "Offset Value (X,Y)", text="0,0")  # X,Y Offset
         self.add_text_input("margin", "Margin", text="5")  # Box Margin
-        self.add_text_input("polling_freq", "Polling Frequency (ms)", text="500")  # New input
+        self.add_text_input("polling_freq", "Polling Frequency (ms)", text="500")  # Polling Rate
+        self.add_combo_menu("ocr_engine", "Type", items=["CPU", "GPU"])
+        self.set_property("ocr_engine", "CPU")  # Set default value after adding the menu
 
         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))
@@ -45,7 +46,7 @@ class IdentificationOverlayNode(BaseNode):
         """
         polling_text = self.get_property("polling_freq")
         try:
-            polling_interval = max(100, int(polling_text))  # Minimum 100ms to avoid overloading
+            polling_interval = max(50, int(polling_text))  # Minimum 50ms for near real-time
         except ValueError:
             polling_interval = 500  # Default to 500ms
 
@@ -58,6 +59,7 @@ class IdentificationOverlayNode(BaseNode):
         search_term = self.get_property("search_term")
         offset_text = self.get_property("offset_value")
         margin_text = self.get_property("margin")
+        ocr_engine = self.get_property("ocr_engine")
 
         # Read and apply polling frequency updates
         self.update_polling_frequency()
@@ -77,8 +79,10 @@ class IdentificationOverlayNode(BaseNode):
         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)
+        # Get detected word positions using the selected OCR engine
+        detected_positions = data_collector.find_word_positions(
+            self.region_id, search_term, offset_x, offset_y, margin, ocr_engine
+        )
 
         # Draw detected word boxes
         data_collector.draw_identification_boxes(self.region_id, detected_positions, color=(0, 0, 255))