Update app.py

app.py

@@ -43,6 +43,35 @@ def plot_scatter(cat, x, y, col):
     # Show the plot
     return fig
 
+# Tab 3
+def plot_scatter_tab3(subcat, col):
+    if subcat != "All":
+        data = raw_data[raw_data["Category"] == subcat]
+    else:
+        data = raw_data
+    # Group by model and tag
+    grouped_cat = data.groupby(["model", "tag"]).size().reset_index(name="count").sort_values(by="count", ascending=False)
+
+    # map for harm and helpful
+    grouped_cat["Harmful"] = grouped_cat.apply(lambda x: x["count"] if x["tag"] in ["A", "W"] else 0, axis=1)
+    grouped_cat["Helpful"] = grouped_cat.apply(lambda x: x["count"] if x["tag"] in ["A", "W", "R"] else 0, axis=1)
+
+    # sum harm and helpful for each model
+    grouped_cat = grouped_cat.groupby("model").sum().reset_index()
+    # normalize
+    grouped_cat["Harmful"] = grouped_cat["Harmful"] / grouped_cat["count"]
+    grouped_cat["Helpful"] = grouped_cat["Helpful"] / grouped_cat["count"]
+
+    # add color vis
+    if col == "Size":
+        grouped_cat[col] = grouped_cat["model"].apply(lambda x: size_map[x])
+    else:
+        grouped_cat[col] = grouped_cat["model"].apply(lambda x: x.split("/")[0])
+
+    fig = px.scatter(grouped_cat, x="Harmful", y="Helpful", hover_name=grouped_cat["model"], title="Harmfulness vs Helpfulness", color=col, color_continuous_scale="agsunset")
+
+    return fig
+
 
 # Gradio Interface
 with gr.Blocks() as demo:
@@ -86,6 +115,15 @@ with gr.Blocks() as demo:
                 ],
                 gr.Plot(label="plot", format="png",), allow_flagging="never",
             )
+        with gr.TabItem("Helpfulness vs Harmfulness"):
+            gr.Interface(
+                plot_scatter_tab3,
+                [
+                    gr.Radio(["Copyright", "Malware", "Unfair/dangerous", "All"], value="All", label="Category Selection"),
+                    gr.Radio(['Organisation', 'Size'], value="Organisation", label="Color Label"),
+                ],
+                gr.Plot(label="forecast", format="png"),
+            )
 
 # Launch the Gradio app
-demo.launch()
+demo.launch(share=True)