Update utils.py

utils.py

@@ -2,7 +2,8 @@
 
 import os
 
-LINK_FIELDS = ["Tag"]
+# Definir los campos de enlace relevantes
+LINK_FIELDS = ["Tag"]  # Utilizamos solo el campo 'Tag' según tu ejemplo de datos
 
 def check_image_exists(path):
     """
@@ -26,13 +27,14 @@ def process_tags_chat(search_results):
     Returns:
         tuple: (full_response (str), image_url (str o None))
     """
-    # Manejar la lógica para procesar tags e imágenes internamente
+    # Extraer enlaces relevantes de los metadatos utilizando LINK_FIELDS
     relevant_links = [
         result['metadata'].get(field)
         for result in search_results
         for field in LINK_FIELDS
         if field in result['metadata']
     ]
+    
     full_response = ""
     image_url = None
     tags_detected = []
@@ -40,7 +42,7 @@ def process_tags_chat(search_results):
     filtered_links = []
     if relevant_links:
         # Limitar a solo dos tags
-        tag_list = ["rld 0", "rld 1"]
+        tag_list = ["rld 0", "rld 1"]  # Ajusta estos tags según tus necesidades
         for link in relevant_links:
             if any(tag in link for tag in tag_list):
                 tags_detected.append(link)  # Guardar el tag pero no mostrarlo
@@ -52,15 +54,20 @@ def process_tags_chat(search_results):
             full_response += f"\n\nRespuestas relevantes:\n{filtered_links[0]}"
 
         # Manejar las imágenes basadas en los tags detectados
+        current_dir = os.path.dirname(os.path.abspath(__file__))
+        images_dir = os.path.join(current_dir, "images")  # Asegúrate de que la carpeta 'images' exista
+
         tags_to_images = {
-            "rld 0": "images/rld0.png",
-            "rld 1": "images/rld1.png",
+            "rld 0": os.path.join(images_dir, "rld_0.png"),
+            "rld 1": os.path.join(images_dir, "rld_1.png"),
+            "rld 2": os.path.join(images_dir, "rld_2.png"),
+            # Agrega más mappings según tus tags e imágenes
         }
 
         for tag in tags_detected:
-            for key, path in tags_to_images.items():
-                if key in tag and check_image_exists(path):
+            if tag in tags_to_images:
+                path = tags_to_images[tag]
+                if check_image_exists(path):
                     image_url = path
                     break  # Asumimos solo una imagen relevante
-
     return full_response, image_url