Return 404 for missing video posters

app.py

@@ -27,6 +27,7 @@ from dotenv import load_dotenv
 from fastapi import FastAPI, File, Form, HTTPException, Request, UploadFile
 from fastapi.responses import FileResponse, JSONResponse
 from huggingface_hub import HfApi, hf_hub_download, snapshot_download
+from huggingface_hub.errors import RemoteEntryNotFoundError
 
 load_dotenv(Path(__file__).with_name(".env"))
 jobs = {}
@@ -544,6 +545,9 @@ def video_poster(session_id: UUID, filename: str, request: Request):
     row = session_row(session_id, request.state.user_id)
     if Path(filename).name != filename or "\\" in filename or not filename.endswith(".webp"):
         raise HTTPException(404, "Preview not found")
-    path = hf_hub_download(repo_id=os.environ["STORAGE_REPO_ID"], repo_type="dataset",
-        filename=f'{row["path"]}/posters/{filename}', token=os.environ["HF_TOKEN"])
+    try:
+        path = hf_hub_download(repo_id=os.environ["STORAGE_REPO_ID"], repo_type="dataset",
+            filename=f'{row["path"]}/posters/{filename}', token=os.environ["HF_TOKEN"])
+    except RemoteEntryNotFoundError as error:
+        raise HTTPException(404, "Preview not found") from error
     return FileResponse(path, media_type="image/webp", headers={"Cache-Control": "private, max-age=86400"})