Display loading message

app.py

@@ -4,11 +4,11 @@ import gradio as gr
 
 import src.constants as constants
 from src.details import update_subtasks_component, update_load_details_component, load_details_dataframes, \
-    display_details, update_sample_idx_component, clear_details, update_task_description_component
+    display_details, update_sample_idx_component, clear_details, update_task_description_component, \
+    display_loading_message_for_details
 from src.results import update_load_results_component, \
     load_results_dataframes, display_results, update_tasks_component, clear_results, \
-    sort_result_paths_per_model, fetch_result_paths
-
+    sort_result_paths_per_model, fetch_result_paths, display_loading_message_for_results
 
 # if __name__ == "__main__":
 result_paths_per_model = sort_result_paths_per_model(fetch_result_paths())
@@ -99,6 +99,9 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
     )
     gr.on(
         triggers=[load_results_btn.click, load_configs_btn.click],
+        fn=display_loading_message_for_results,
+        outputs=[results, configs],
+    ).then(
         fn=load_results_dataframes,
         inputs=[model_id_1, model_id_2],
         outputs=[dataframe_1, dataframe_2],
@@ -132,6 +135,7 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
         outputs=[model_id_1, model_id_2, dataframe_1, dataframe_2, load_results_btn, load_configs_btn, results_task, configs_task],
     )
 
+    # DETAILS:
     details_task.change(
         fn=update_task_description_component,
         inputs=details_task,
@@ -148,6 +152,9 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
         outputs=load_details_btn,
     )
     load_details_btn.click(
+        fn=display_loading_message_for_details,
+        outputs=details,
+    ).then(
         fn=load_details_dataframes,
         inputs=[subtask, model_id_1, model_id_2],
         outputs=[details_dataframe_1, details_dataframe_2],

src/details.py

@@ -96,3 +96,7 @@ def clear_details():
         gr.Button("Load Details", interactive=False),
         gr.Number(label="Sample Index", info="Index of the sample to be displayed", value=0, minimum=0,visible=False),
     )
+
+
+def display_loading_message_for_details():
+    return "<h3 style='text-align: center;'>Loading...</h3>"

src/results.py

@@ -113,3 +113,7 @@ def highlight_min_max(s):
         return np.where(s == np.nanmax(s.values), "background-color:green", "background-color:#D81B60")
     else:
         return [""] * len(s)
+
+
+def display_loading_message_for_results():
+    return ("<h3 style='text-align: center;'>Loading...</h3>", ) * 2