Update backend/app.py

backend/app.py

@@ -1,104 +1,105 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse, FileResponse
-from fastapi.staticfiles import StaticFiles
-import os
-import shutil
-from pathlib import Path
-import json
-from tracker import VideoTracker
-from utils import save_trajectory_json, create_output_dirs
-
-app = FastAPI(title="3D Trajectory Tracker")
-
-# CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Create necessary directories
-UPLOAD_DIR = Path("uploads")
-OUTPUT_DIR = Path("outputs")
-FRONTEND_DIR = Path("frontend")
-
-create_output_dirs([UPLOAD_DIR, OUTPUT_DIR])
-
-# Mount static files
-if FRONTEND_DIR.exists():
-    app.mount("/static", StaticFiles(directory=FRONTEND_DIR), name="static")
-
-@app.get("/")
-async def read_root():
-    """Serve the main HTML page"""
-    index_path = FRONTEND_DIR / "index.html"
-    if index_path.exists():
-        return FileResponse(index_path)
-    return {"message": "3D Trajectory Tracker API"}
-
-@app.post("/api/upload")
-async def upload_video(file: UploadFile = File(...)):
-    """Upload and process video file"""
-    try:
-        # Validate file type
-        if not file.filename.lower().endswith(('.mp4', '.avi', '.mov', '.mkv')):
-            raise HTTPException(400, "Invalid file type. Please upload a video file.")
-        
-        # Save uploaded file
-        file_path = UPLOAD_DIR / file.filename
-        with open(file_path, "wb") as buffer:
-            shutil.copyfileobj(file.file, buffer)
-        
-        # Process video
-        tracker = VideoTracker(str(file_path))
-        trajectories = tracker.process_video()
-        
-        # Save results
-        output_path = OUTPUT_DIR / f"{Path(file.filename).stem}_trajectories.json"
-        save_trajectory_json(trajectories, output_path)
-        
-        # Clean up uploaded file
-        os.remove(file_path)
-        
-        return JSONResponse({
-            "success": True,
-            "trajectories": trajectories,
-            "output_file": str(output_path)
-        })
-    
-    except Exception as e:
-        raise HTTPException(500, f"Processing error: {str(e)}")
-
-@app.get("/api/trajectories/{filename}")
-async def get_trajectory(filename: str):
-    """Retrieve saved trajectory data"""
-    file_path = OUTPUT_DIR / filename
-    if not file_path.exists():
-        raise HTTPException(404, "Trajectory file not found")
-    
-    with open(file_path, "r") as f:
-        data = json.load(f)
-    
-    return JSONResponse(data)
-
-@app.get("/api/list")
-async def list_trajectories():
-    """List all processed trajectories"""
-    files = [f.name for f in OUTPUT_DIR.glob("*.json")]
-    return JSONResponse({"files": files})
-
-@app.delete("/api/clear")
-async def clear_data():
-    """Clear all uploaded and processed files"""
-    for folder in [UPLOAD_DIR, OUTPUT_DIR]:
-        for file in folder.glob("*"):
-            if file.is_file():
-                os.remove(file)
-    return JSONResponse({"success": True, "message": "All data cleared"})
-
-if __name__ == "__main__":
-    import uvicorn
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse, FileResponse
+from fastapi.staticfiles import StaticFiles
+import os
+import shutil
+from pathlib import Path
+import json
+from tracker import VideoTracker
+from utils import save_trajectory_json, create_output_dirs
+
+app = FastAPI(title="3D Trajectory Tracker")
+
+# CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Create necessary directories
+BASE_DIR = Path(__file__).parent.parent
+UPLOAD_DIR = BASE_DIR / "uploads"
+OUTPUT_DIR = BASE_DIR / "outputs"
+FRONTEND_DIR = BASE_DIR / "frontend"
+
+create_output_dirs([UPLOAD_DIR, OUTPUT_DIR])
+
+# Mount static files
+if FRONTEND_DIR.exists():
+    app.mount("/static", StaticFiles(directory=FRONTEND_DIR), name="static")
+
+@app.get("/")
+async def read_root():
+    """Serve the main HTML page"""
+    index_path = FRONTEND_DIR / "index.html"
+    if index_path.exists():
+        return FileResponse(index_path)
+    return {"message": "3D Trajectory Tracker API"}
+
+@app.post("/api/upload")
+async def upload_video(file: UploadFile = File(...)):
+    """Upload and process video file"""
+    try:
+        # Validate file type
+        if not file.filename.lower().endswith(('.mp4', '.avi', '.mov', '.mkv')):
+            raise HTTPException(400, "Invalid file type. Please upload a video file.")
+        
+        # Save uploaded file
+        file_path = UPLOAD_DIR / file.filename
+        with open(file_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+        
+        # Process video
+        tracker = VideoTracker(str(file_path))
+        trajectories = tracker.process_video()
+        
+        # Save results
+        output_path = OUTPUT_DIR / f"{Path(file.filename).stem}_trajectories.json"
+        save_trajectory_json(trajectories, output_path)
+        
+        # Clean up uploaded file
+        os.remove(file_path)
+        
+        return JSONResponse({
+            "success": True,
+            "trajectories": trajectories,
+            "output_file": str(output_path)
+        })
+    
+    except Exception as e:
+        raise HTTPException(500, f"Processing error: {str(e)}")
+
+@app.get("/api/trajectories/{filename}")
+async def get_trajectory(filename: str):
+    """Retrieve saved trajectory data"""
+    file_path = OUTPUT_DIR / filename
+    if not file_path.exists():
+        raise HTTPException(404, "Trajectory file not found")
+    
+    with open(file_path, "r") as f:
+        data = json.load(f)
+    
+    return JSONResponse(data)
+
+@app.get("/api/list")
+async def list_trajectories():
+    """List all processed trajectories"""
+    files = [f.name for f in OUTPUT_DIR.glob("*.json")]
+    return JSONResponse({"files": files})
+
+@app.delete("/api/clear")
+async def clear_data():
+    """Clear all uploaded and processed files"""
+    for folder in [UPLOAD_DIR, OUTPUT_DIR]:
+        for file in folder.glob("*"):
+            if file.is_file():
+                os.remove(file)
+    return JSONResponse({"success": True, "message": "All data cleared"})
+
+if __name__ == "__main__":
+    import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)