Update main_process/main_router.py

main_process/main_router.py

@@ -19,7 +19,7 @@ from storage.embeddings_routers import get_embeddings_json
 EMBEDDINGS_ROOT = Path("/data/embeddings")
 MEDIA_ROOT = Path("/data/media")
 os.environ["CUDA_VISIBLE_DEVICES"] = "1"
-router = APIRouter(prefix="/transcription", tags=["Transcription Process"])
+router = APIRouter(prefix="/transcription", tags=["Initial Transcription Process"])
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 def get_casting(video_sha1: str):
@@ -238,7 +238,7 @@ def pipeline_preprocessing_audio(video_path: str, voice_col):
 
     return full_transcription, diarization_info
 
-@router.post("/generate_srt", tags=["Transcription Process"])
+@router.post("/generate_srt", tags=["Initial Transcription Process"])
 async def pipeline_video_analysis(
     sha1: str,
     token: str = Query(..., description="Token required for authorization")
@@ -363,7 +363,7 @@ def get_initial_srt_path(sha1:str):
 
     return final_path
     
-@router.get("/download_initial_srt", tags=["Transcription Process"])
+@router.get("/download_initial_srt", tags=["Initial Transcription Process"])
 def download_initial_srt(
     sha1: str,
     token: str = Query(..., description="Token required for authorization")
@@ -388,7 +388,7 @@ def download_initial_srt(
         filename="initial.srt"
     )
 
-@router.get("/download_initial_info", tags=["Transcription Process"])
+@router.get("/download_initial_info", tags=["Initial Transcription Process"])
 def download_initial_info(
     sha1: str,
     token: str = Query(..., description="Token required for authorization")