Update app.py

app.py

@@ -41,22 +41,26 @@ def plot_toxicity(comment):
     categories = list(toxicity_scores.keys())
     scores = list(toxicity_scores.values())
     
-    plt.figure(figsize=(8, 5), facecolor='black')
+    plt.figure(figsize=(10, 6), dpi=200, facecolor='black')
     ax = plt.gca()
     ax.set_facecolor('black')
     bars = plt.bar(categories, scores, color='#20B2AA', edgecolor='white')  # Sea green
     
-    plt.xticks(color='white', fontsize=12)
-    plt.yticks(color='white', fontsize=12)
-    plt.title("Toxicity Score Analysis", color='white', fontsize=14)
+    plt.xticks(color='white', fontsize=14)
+    plt.yticks(color='white', fontsize=14)
+    plt.title("Toxicity Score Analysis", color='white', fontsize=16)
     plt.ylim(0, 1)
     
     for bar in bars:
         yval = bar.get_height()
-        plt.text(bar.get_x() + bar.get_width()/2, yval + 0.02, f'{yval:.2f}', ha='center', color='white', fontsize=10)
+        plt.text(bar.get_x() + bar.get_width()/2, yval + 0.02, f'{yval:.2f}', ha='center', color='white', fontsize=12)
+    
+    # Add a curly bracket to group subcategories under "toxic"
+    plt.plot([-0.3, -0.3, 5.3, 5.3], [-0.05, 0.85, 0.85, -0.05], color='white', linewidth=2)
+    plt.text(-0.5, 0.9, "Toxic Subcategories", color='white', fontsize=14, rotation=90, verticalalignment='center')
     
     plt.tight_layout()
-    plt.savefig("toxicity_chart.png", facecolor='black')
+    plt.savefig("toxicity_chart.png", facecolor='black', bbox_inches='tight')
     plt.close()
     
     return "toxicity_chart.png"
@@ -72,4 +76,4 @@ demo = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()