Update download_api.py

download_api.py

@@ -15,12 +15,12 @@ from datetime import datetime
 
 import torch
 
-# Import from vision_analyzer (previously cursor_tracker)
+# Import core functionality
 from vision_analyzer import (
     main_processing_loop,
     processing_status,
     log_message,
-    FRAMES_OUTPUT_FOLDER  # Add this import for frames directory
+    FRAMES_OUTPUT_FOLDER
 )
 
 # FastAPI App Definition
@@ -42,6 +42,7 @@ processing_thread = None
 frame_locks = {}  # Dict to track frame locks: {course: {frame: {"locked_by": id, "locked_at": timestamp}}}
 processed_frames = {}  # Dict to track processed frames: {course: {frame: {"processed_by": id, "processed_at": timestamp}}}
 LOCK_TIMEOUT = 300  # 5 minutes timeout for locks
+TRACKING_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "frame_tracking.json")
 
 def save_tracking_state():
     """Save frame tracking state to disk"""
@@ -50,7 +51,7 @@ def save_tracking_state():
         "processed_frames": processed_frames
     }
     try:
-        with open("frame_tracking.json", "w") as f:
+        with open(TRACKING_FILE, "w") as f:
             json.dump(state, f, indent=2)
     except Exception as e:
         log_message(f"Error saving tracking state: {e}")
@@ -59,7 +60,7 @@ def load_tracking_state():
     """Load frame tracking state from disk"""
     global frame_locks, processed_frames
     try:
-        with open("frame_tracking.json", "r") as f:
+        with open(TRACKING_FILE, "r") as f:
             state = json.load(f)
             frame_locks = state.get("frame_locks", {})
             processed_frames = state.get("processed_frames", {})