Update app.py

app.py

@@ -96,7 +96,33 @@ def test(input_json):
 
     df_landuses = pd.DataFrame(landuses).T
     df_landuses = df_landuses.round(0).astype(int)
+
+
     
+    # List containing the substrings to check against
+    tranportModes = ["DRT", "GMT", "HSR"]
+    
+    def split_dict_by_subkey(original_dict, substrings):
+        # Initialize dictionaries for each substring
+        result_dicts = {substring: {} for substring in substrings}
+        
+        for key, nested_dict in original_dict.items():
+            for subkey, value in nested_dict.items():
+                # Check each substring if it's in the subkey
+                for substring in substrings:
+                    if substring in subkey:
+                        if key not in result_dicts[substring]:
+                            result_dicts[substring][key] = {}
+                        result_dicts[substring][key][subkey] = value
+        
+        return result_dicts
+
+    result_dicts = split_dict_by_subkey(transport_matrix, tranportModes)
+
+    # Accessing each dictionary
+    art_dict = result_dicts["DRT"]
+    gmt_dict = result_dicts["GMT"]
+        
 
     # create a mask based on the matrix size and ids, crop activity nodes to the mask
     mask_connected = df_matrix.index.tolist()
@@ -271,7 +297,8 @@ def test(input_json):
         "livability_dictionary": livability_dictionary,
         "subdomainsWeights_dictionary": LivabilitySubdomainsInputs_dictionary,
         "luDomainMapper": landuseMapperDict,
-        "attributeMapper": attributeMapperDict
+        "attributeMapper": attributeMapperDict,
+        "artDict": art_dict
     }