Update cursor_tracker.py

cursor_tracker.py

@@ -1,7 +1,79 @@
+"""
+Cursor detection and action logging utilities.
+"""
+
+import os
+import json
+import cv2
+import numpy as np  # ✅ FIXED MISSING IMPORT
+from pathlib import Path
+from email.message import EmailMessage
+import smtplib
 from typing import List, Tuple
-from PIL import Image
-from io import BytesIO
-import requests
+
+def to_rgb(img):
+    if img is None:
+        return None
+    if len(img.shape) == 2:
+        return cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
+    if img.shape[2] == 4:
+        return cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)
+    return img
+
+def get_mask_from_alpha(template_img):
+    if template_img is not None and len(template_img.shape) == 3 and template_img.shape[2] == 4:
+        return (template_img[:, :, 3] > 0).astype(np.uint8) * 255
+    return None
+
+def detect_cursor_in_frame_multi(frame, cursor_templates, threshold=0.8):
+    best_pos = None
+    best_conf = -1
+    best_template_name = None
+    frame_rgb = to_rgb(frame)
+    for template_name, cursor_template in cursor_templates.items():
+        template_rgb = to_rgb(cursor_template)
+        mask = get_mask_from_alpha(cursor_template)
+        if template_rgb is None or frame_rgb is None or template_rgb.shape[2] != frame_rgb.shape[2]:
+            print(f"[WARN] Skipping template {template_name} due to channel mismatch or load error.")
+            continue
+        try:
+            result = cv2.matchTemplate(frame_rgb, template_rgb, cv2.TM_CCOEFF_NORMED, mask=mask)
+        except Exception as e:
+            print(f"[WARN] matchTemplate failed for {template_name}: {e}")
+            continue
+        _, max_val, _, max_loc = cv2.minMaxLoc(result)
+        if max_val > best_conf:
+            best_conf = max_val
+            if max_val >= threshold:
+                cursor_w, cursor_h = template_rgb.shape[1], template_rgb.shape[0]
+                cursor_x = max_loc[0] + cursor_w // 2
+                cursor_y = max_loc[1] + cursor_h // 2
+                best_pos = (cursor_x, cursor_y)
+                best_template_name = template_name
+    if best_conf >= threshold:
+        return best_pos, best_conf, best_template_name
+    return None, best_conf, None
+
+def ensure_dir(path):
+    os.makedirs(path, exist_ok=True)
+
+def send_email_with_attachment(subject, body, to_email, from_email, app_password, attachment_path):
+    msg = EmailMessage()
+    msg['Subject'] = subject
+    msg['From'] = from_email
+    msg['To'] = to_email
+    msg.set_content(body)
+    with open(attachment_path, 'rb') as f:
+        file_data = f.read()
+        file_name = Path(attachment_path).name
+    msg.add_attachment(file_data, maintype='application', subtype='octet-stream', filename=file_name)
+    try:
+        with smtplib.SMTP_SSL('smtp.gmail.com', 465) as smtp:
+            smtp.login(from_email, app_password)
+            smtp.send_message(msg)
+        print(f"[SUCCESS] Email sent to {to_email}")
+    except Exception as e:
+        print(f"[ERROR] Failed to send email: {e}")
 
 def track_cursor_from_images(
     images: List[Tuple[str, np.ndarray]],
@@ -10,7 +82,7 @@ def track_cursor_from_images(
     threshold=0.8,
     email_results=False
 ):
-    """Detect cursor in a list of in-memory images (filename, image)."""
+    """Detect cursor in a list of in-memory images (filename, image ndarray)."""
 
     cursor_templates_dir = Path(cursor_templates_dir).resolve()
     output_json_path = Path(output_json_path).resolve()