Update app.py

app.py

@@ -2,24 +2,26 @@ import gradio as gr
 from detoxify import Detoxify
 import pandas as pd
 
-# Load model once at startup
+# Load model once
 model = Detoxify('original')
 
-def classify_toxicity(text):
-    if not text.strip():
-        return "Please enter a valid comment."
+def classify_multiple(comments):
+    # Split input by newlines and clean
+    comment_list = [c.strip() for c in comments.split('\n') if c.strip()]
+    if not comment_list:
+        return "Please enter at least one valid comment."
 
-    results = model.predict(text)
-    df = pd.DataFrame([results], index=["Score"]).T
-    df = df.round(4)
+    results = model.predict(comment_list)  # Returns a dict of lists
+
+    df = pd.DataFrame(results, index=comment_list).round(4)
     return df
 
 iface = gr.Interface(
-    fn=classify_toxicity,
-    inputs=gr.Textbox(label="Enter a Comment", lines=3, placeholder="Example: You are so annoying..."),
-    outputs=gr.Dataframe(label="Toxicity Classification"),
-    title="💬 Toxic Comment Classifier",
-    description="Classifies input comments as toxic, severe toxic, obscene, threat, insult, and identity hate using Detoxify (unitary/toxic-bert)."
+    fn=classify_multiple,
+    inputs=gr.Textbox(lines=8, placeholder="Enter one or more comments, each on a new line..."),
+    outputs=gr.Dataframe(label="Toxicity Predictions"),
+    title="💬 Toxic Comment Classifier (Multi-Comment)",
+    description="Paste one or more comments. Each will be scored for toxicity, severe toxicity, insult, threat, obscene, and identity hate using Detoxify."
 )
 
 if __name__ == "__main__":