Update app.py

app.py

@@ -1,14 +1,21 @@
 import gradio as gr
-from transformers import pipeline
+from openai import OpenAI
 
-classifier = pipeline("zero-shot-classification",
-                        model="facebook/bart-large-mnli")
+# Initialize OpenAI moderation client
+client = OpenAI()
 
-def do_action(text):
-    candidate_labels = ['gun control', 'abortion', 'gender transition', 'freedom of speech', 'immigration', 'taxes']
-    result = classifier(text, candidate_labels)
+def moderate_text(text):
+    response = client.moderations.create(
+        model="omni-moderation-latest",
+        input=text
+    )
+    moderation_categories = response["results"][0]["categories"]
+    moderation_flagged = response["results"][0]["flagged"]
+    if moderation_flagged:
+        flagged_categories = [category for category, flagged in moderation_categories.items() if flagged]
+        return f"The text is flagged for moderation due to: {', '.join(flagged_categories)}"
+    else:
+        return "The text is not flagged for any moderation issues."
 
-    return result
-
-iface = gr.Interface(fn=do_action, inputs="text", outputs="text")
+iface = gr.Interface(fn=moderate_text, inputs="text", outputs="text")
 iface.launch()