Update app.py

app.py

@@ -118,13 +118,25 @@ def plot_scatter_tab5(cat, x, y, z, col):
     grouped_cat["count"] = grouped_cat.groupby(["model"])["count"].transform(lambda x: x / x.sum())
 
     # Pivot the data for stacking
-    pivot_df = grouped_cat.pivot(index='model', columns='tag', values='count').fillna(0)
-    # pivot_df = pivot_df.sort_values(by="A", ascending=False)
-    # add color vis
+    # pivot_df = grouped_cat.pivot(index='model', columns='tag', values='count').fillna(0)
+    # if col == "Size":
+    #     pivot_df[col] = pivot_df.index.map(size_map)
+    # else:
+    #     pivot_df[col] = pivot_df.index.str.split("/").str[0]
+
+    pivot_df = grouped_cat.pivot(index='model', columns='tag', values='count').fillna(0).reset_index()
+
     if col == "Size":
-        pivot_df[col] = pivot_df.index.map(size_map)
+        pivot_df[col] = pivot_df["model"].map(size_map)
     else:
-        pivot_df[col] = pivot_df.index.str.split("/").str[0]
+        pivot_df[col] = pivot_df["model"].str.split("/").str[0]
+    
+    fig = px.scatter_3d(pivot_df, x=x, y=y, z=z,
+        hover_name="model",
+        title=f'{x} vs {y} vs {z}',
+        color=col,
+        color_continuous_scale="agsunset")
+
 
     # Create an interactive scatter plot
     # fig = px.scatter(pivot_df, x=x, y=y, hover_name=pivot_df.index, title=f'{x} vs {y}', color=col, color_continuous_scale="agsunset")