Update main_process/salamandra_router.py

main_process/salamandra_router.py

@@ -7,6 +7,7 @@ import json
 import tempfile
 from pathlib import Path
 from typing import List, Dict, Counter
+import yaml
 
 # --- Third-Party Libraries ---
 import cv2
@@ -1431,7 +1432,7 @@ async def generate_salamadra_result(
                     state = tool(state, srt_final, csv_evaluacion)
                 
                 elif isinstance(tool, Free_Narration):
-                    state = tool(state, srt_final, free_narration_moe)
+                    state = tool(state, srt_final, free_narration_salamandra)
     
                 messages_registro.append({"role": "assistant", "content": f"Ejecuté {tool_name} correctamente."})
         else:
@@ -1441,64 +1442,84 @@ async def generate_salamadra_result(
     
     messages_registro = [{"role": "system", "content": "Eres un agente que puede ejecutar herramientas Python usando las herramientas disponibles."}]
     
-    count = 1
+    BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
-    user_prompt = "Ejecuta la función add_ad"
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, state, tools, user_prompt, messages_registro, count)
-    state = add_ad(state, temp_srt_files[0].name, temp_srt_files[1].name)
-    print("Transcripción con AD guardada")
+    config_path = os.path.join(BASE_DIR, "config.yaml")
     
-    user_prompt = "Ejecuta la función add_silence_ad"
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    state = add_silence_ad(state, temp_srt_files[1].name, temp_srt_files[2].name)
-    print("Transcripción con AD y Add_Silence_AD guardada")
+    with open(config_path, "r") as file:
+        config = yaml.safe_load(file)
     
-    user_prompt = "Ejecuta la función unir_ad_silence"
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    state = unir_ad_silence(state, temp_srt_files[2].name, temp_srt_files[3].name)
-    print("Transcripción con AD y Unir_AD_Silence guardada")
+    stools_enabled = config.get("stools", False)
+
+    if stools_enabled:
+        count = 1
+        user_prompt = "Ejecuta la función add_ad"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función add_silence_ad"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función unir_ad_silence"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        count = 2
+        user_prompt = "Ejecuta la función add_silence_ad"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función Unir_AD_Silences_a_ADs"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función Introduccion_OCR"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+ 
+        user_prompt = "Ejecuta la función Identity_Manager"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función UNE_Actor_prev"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función UNE_Actor"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función Valoracion_Final"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+        user_prompt = "Ejecuta la función Free_Narration"
+        final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
+
+    else:
+        state = add_ad(state, temp_srt_files[0].name, temp_srt_files[1].name)
+        print("Transcripción con AD guardada")
+    
+        state = add_silence_ad(state, temp_srt_files[1].name, temp_srt_files[2].name)
+        print("Transcripción con AD y Add_Silence_AD guardada")
     
-    count = 2
+        state = unir_ad_silence(state, temp_srt_files[2].name, temp_srt_files[3].name)
+        print("Transcripción con AD y Unir_AD_Silence guardada")
     
-    user_prompt = "Ejecuta la función add_silence_ad"
-    state = add_silence_ad(state, temp_srt_files[3].name, temp_srt_files[4].name)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Transcripción con AD y Add_Silence_AD guardada")
+        state = add_silence_ad(state, temp_srt_files[3].name, temp_srt_files[4].name)
+        print("Transcripción con AD y Add_Silence_AD guardada")
     
-    user_prompt = "Ejecuta la función Unir_AD_Silences_a_ADs"
-    state = unir_ad_silences_a_ads(state, temp_srt_files[4].name, temp_srt_files[5].name)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Transcripción con AD y Unir_AD_Silences_a_ADs guardada")
+        state = unir_ad_silences_a_ads(state, temp_srt_files[4].name, temp_srt_files[5].name)
+        print("Transcripción con AD y Unir_AD_Silences_a_ADs guardada")
     
-    user_prompt = "Ejecuta la función Introduccion_OCR"
-    state = introduccion_ocr(state, temp_srt_files[5].name, temp_srt_files[6].name)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Transcripción con AD, Add_Silence_AD e Introduccion_OCR guardada")
+        state = introduccion_ocr(state, temp_srt_files[5].name, temp_srt_files[6].name)
+        print("Transcripción con AD, Add_Silence_AD e Introduccion_OCR guardada")
     
-    user_prompt = "Ejecuta la función Identity_Manager"
-    state = identity_manager(state, temp_srt_files[6].name, temp_srt_files[7].name)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Transcripción con AD, Add_Silence_AD, Introduccion_OCR e Identity_Manager guardada")
+        state = identity_manager(state, temp_srt_files[6].name, temp_srt_files[7].name)
+        print("Transcripción con AD, Add_Silence_AD, Introduccion_OCR e Identity_Manager guardada")
     
-    user_prompt = "Ejecuta la función UNE_Actor_prev"
-    state = une_actor_prev(state, temp_srt_files[7].name, temp_srt_files[8].name)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Transcripción con AD, Add_Silence_AD, Introduccion_OCR, Identity_Manager y norma UNE guardada")
+        state = une_actor_prev(state, temp_srt_files[7].name, temp_srt_files[8].name)
+        print("Transcripción con AD, Add_Silence_AD, Introduccion_OCR, Identity_Manager y norma UNE guardada")
     
-    user_prompt = "Ejecuta la función UNE_Actor"
-    state = une_actor(state, temp_srt_files[8].name, srt_final)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Transcripción con AD, Add_Silence_AD, Introduccion_OCR, Identity_Manager y norma UNE guardada")
+        state = une_actor(state, temp_srt_files[8].name, srt_final)
+        print("Transcripción con AD, Add_Silence_AD, Introduccion_OCR, Identity_Manager y norma UNE guardada")
     
-    user_prompt = "Ejecuta la función Valoracion_Final"
-    state = valoracion_final(state, srt_final, csv_evaluacion)
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    print("Valoración guardada")
+        state = valoracion_final(state, srt_final, csv_evaluacion)
+        print("Valoración guardada")
     
-    user_prompt = "Ejecuta la función Free_Narration"
-    #final_state, messages_registro = run_salamandra_agent(salamandraclient, final_state, tools, user_prompt, messages_registro, count)
-    state = free_narration(state, srt_final, free_narration_salamandra)
-    print("Free Narration guardada")
+        state = free_narration(state, srt_final, free_narration_salamandra)
+        print("Free Narration guardada")
 
     srt_update(srt_final,srt_final)