up

app.py

@@ -59,11 +59,11 @@ def fetch_datasets(base_folder: str):
     return datasets, gr.update(choices=datasets, value=None), fetch_groups(base_folder, datasets, None, "union")
 
 
-def export_data(exported_data):
+def export_data(exported_data, stat_name):
     if not exported_data:
         return None
     # Assuming exported_data is a dictionary where the key is the dataset name and the value is the data to be exported
-    with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".json") as temp:
+    with tempfile.NamedTemporaryFile(mode="w", delete=False, prefix=stat_name, suffix=".json") as temp:
         json.dump(exported_data, temp)
         temp_path = temp.name
     return gr.update(visible=True, value=temp_path)
@@ -92,7 +92,6 @@ def fetch_groups(base_folder, datasets, old_groups, type="intersection"):
 
 
 def fetch_stats(base_folder, datasets, group, old_stats, type="intersection"):
-    print("Fetching stats")
     with ThreadPoolExecutor() as executor:
         STATS = list(executor.map(lambda run: [Path(x).name for x in find_folders(base_folder, f"{run}/{group}")], datasets))
     if len(STATS) == 0:
@@ -141,10 +140,12 @@ def prepare_non_grouped_data(dataset_path, base_folder, grouping, stat_name, nor
     stats = load_stats(base_folder, dataset_path, stat_name, grouping)
     stats_rounded = defaultdict(lambda: 0)
     for key, value in stats.items():
-        stats_rounded[float(key)] += value.total
+        stats_rounded[round(float(key), 2)] += value.total
     if normalization:
         normalizer = sum(stats_rounded.values())
         stats_rounded = {k: v / normalizer for k, v in stats_rounded.items()}
+        # check that the sum of the values is 1
+        summed = sum(stats_rounded.values())
     return stats_rounded
 
 
@@ -432,7 +433,7 @@ Groupings:
 
     export_data_button.click(
         fn=export_data,
-        inputs=[exported_data],
+        inputs=[exported_data, stat_name_dropdown],
         outputs=export_data_json,
     )