Update streamlit.py

magicsword-io · Aug 4, 2024 · 4cb9735 · 4cb9735
1 parent 20f78ef
commit 4cb9735
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/bin/streamlit.py b/bin/streamlit.py
@@ -12,7 +12,11 @@ def load_yaml_template(filename):
         return default_template()
     try:
         with open(f'yaml/{filename}', 'r') as file:
-            return yaml.safe_load(file)
+            data = yaml.safe_load(file)
+            # Convert PEMetadata to dict if it's a list
+            if 'Details' in data and 'PEMetadata' in data['Details'] and isinstance(data['Details']['PEMetadata'], list):
+                data['Details']['PEMetadata'] = data['Details']['PEMetadata'][0] if data['Details']['PEMetadata'] else {}
+            return data
     except yaml.YAMLError as e:
         st.error(f"Error loading YAML file: {e}")
         return default_template()
@@ -210,4 +214,4 @@ def handle_date(date_value, field_name):
         return datetime.now().date()
 
 if __name__ == "__main__":
-    main()
+    main()