From 0cc4a1f215adc68b19cc5918e08dddf89a065f2e Mon Sep 17 00:00:00 2001
From: Florez Ospina Juan Felipe <juan.florez-ospina@psi.ch>
Date: Mon, 3 Jun 2024 14:17:42 +0200
Subject: [PATCH] Updated treemap visualization to select only root metadata,
 which is of string type.

---
 src/hdf5_vis.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/hdf5_vis.py b/src/hdf5_vis.py
index e771b30..40c8977 100644
--- a/src/hdf5_vis.py
+++ b/src/hdf5_vis.py
@@ -28,8 +28,15 @@ def display_group_hierarchy_on_a_treemap(filename: str):
         metadata_dict={}
         for key in file.attrs.keys():            
             #if 'metadata' in key:
-            metadata_dict[key[key.find('_')+1::]]= file.attrs[key]
-            metadata_list.append(key[key.find('_')+1::]+':'+file.attrs[key])
+            if isinstance(file.attrs[key], str):  # Check if the attribute is a string
+                metadata_key = key[key.find('_') + 1:]
+                metadata_value = file.attrs[key]
+                metadata_dict[metadata_key] = metadata_value
+                metadata_list.append(f'{metadata_key}: {metadata_value}')
+            
+            #metadata_dict[key[key.find('_')+1::]]= file.attrs[key]
+            #metadata_list.append(key[key.find('_')+1::]+':'+file.attrs[key])
+
         metadata = '<br>'.join(['<br>'] + metadata_list)
 
     customdata_series = pd.Series(nodes)