Update app.py

app.py

@@ -41,25 +41,21 @@ def plot_toxicity(comment):
     categories = list(toxicity_scores.keys())
     scores = list(toxicity_scores.values())
     
-    plt.figure(figsize=(10, 6), dpi=200, facecolor='black')
+    plt.figure(figsize=(12, 7), dpi=300, facecolor='black')
     ax = plt.gca()
     ax.set_facecolor('black')
-    bars = plt.bar(categories, scores, color='#20B2AA', edgecolor='white')  # Sea green
+    bars = plt.bar(categories, scores, color='#20B2AA', edgecolor='white', width=0.5)  # Sea green
     
-    plt.xticks(color='white', fontsize=14)
+    plt.xticks(color='white', fontsize=14, rotation=25, ha='right')
     plt.yticks(color='white', fontsize=14)
     plt.title("Toxicity Score Analysis", color='white', fontsize=16)
-    plt.ylim(0, 1)
+    plt.ylim(0, 1.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=12)
+        plt.text(bar.get_x() + bar.get_width()/2, yval + 0.03, f'{yval:.2f}', ha='center', color='white', fontsize=12, fontweight='bold')
     
-    # 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.tight_layout(pad=2)
     plt.savefig("toxicity_chart.png", facecolor='black', bbox_inches='tight')
     plt.close()
     
@@ -76,4 +72,4 @@ demo = gr.Interface(
 
 # Launch the Gradio app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()