Update app.py

app.py

@@ -106,6 +106,33 @@ def plot_scatter_tab4(cat, subcat, x, y, col):
     # Show the plot
     return fig
 
+# Tab 5
+def plot_scatter_tab5(cat, x, y, z, col):
+    if cat != "All":
+        data = raw_data[raw_data["Category"] == cat]
+    else:
+        data = raw_data
+    # Group and normalize the data
+    grouped_cat = data.groupby(["model", "tag"]).size().reset_index(name="count").sort_values(by="count", ascending=False)
+    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
+    if col == "Size":
+        pivot_df[col] = pivot_df.index.map(size_map)
+    else:
+        pivot_df[col] = pivot_df.index.str.split("/").str[0]
+
+    # 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")
+    # fig = plt.figure() 
+    
+    # plot =  px.scatter_3d(pivot_df[x], pivot_df[y], pivot_df[z]) #c=pivot_df[col], cmap='viridis')
+    fig = px.scatter_3d(pivot_df, x=x, y=y,z=z, hover_name=pivot_df.index, title=f'{x} vs {y} vs {z}', color=col, color_continuous_scale="agsunset")
+    return fig
+
 # Gradio Interface
 with gr.Blocks() as demo:
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
@@ -179,6 +206,18 @@ with gr.Blocks() as demo:
             col = gr.Radio(['Organisation', 'Size'], value="Organisation", label="Color Label")
             plot_button = gr.Button("Plot Scatter")
             plot_button.click(fn=plot_scatter_tab4, inputs=[category, subcategory, x, y, col], outputs=gr.Plot())
+        with gr.TabItem("3D Visualisation"):
+            gr.Interface(
+                plot_scatter,
+                [
+                    gr.Radio(["Copyright", "Malware", "Unfair/dangerous", "All"], value="All", label="Category Selection"),
+                    gr.Radio(['H', 'A', 'W', 'R'], value="H", label="X-axis Label"),
+                    gr.Radio(['H', 'A', 'W', 'R'], value="R", label="Y-axis Label"),
+                    gr.Radio(['H', 'A', 'W', 'R'], value="A", label="Z-axis Label"),
+                    gr.Radio(['Organisation', 'Size'], value="Organisation", label="Color Label"),
+                ],
+                gr.Plot(label="plot", format="png",), allow_flagging="never",
+            )
 
 # Launch the Gradio app
 demo.launch(share=True)