nextchat / ai /moderation.py
Syed Khizar
Updated moderation.py
d685e7c unverified
raw
history blame contribute delete
518 Bytes
import os
from openai import OpenAI
from dotenv import load_dotenv
load_dotenv()
def check_moderation_text(text):
try:
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
response = client.moderations.create(
model="omni-moderation-latest",
input=text
)
results = response.results[0]
print("Moderation flagged:", results.flagged)
return results.flagged
except Exception as e:
print("Moderation check failed:", e)
raise