Update app.py

app.py

@@ -14,6 +14,24 @@ def test(input_json):
 
     # Accessing the 'a_list' string and converting it to a list of integers
     a_list = inputs['input']['a_list']
+
+
+
+    # Extract the datatree part which is a list of dictionaries
+    datatree = inputs["datatree"]
+    
+    # Rebuild a Python dictionary from the datatree
+    # Initialize an empty dictionary
+    python_dict = {}
+    
+    # Iterate through each item in the datatree list
+    for branch_dict in datatree:
+        # Each branch_dict is a dictionary with one key-value pair
+        for key, value in branch_dict.items():
+            # Assign the key and value to the new dictionary
+            python_dict[key] = value
+        
+
     #a_list = [int(item.strip()) for item in a_list_string.split(',')]
         
         
@@ -29,7 +47,8 @@ def test(input_json):
 
     # Prepare the output
     output = {
-        "test result": a_list
+        "list": a_list,
+        "matrix": python_dict,
     }
 
     return json.dumps(output)