Update main_process/salamandra_router.py

main_process/salamandra_router.py

@@ -32,7 +32,7 @@ from asr_client import (
     identificar_veu
 )
 
-from schat_client import resumir_frases_salamandra, identificar_personajes
+from schat_client import resumir_frases_salamandra, identificar_personajes, free_narration_schat
 
 from storage.common import validate_token
 from storage.files.file_manager import FileManager
@@ -774,7 +774,11 @@ class Identity_Manager:
                             else:
                                 personas += person
 
-                        result = identificar_personajes(content_escena, personas)
+                        if personas == "":
+                            result = content_escena
+                            
+                        else:
+                            result = identificar_personajes(content_escena, personas)
 
                         out = f"(AD_Descripción): {result}"
 
@@ -983,29 +987,10 @@ class Valoracion_Final:
 class Free_Narration:
     def __call__(self, state: NState, audio_descripcion_path_sin_une, story_path) -> NState:
 
-        with open(audio_descripcion_path_sin_une, "r", encoding="utf-8-sig") as f:
-            audio_descripcion = f.read()
-
-        sys_prompt = f"""
-        Ets un relator objectiu. Tens la següent informació:
-
-        1. Audiodescripció del vídeo (incloent diàlegs i descripcions visuals): {audio_descripcion}
-
-        Objectiu:
-        - Resumeix de forma precisa i cronològica tot el que passa al vídeo.
-        - Inclou només els esdeveniments essencials i les accions principals dels personatges.
-        - Elimina qualsevol detall visual, emocional o descriptiu que no sigui necessari per entendre què passa.
-        - No afegeixis cap informació que no aparegui explícitament a la font.
-        - No reprodueixis diàlegs, només explica el que succeeix.
-        - Mantén el relat neutre, breu i clar.
-        - Usa els noms correctes dels personatges segons apareguin a la història.
+        text = extract_text_from_srt(audio_descripcion_path_sin_une)
+        print(text)
 
-        Sortida:
-        - Un únic text narratiu continu.
-        """
-
-        messages = [{'role':'system','content': sys_prompt}]
-        out = state['llm_GPT'](messages)
+        out = free_narration_schat(text)
 
         with open(story_path, "w", encoding="utf-8-sig") as f:
             f.write(out)