Update app.py

app.py

@@ -1,47 +1,40 @@
 import os
-import zipfile
 from fastapi import FastAPI
-from fastapi.responses import FileResponse
+from fastapi.responses import FileResponse, JSONResponse
 from scripts.extract_frames import extract_frames
-from scripts.cursor_tracker import track_cursor
 from contextlib import asynccontextmanager
 
 FRAMES_DIR = "frames"
-ANNOTATIONS_DIR = "annotations"
-FRAMES_ZIP = "frames.zip"
-ANNOTATIONS_ZIP = "annotations.zip"
+VIDEO_DIR = "videos"
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    video_dir = "videos"
-    cursor_dir = "cursors"
     os.makedirs(FRAMES_DIR, exist_ok=True)
-    os.makedirs(cursor_dir, exist_ok=True)
-    os.makedirs(ANNOTATIONS_DIR, exist_ok=True)
-    # Step 1: Extract frames
-    for video_file in os.listdir(video_dir):
+    
+    # Step 1: Extract frames from each video
+    for video_file in os.listdir(VIDEO_DIR):
         if video_file.lower().endswith(('.mp4', '.avi', '.mov')):
-            extract_frames(os.path.join(video_dir, video_file), FRAMES_DIR)
-    # Step 2: Track cursor
-    track_cursor(FRAMES_DIR, cursor_dir, os.path.join(ANNOTATIONS_DIR, "cursor_positions.json"))
-    # Step 3: Zip frames and annotations
-    with zipfile.ZipFile(FRAMES_ZIP, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        for root, _, files in os.walk(FRAMES_DIR):
-            for file in files:
-                zipf.write(os.path.join(root, file), arcname=file)
-    with zipfile.ZipFile(ANNOTATIONS_ZIP, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        for root, _, files in os.walk(ANNOTATIONS_DIR):
-            for file in files:
-                zipf.write(os.path.join(root, file), arcname=file)
-    print("Pipeline complete and outputs zipped.")
+            video_path = os.path.join(VIDEO_DIR, video_file)
+            extract_frames(video_path, FRAMES_DIR)
+    
+    print("✅ Frame extraction complete.")
     yield
 
 app = FastAPI(lifespan=lifespan)
 
-@app.get("/download/frames")
-def download_frames():
-    return FileResponse(FRAMES_ZIP, filename=FRAMES_ZIP)
+# Endpoint to list all available frames
+@app.get("/frames")
+def list_frames():
+    try:
+        frames = [f for f in os.listdir(FRAMES_DIR) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+        return JSONResponse(content={"frames": frames})
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
 
-@app.get("/download/annotations")
-def download_annotations():
-    return FileResponse(ANNOTATIONS_ZIP, filename=ANNOTATIONS_ZIP)
+# Endpoint to download a specific frame
+@app.get("/frames/{filename}")
+def get_frame(filename: str):
+    file_path = os.path.join(FRAMES_DIR, filename)
+    if os.path.exists(file_path):
+        return FileResponse(file_path, filename=filename)
+    return JSONResponse(content={"error": "File not found"}, status_code=404)