Update app.py

app.py

@@ -14,16 +14,31 @@ from ingestion_chroma import retrieve_info_from_db
 ################################### TOOLS ##################################################
 ############################################################################################
 
-def find_key(data, target_key):
-    if isinstance(data, dict):
-        for key, value in data.items():
-            if key == target_key:
-                return value
-            else:
-                result = find_key(value, target_key)
-                if result is not None:
-                    return result
-    return "Indicator not found"
+def search_key(d, target_key):
+    """
+    Recherche une clé dans un dictionnaire imbriqué.
+
+    :param d: Le dictionnaire dans lequel chercher.
+    :param target_key: La clé à chercher.
+    :return: Une liste des valeurs associées à la clé trouvée.
+    """
+    results = []
+
+    def recursive_search(d):
+        if isinstance(d, dict):
+            for key, value in d.items():
+                if key == target_key:
+                    results.append(value)
+                if isinstance(value, dict):
+                    recursive_search(value)
+                elif isinstance(value, list):
+                    for item in value:
+                        if isinstance(item, dict):
+                            recursive_search(item)
+
+    recursive_search(d)
+    
+    return results
 
 ############################################################################################
 
@@ -65,7 +80,7 @@ class ESRS_info_tool(Tool):
         with open('./data/dico_esrs.json') as json_data:
             dico_esrs = json.load(json_data)
         
-        result = find_key(dico_esrs, indicator)
+        result = search_key(dico_esrs, indicator)
 
         return result