loading all data on block reload

app.py

@@ -38,7 +38,6 @@ def parse_metrics_accuracy(meta):
     result = meta["model-index"][0]["results"]
     metrics = result[0]["metrics"]
     accuracy = metrics[0]["value"]
-    #print("ACCURACY", accuracy)
     return accuracy
 
 # We keep the worst case episode
@@ -64,9 +63,7 @@ def get_data(rl_env):
         row["User"] = user_id
         row["Model"] = model_id
         accuracy = parse_metrics_accuracy(meta)
-        #print("RETURNED ACCURACY", accuracy)
         mean_reward, std_reward = parse_rewards(accuracy)
-        #print("MEAN REWARD", mean_reward)
         row["Results"] = mean_reward - std_reward
         row["Mean Reward"] = mean_reward
         row["Std Reward"] = std_reward
@@ -84,8 +81,6 @@ def get_data_per_env(rl_env):
         dataframe["Model"] = dataframe["Model"].apply(make_clickable_model)
         dataframe = dataframe.sort_values(by=['Results'], ascending=False)
         table_html = dataframe.to_html(escape=False, index=False,justify = 'left')
-     
-        #table_html = table_html.replace("<thead>", '<thead align="left">')  # left-align the headers
         return table_html,dataframe,dataframe.empty
     else: 
         html = """<div style="color: green">
@@ -130,12 +125,19 @@ def get_info_display(len_dataframe,env_name,name_leaderboard,is_empty):
         """.format(name_leaderboard =  name_leaderboard)
     return markdown    
 
+def reload_all_data():
+
+    print(f'Loading all data...')
+    global RL_DETAILS,RL_ENVS
+
+    for rl_env in RL_ENVS:
+        RL_DETAILS[rl_env]['data'] = get_data_per_env(rl_env)
+
+
 def reload_leaderboard(rl_env):
     global RL_DETAILS
-    RL_DETAILS[rl_env]['data'] = get_data_per_env(rl_env)
-  
+ 
     data_html,data_dataframe,is_empty = RL_DETAILS[rl_env]['data'] 
-    data_html = data_html.replace('thead align="left"', '<thead align="left" style="color:red">')  # left-align the headers
 
     markdown = get_info_display(len(data_dataframe),rl_env,RL_DETAILS[rl_env]['title'],is_empty)            
     
@@ -146,17 +148,19 @@ with open('app.css','r') as f:
 
 block = gr.Blocks(css=BLOCK_CSS)
 with block:
+    block.load(reload_all_data,[],[])
     
     with gr.Tabs():
         for rl_env in RL_ENVS:
-            with gr.TabItem(rl_env):
+            with gr.TabItem(rl_env) as rl_tab:
                 data_html,data_dataframe,is_empty = RL_DETAILS[rl_env]['data'] 
                 markdown = get_info_display(len(data_dataframe),rl_env,RL_DETAILS[rl_env]['title'],is_empty)            
-                reload = gr.Button('Reload Leaderboard')
                 env_state =gr.Variable(default_value=rl_env)  
                 output_markdown = gr.HTML(markdown)
+                reload = gr.Button('Reload Leaderboard')
+
                 output_html = gr.HTML(data_html)
+
                 reload.click(reload_leaderboard,inputs=[env_state],outputs=[output_markdown,output_html])
-            
 
 block.launch()