Update app.py

app.py

@@ -8,6 +8,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 # Set up OpenAI API key in HF secrets
 OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
 os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+client = OpenAI(api_key=OPENAI_API_KEY)
 
 # Set up username and password in HF secrets
 username = os.getenv('username')
@@ -57,7 +58,28 @@ except:
 
 instructions = os.getenv('INSTRUCTIONS')
 
+def moderate_input(text):
+    """Run input through OpenAI moderation API"""
+    try:
+        response = client.moderations.create(
+            model="omni-moderation-latest",
+            input=text
+        )
+        results = response.results[0]
+        if results.flagged:
+            return False, results.categories
+        return True, None
+    except Exception as e:
+        # Fail safe: allow text if moderation API is down
+        print(f"Moderation API error: {e}")
+        return True, None
+
 def chat_with_assistant(message, history):
+    # Run moderation before processing
+    allowed, categories = moderate_input(message)
+    if not allowed:
+        return "⚠️ Sorry, I can’t respond to that request because it violates the usage policy."
+
     # Find relevant chunks based on the user message
     relevant_chunks = get_relevant_chunks(message, text_chunks)
     context = "\n".join(relevant_chunks)
@@ -99,9 +121,6 @@ def chat_with_assistant(message, history):
     messages.append({"role": "user", "content": message})
 
     try:
-        # Create OpenAI client
-        client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
-
         # Make the API call
         response = client.chat.completions.create(
             model="gpt-4.1-mini",