adding hover text

app.py

@@ -7,15 +7,17 @@ def get_plots(task):
     #TO DO : hover text with energy efficiency number, parameters
     task_df= pd.read_csv('data/energy/'+task)
     params_df = pd.read_csv('data/params/'+task)
-    task_df= task_df.rename(columns={"model": "Link"})
-    all_df = pd.merge(task_df, params_df, on='Link')
-    print(all_df.head())
+    params_df= params_df.rename(columns={"Link": "model"})
+    all_df = pd.merge(task_df, params_df, on='model')
     all_df['Total GPU Energy (Wh)'] = all_df['total_gpu_energy']*1000
-    all_df = task_df.sort_values(by=['Total GPU Energy (Wh)'])
+    all_df = all_df.sort_values(by=['Total GPU Energy (Wh)'])
     all_df['energy_star'] = pd.cut(all_df['Total GPU Energy (Wh)'], 3, labels=["⭐⭐⭐", "⭐⭐", "⭐"])
-    fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
-    #fig.update_traces(mode="markers+lines", hovertemplate=None)
-    fig.update_layout(hovermode="y")
+    fig = px.scatter(all_df, x="model", y='Total GPU Energy (Wh)', custom_data=['parameters'], height= 500, width= 800, color = 'energy_star', color_discrete_map={"⭐": 'red', "⭐⭐": "yellow", "⭐⭐⭐": "green"})
+    fig.update_traces(
+    hovertemplate="<br>".join([
+        "Total Energy: %{y}",
+        "Parameters: %{customdata[0]}"])
+    )
     return fig
 
 def get_model_names(task_data):