Update vision_analyzer.py

vision_analyzer.py

@@ -43,7 +43,7 @@ MIN_FREE_SPACE_GB = 2  # Minimum free space in GB before processing
 DEFAULT_FPS = 3 # Default frames per second for extraction
 
 # Cursor Tracking Parameters
-CURSOR_THRESHOLD = 0.8
+
 
 # Initialize HF API
 hf_api = HfApi(token=HF_TOKEN)
@@ -57,8 +57,7 @@ processing_status = {
     "failed_files": 0,
     "extracted_courses": 0,
     "extracted_videos": 0,
-    "extracted_frames_count": 0,
-    "tracked_cursors_count": 0,
+: 0,
     "last_update": None,
     "logs": []
 }
@@ -263,21 +262,7 @@ def extract_frames(video_path, output_dir, fps=DEFAULT_FPS):
     log_message(f"Extracted {saved_idx-1} frames from {video_path} to {output_dir}")
     return saved_idx - 1
 
-# --- Cursor Tracking Utilities ---
-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:
-        # Use alpha channel as mask (nonzero alpha = 255)
-        return (template_img[:, :, 3] > 0).astype(np.uint8) * 255
-    return None
 
 
 
@@ -331,11 +316,7 @@ def process_rar_file(rar_path: str) -> bool:
                 processing_status["extracted_videos"] += 1
                 log_message(f"[INFO] Extracted {extracted_frames_count} frames from {video_filename}")
 
-                # Perform cursor tracking on the extracted frames
-                cursor_output_json = os.path.join(CURSOR_TRACKING_OUTPUT_FOLDER, f"{course_name}_{video_filename.replace('.', '_')}_cursor_data.json")
-                tracked_cursors = track_cursor(frames_output_dir, CURSOR_TEMPLATES_DIR, cursor_output_json, threshold=CURSOR_THRESHOLD)
-                processing_status["tracked_cursors_count"] += tracked_cursors
-                log_message(f"[INFO] Tracked {tracked_cursors} cursors in frames from {video_filename}")
+
             else:
                 log_message(f"[WARN] No frames extracted from {video_filename}")
 
@@ -425,12 +406,7 @@ def main_processing_loop(start_index: int = 0):
             next_index += 1
             save_json_state(DOWNLOAD_STATE_FILE, {"next_download_index": next_index})
             
-            # Status update
-            log_message(f"📊 Progress: {next_index}/{len(rar_files)} files processed")
-            log_message(f'📊 Extracted: {processing_status["extracted_courses"]} courses')
-            log_message(f'📊 Videos Processed: {processing_status["extracted_videos"]}')
-            log_message(f'📊 Frames Extracted: {processing_status["extracted_frames_count"]}')
-            
+            # Status update            log_message(f"📊 Frames Extracted: {processing_status["extracted_frames_count"]}")   
             if next_index < len(rar_files):
                 log_message(f"🔄 Run the script again to process the next file: {os.path.basename(rar_files[next_index])}")
             else:
@@ -439,7 +415,7 @@ def main_processing_loop(start_index: int = 0):
             log_message("✅ All files have been processed!")
         
         log_message("🎉 Processing complete!")
-        log_message(f'📊 Final stats: {processing_status["extracted_courses"]} courses extracted, {processing_status["extracted_videos"]} videos processed, {processing_status["extracted_frames_count"]} frames extracted, {processing_status["tracked_cursors_count"]} cursors tracked')
+        log_message(f'📊 Final stats: {processing_status["extracted_courses"]} courses extracted, {processing_status["extracted_videos"]} videos processed, {processing_status["extracted_frames_count"]} frames extracted')
         
     except KeyboardInterrupt:
         log_message("⏹️ Processing interrupted by user")
@@ -460,5 +436,3 @@ __all__ = [
 ]
 
 
-
-