| import os |
| from fastapi import FastAPI |
| from fastapi.responses import FileResponse, JSONResponse |
| from scripts.extract_frames import extract_frames |
| from contextlib import asynccontextmanager |
|
|
| FRAMES_DIR = "frames" |
| VIDEO_DIR = "videos" |
|
|
| @asynccontextmanager |
| async def lifespan(app: FastAPI): |
| os.makedirs(FRAMES_DIR, exist_ok=True) |
| |
| |
| for video_file in os.listdir(VIDEO_DIR): |
| if video_file.lower().endswith(('.mp4', '.avi', '.mov')): |
| 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("/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("/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) |
|
|
