Update app.py

app.py

@@ -33,32 +33,27 @@ def predict_toxicity(comment):
     
     return {labels[i]: float(probabilities[i]) for i in range(len(labels))}
 
-def visualize_toxicity(comment):
-    """Generates a bar chart showing toxicity levels."""
+def format_toxicity_data(comment):
+    """Formats the toxicity scores for a modern bar graph."""
     scores = predict_toxicity(comment)
-    
-    # Create bar chart
-    plt.figure(figsize=(6, 4))
-    plt.bar(scores.keys(), scores.values(), color=['blue', 'red', 'green', 'purple', 'orange', 'brown'])
-    plt.ylim(0, 1)
-    plt.ylabel("Toxicity Score")
-    plt.title("Toxicity Analysis")
-    plt.xticks(rotation=45)
-    plt.grid(axis='y', linestyle='--', alpha=0.7)
-    
-    # Save plot to display in Gradio
-    plt.savefig("toxicity_plot.png")
-    plt.close()
-    
-    return "toxicity_plot.png"
+    data = [[label, score] for label, score in scores.items()]
+    return data
 
 # Gradio interface
 demo = gr.Interface(
-    fn=visualize_toxicity,
+    fn=format_toxicity_data,
     inputs=gr.Textbox(label="Enter a comment:"),
-    outputs=gr.Image(type="filepath", label="Toxicity Scores"),
+    outputs=gr.BarPlot(
+        x="Category",
+        y="Score",
+        title="Toxicity Analysis",
+        y_lim=[0, 1],
+        color="blue",
+        label="Toxicity Scores",
+        interactive=False
+    ),
     title="Toxicity Detection with RogueBERT",
-    description="Enter a comment to analyze its toxicity levels. The results will be displayed as a bar chart."
+    description="Enter a comment to analyze its toxicity levels. The results will be displayed as a modern bar chart."
 )
 
 demo.launch()