Update app.py

app.py

@@ -18,19 +18,22 @@ def test(input_json):
 
 
     # Extract the datatree part which is a list of dictionaries
-    datatree = inputs["datatree"]
+    datatree = inputs['input']["datatree"]
     
     # Rebuild a Python dictionary from the datatree
     # Initialize an empty dictionary
-    python_dict = {}
-    
+    #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(',')]
         
@@ -48,7 +51,7 @@ def test(input_json):
     # Prepare the output
     output = {
         "list": a_list,
-        "matrix": python_dict,
+        "matrix": datatree,
     }
 
     return json.dumps(output)