ai/moderation.py

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