Update app.py

app.py

@@ -38,7 +38,7 @@ def make_link(mname):
     display_name = parts[1] if len(parts) > 1 else mname
     return f'[{display_name}](https://huggingface.co/{mname})'
 
-# --- Plot Functions (Bar Chart) ---
+# --- Plot Functions (Bar Chart - Modified) ---
 
 def get_plots(task):
     df = pd.read_csv('data/energy/' + task)
@@ -50,6 +50,12 @@ def get_plots(task):
     # Create a display model column for labeling
     df['Display Model'] = df['model'].apply(lambda m: m.split('/')[-1])
 
+    # --- Debugging: Check data type and dataframe ---
+    print(f"Data type of total_gpu_energy: {df['total_gpu_energy'].dtype}")
+    print("DataFrame head before plotting:")
+    print(df.head())
+    # --- End Debugging ---
+
     # Use the energy score to control color
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
 
@@ -75,10 +81,11 @@ def get_plots(task):
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        yaxis_tickformat=".4f", # Add this line to format y-axis ticks - might not be needed for bar chart
-        yaxis = dict(
-            tickformat=".4f" # Ensure tickformat is set within yaxis dict as well - might not be needed for bar chart
-        )
+        # --- Removed explicit y-axis formatting for now ---
+        # yaxis_tickformat=".4f",
+        # yaxis = dict(
+        #     tickformat=".4f"
+        # )
     )
     return fig
 
@@ -94,6 +101,13 @@ def get_all_plots():
         all_df = pd.concat([all_df, df], ignore_index=True)
     all_df = all_df.drop_duplicates(subset=['model'])
 
+    # --- Debugging: Check data type and dataframe ---
+    print(f"Data type of total_gpu_energy (all_plots): {all_df['total_gpu_energy'].dtype}")
+    print("DataFrame head before plotting (all_plots):")
+    print(all_df.head())
+    # --- End Debugging ---
+
+
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
 
     fig = px.bar(
@@ -116,14 +130,15 @@ def get_all_plots():
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        yaxis_tickformat=".4f", # Add this line to format y-axis ticks - might not be needed for bar chart
-        yaxis = dict(
-            tickformat=".4f" # Ensure tickformat is set within yaxis dict as well - might not be needed for bar chart
-        )
+        # --- Removed explicit y-axis formatting for now ---
+        # yaxis_tickformat=".4f",
+        # yaxis = dict(
+        #     tickformat=".4f"
+        # )
     )
     return fig
 
-# --- New functions for Text Generation filtering by model class (with Bar Chart) ---
+# --- New functions for Text Generation filtering by model class (with Bar Chart - Modified) ---
 
 def get_text_generation_plots(model_class):
     df = pd.read_csv('data/energy/text_generation.csv')
@@ -136,6 +151,13 @@ def get_text_generation_plots(model_class):
     df['energy_score'] = df['energy_score'].astype(int).astype(str)
     df['Display Model'] = df['model'].apply(lambda m: m.split('/')[-1])
 
+    # --- Debugging: Check data type and dataframe ---
+    print(f"Data type of total_gpu_energy (text_gen): {df['total_gpu_energy'].dtype}")
+    print("DataFrame head before plotting (text_gen):")
+    print(df.head())
+    # --- End Debugging ---
+
+
     color_map = {"1": "red", "2": "orange", "3": "yellow", "4": "lightgreen", "5": "green"}
 
     fig = px.bar(
@@ -158,10 +180,11 @@ def get_text_generation_plots(model_class):
     fig.update_layout(
         xaxis_title="Model",
         yaxis_title="GPU Energy (Wh)",
-        yaxis_tickformat=".4f", # Add this line to format y-axis ticks - might not be needed for bar chart
-        yaxis = dict(
-            tickformat=".4f" # Ensure tickformat is set within yaxis dict as well - might not be needed for bar chart
-        )
+        # --- Removed explicit y-axis formatting for now ---
+        # yaxis_tickformat=".4f",
+        # yaxis = dict(
+        #     tickformat=".4f"
+        # )
     )
     return fig