Update api.py

api.py

@@ -67,7 +67,91 @@ def validate_token(token: str):
     if token != HF_TOKEN:
         raise HTTPException(status_code=401, detail="Invalid token")
 
-@app.post("/upload_video", tags=["Media Manager"])
+
+@app.post("/upload_cast_csv/{sha1}", tags=["Media Manager"])
+async def upload_cast_csv(
+    sha1: str,
+    cast_file: UploadFile = File(...),
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Upload a cast CSV file for a specific video identified by its SHA-1.
+
+    The CSV will be stored under:
+        /data/media/<sha1>/cast/cast.csv
+
+    Steps:
+    - Validate the token.
+    - Ensure /data/media/<sha1> exists.
+    - Create /cast folder if missing.
+    - Save the CSV file inside /cast.
+    """
+    validate_token(token)
+
+    base_folder = MEDIA_ROOT / sha1
+    if not base_folder.exists() or not base_folder.is_dir():
+        raise HTTPException(status_code=404, detail="SHA1 folder not found")
+
+    cast_folder = base_folder / "cast"
+    cast_folder.mkdir(parents=True, exist_ok=True)
+
+    final_path = cast_folder / "cast.csv"
+
+    file_bytes = await cast_file.read()
+    save_result = file_manager.upload_file(io.BytesIO(file_bytes), final_path)
+    if not save_result["operation_success"]:
+        raise HTTPException(status_code=500, detail=save_result["error"])
+
+    return JSONResponse(
+        status_code=200,
+        content={"status": "ok", "saved_to": str(final_path)}
+    )
+
+
+@app.get("/download_cast_csv/{sha1}", tags=["Media Manager"])
+def download_cast_csv(
+    sha1: str,
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Download the cast CSV for a specific video identified by its SHA-1.
+
+    The CSV is expected under:
+        /data/media/<sha1>/cast/cast.csv
+
+    Steps:
+    - Validate the token.
+    - Ensure /data/media/<sha1> and /cast exist.
+    - Return the CSV as a FileResponse.
+    - Raise 404 if any folder or file is missing.
+    """
+    validate_token(token)
+
+    base_folder = MEDIA_ROOT / sha1
+    cast_folder = base_folder / "cast"
+    csv_path = cast_folder / "cast.csv"
+
+    if not base_folder.exists() or not base_folder.is_dir():
+        raise HTTPException(status_code=404, detail="SHA1 folder not found")
+    if not cast_folder.exists() or not cast_folder.is_dir():
+        raise HTTPException(status_code=404, detail="Cast folder not found")
+    if not csv_path.exists() or not csv_path.is_file():
+        raise HTTPException(status_code=404, detail="Cast CSV not found")
+
+    # Convert to relative path for FileManager
+    relative_path = csv_path.relative_to(MEDIA_ROOT)
+    handler = file_manager.get_file(relative_path)
+    if handler is None:
+        raise HTTPException(status_code=404, detail="Cast CSV not accessible")
+    handler.close()
+
+    return FileResponse(
+        path=csv_path,
+        media_type="text/csv",
+        filename="cast.csv"
+    )
+
+@app.post("/upload_original_video", tags=["Media Manager"])
 async def upload_video(
     video: UploadFile = File(...),
     token: str = Query(..., description="Token required for authorization")
@@ -128,7 +212,7 @@ async def upload_video(
     )
 
 
-@app.get("/download_video/{sha1}", tags=["Media Manager"])
+@app.get("/download_original_video/{sha1}", tags=["Media Manager"])
 def download_video(
     sha1: str,
     token: str = Query(..., description="Token required for authorization")
@@ -166,7 +250,7 @@ def download_video(
         - 404 if the file cannot be retrieved using FileManager.
     """
     validate_token(token)
-    
+
     sha1_folder = MEDIA_ROOT / sha1
     clip_folder = sha1_folder / "clip"