add moderation features...

core/moderation.py

@@ -0,0 +1,37 @@
+#coding: utf-8
+
+from typing import Any
+from typing import Dict
+#from typing import List
+#from typing import Tuple
+
+from typing import Optional
+#from typing import Union
+from os import getenv 
+import json
+
+from openai import OpenAI
+from dotenv import load_dotenv
+
+def moderate_text(input_text: str, 
+                  model: Optional[str] = "omni-moderation-latest") -> Dict[str, Any]:
+    
+    # Load environment variables from the .env file
+    load_dotenv()
+
+    # correct values for `model` is "omni-moderation-latest" and "text-moderation-latest"
+    try:
+        client = OpenAI(api_key=getenv("OPENAI_API_KEY"))
+        response = client.moderations.create(
+            model=f"{model}",
+            input=f"{input_text}"
+        )
+        return response
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return None
+
+
+
+
+