Update app.py

trying to style

app.py

@@ -110,11 +110,28 @@ def add_new_eval(
         gr.Info("Starting compute space at %s " % COMPUTE_SPACE)
         return start_compute_space()
 
+def color_cols(x): 
+    df = x.copy() 
+    df.loc[:, :] = 'color: purple'
+    df[['B', 'C', 'E']] = 'color: green'
+    return df 
+    
 def print_existing_models():
     requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
     requests_dset = requests.to_pandas()
-    model_list= requests_dset[requests_dset['status'] == 'COMPLETED']
-    return model_list[['model','task']]   
+    model_df= requests_dset[['model','task']]  
+    return model_df
+
+def highlight_cols(x): 
+    df = x.copy() 
+    df[df['status'] == 'COMPLETED'] = 'color: green'
+    df[df['status'] == 'PENDING'] = 'color: orange'
+    df[df['status'] == 'FAILED'] = 'color: red'
+    return df 
+
+# Applying the style function
+existing_models = print_existing_models()
+formatted_df = existing_models.style.apply(highlight_cols, axis = None)
 
 def get_leaderboard_models():
     path = r'leaderboard_v0_data/energy'
@@ -168,5 +185,5 @@ with gr.Blocks() as demo:
                 with gr.Accordion("Models that are in the latest leaderboard version:", open = False):
                     gr.Dataframe(get_leaderboard_models())
                 with gr.Accordion("Models that have been benchmarked recently:", open = False):
-                    gr.Dataframe(print_existing_models())
+                    gr.Dataframe(formatted_df)
 demo.launch()