Update main_process/main_router.py

main_process/main_router.py

@@ -100,8 +100,8 @@ def generate_srt_from_segments(segments: List[Dict], sha1: str) -> str:
     video_root = MEDIA_ROOT / sha1
     video_root.mkdir(parents=True, exist_ok=True)
 
-    # Path: /data/media/<sha1>/srt
-    srt_dir = video_root / "srt"
+    # Path: /data/media/<sha1>/origin_srt
+    srt_dir = video_root / "initial_srt"
     srt_dir.mkdir(parents=True, exist_ok=True)
 
     # Delete old SRT files
@@ -308,5 +308,29 @@ async def pipeline_video_analysis(
     # Generate the final SRT subtitle file
     srt = generate_srt_from_segments(info_clips, sha1)
 
-    # The endpoint returns the SRT file as plain text
-    return srt
+    # Create result JSON
+    result_json = {
+            "full_transcription": full_transcription,
+            "info_escenas": info_escenas,
+            "info_clips": info_clips
+    }
+
+    # Path: /data/media/<sha1>
+    video_root = MEDIA_ROOT / sha1
+    video_root.mkdir(parents=True, exist_ok=True)
+
+    # Path: /data/media/<sha1>/origin_srt
+    srt_dir = video_root / "initial_srt"
+    srt_dir.mkdir(parents=True, exist_ok=True)
+
+    final_path = srt_dir / "initial_info.json"
+    
+    with final_path.open("w", encoding="utf-8") as f:
+        json.dump({
+            "full_transcription": full_transcription,
+            "info_escenas": info_escenas,
+            "info_clips": info_clips
+        }, f, ensure_ascii=False, indent=4)
+
+    # The endpoint returns OK message info
+    return {"status": "ok", "message": "Initial SRT and info JSON generated"}