Update download_api.py

download_api.py

@@ -1,20 +1,49 @@
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import FileResponse
-from pathlib import Path
-import os
-
-app = FastAPI()
-
-ANNOTATIONS_DIR = Path("annotations").resolve()
-
-@app.get("/download/{filename}")
-def download_file(filename: str):
-    file_path = ANNOTATIONS_DIR / filename
-    if not file_path.exists() or not file_path.is_file():
-        raise HTTPException(status_code=404, detail="File not found")
-    return FileResponse(str(file_path), filename=filename)
-
-@app.get("/")
-def root():
-    files = [f.name for f in ANNOTATIONS_DIR.glob("*.json")]
-    return {"message": "Use /download/{filename} to download a file.", "available_files": files}
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse
+from pathlib import Path
+import os
+import threading
+import time
+import sys
+sys.path.append(str(Path(__file__).parent / "scripts"))
+from cursor_tracker import track_cursor
+
+app = FastAPI()
+
+ANNOTATIONS_DIR = Path("annotations").resolve()
+
+# Only run once per server start
+cursor_tracking_started = False
+
+def run_cursor_tracker_once():
+    global cursor_tracking_started
+    if cursor_tracking_started:
+        return
+    cursor_tracking_started = True
+    try:
+        # These should match your preset values
+        FRAMES_DIR = 'frames/sample_task'
+        CURSOR_TEMPLATES_DIR = 'cursors'
+        OUTPUT_JSON = 'annotations/test.json'
+        THRESHOLD = 0.8
+        print("[INFO] Running cursor tracking on server startup...")
+        track_cursor(FRAMES_DIR, CURSOR_TEMPLATES_DIR, OUTPUT_JSON, THRESHOLD, start_frame=0, email_results=False)
+    except Exception as e:
+        print(f"[ERROR] Cursor tracking failed: {e}")
+
+@app.on_event("startup")
+def startup_event():
+    # Run in a thread so server starts immediately
+    threading.Thread(target=run_cursor_tracker_once, daemon=True).start()
+
+@app.get("/download/{filename}")
+def download_file(filename: str):
+    file_path = ANNOTATIONS_DIR / filename
+    if not file_path.exists() or not file_path.is_file():
+        raise HTTPException(status_code=404, detail="File not found")
+    return FileResponse(str(file_path), filename=filename)
+
+@app.get("/")
+def root():
+    files = [f.name for f in ANNOTATIONS_DIR.glob("*.json")]
+    return {"message": "Use /download/{filename} to download a file.", "available_files": files}