Update main_process/main_router.py

main_process/main_router.py

@@ -238,7 +238,7 @@ def pipeline_preprocessing_audio(video_path: str, voice_col):
 
     return full_transcription, diarization_info
 
-@router.post("/generate_srt_salamandra", tags=["Transcription Process"])
+@router.post("/generate_srt", tags=["Transcription Process"])
 async def pipeline_video_analysis(
     sha1: str,
     token: str = Query(..., description="Token required for authorization")
@@ -333,4 +333,82 @@ async def pipeline_video_analysis(
         }, f, ensure_ascii=False, indent=4)
 
     # The endpoint returns OK message info
-    return {"status": "ok", "message": "Initial SRT and info JSON generated"}
+    return {"status": "ok", "message": "Initial SRT and info JSON generated"}
+
+def get_initial_info_path(sha1:str):
+    video_root = MEDIA_ROOT / sha1
+    srt_dir = video_root / "initial_srt"
+    final_path = srt_dir / "initial_info.json"
+
+    if not video_root.exists() or not video_root.is_dir():
+        raise HTTPException(status_code=404, detail="SHA1 folder not found")
+    if not srt_dir.exists() or not srt_dir.is_dir():
+        raise HTTPException(status_code=404, detail="initial_srt folder not found")
+    if not final_path.exists() or not final_path.is_file():
+        raise HTTPException(status_code=404, detail="initial_info JSON not found")
+
+    return final_path
+
+def get_initial_srt_path(sha1:str):
+    video_root = MEDIA_ROOT / sha1
+    srt_dir = video_root / "initial_srt"
+    final_path = srt_dir / "initial.srt"
+
+    if not video_root.exists() or not video_root.is_dir():
+        raise HTTPException(status_code=404, detail="SHA1 folder not found")
+    if not srt_dir.exists() or not srt_dir.is_dir():
+        raise HTTPException(status_code=404, detail="initial_srt folder not found")
+    if not final_path.exists() or not final_path.is_file():
+        raise HTTPException(status_code=404, detail="initial.srt SRT not found")
+
+    return final_path
+    
+@router.get("/download_initial_srt", tags=["Transcription Process"])
+def download_initial_srt(
+    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)
+    
+    file_path = get_initial_srt_path(sha1)
+
+    return FileResponse(
+        path=file_path,
+        media_type="text/srt",
+        filename="initial.srt"
+    )
+
+@router.get("/download_initial_info", tags=["Transcription Process"])
+def download_initial_info(
+    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)
+    
+    file_path = get_initial_info_path(sha1)
+
+    return FileResponse(
+        path=file_path,
+        media_type="text/json",
+        filename="initial_info.json"
+    )