Update app.py

app.py

@@ -28,10 +28,20 @@ class Chatbot:
 
 def create_bot(input):
     bot = Chatbot(os.getenv("OPENAI_API_KEY"), index=index)
-  
+
+    # use moderations endpoint to check input
     if input:
-        response = bot.generate_response(input)
-        output = response['content']
+        client = OpenAI()
+        response_mod = client.moderations.create(input=input)
+        response_dict = response_mod.model_dump()
+        flagged = response_dict['results'][0]['flagged']
+        print("Flagged:", flagged)
+    
+        if not flagged:
+            response_bot = bot.generate_response(input)
+            output = response_bot['content']
+        else:
+             output = "Invalid request."
 
         return output