Update main_process/moe_router.py

main_process/moe_router.py

@@ -357,6 +357,71 @@ class Valoracion_Final:
 
         return state
 
+def srt_update(srt_video, srt_video_modified):
+    with open(srt_video, "r", encoding="utf-8") as f:
+        srt_text = f.read()
+
+    srt_blocks = []
+    srt_blocks_modified = []
+
+    pattern = re.compile(
+        r"(\d+)\s+(\d{2}:\d{2}:\d{2},\d{3}) --> (\d{2}:\d{2}:\d{2},\d{3})\s+(.*?)(?=\n\d+\n|\Z)",
+        re.S
+    )
+
+    for match in pattern.finditer(srt_text):
+        srt_blocks.append({
+            "index": int(match.group(1)),
+            "start": match.group(2),
+            "end": match.group(3),
+            "text": match.group(4).strip()
+        })
+
+    for block in srt_blocks:
+        original_text = block["text"]
+
+        match_name = re.match(r'\[([^\]]+)\]:\s*(.*)', original_text)
+        if match_name:
+            name = match_name.group(1).upper()
+            content = match_name.group(2)
+
+            srt_blocks_modified.append({
+                "index": block["index"],
+                "start": block["start"],
+                "end": block["end"],
+                "text": f"{name}: {content}"
+            })
+            continue
+
+        match_ad = re.match(r'\(AD\):\s*(.*)', original_text)
+        if match_ad:
+            content = match_ad.group(1)
+
+            content = content.strip()
+            if content.startswith('"') and content.endswith('"'): # quitamos las ""
+                content = content[1:-1]
+
+            srt_blocks_modified.append({
+                "index": block["index"],
+                "start": block["start"],
+                "end": block["end"],
+                "text": f"(AD) {content}"
+            })
+            continue
+
+        srt_blocks_modified.append(block)
+
+    srt_final = ""
+    for block in srt_blocks_modified:
+        srt_final += (
+            f"{block['index']}\n"
+            f"{block['start']} --> {block['end']}\n"
+            f"{block['text']}\n\n"
+        )
+
+    with open(srt_video_modified, "w", encoding="utf-8") as f:
+        f.write(srt_final)
+
 @router.post("/generate_moe_result", tags=["MoE Process"])
 async def generate_salamadra_result(
     sha1: str,
@@ -480,6 +545,8 @@ async def generate_salamadra_result(
     state = valoracion_final(state, srt_final, csv_evaluacion)
     tmp.close()
 
+    srt_update(srt_final,srt_final)
+    
     return {"status": "ok", "message": "Salamandra SRT, free_narration and CSV evaluation generated"}
 
 @router.get("/download_moe_srt", tags=["MoE Process"])