Update app.py

app.py

@@ -106,11 +106,28 @@ def handle_events(threadId, chat_history, storage):
     chat_history[-1][1] = "Error occured during processing your message. Please try again"
     yield [chat_history, storage]
 
+def check_moderation_flag(message):
+  moderation_response = client.moderations.create(input=message, model="omni-moderation-latest")
+  print("Moderation respones: ", moderation_response)
+  flagged = moderation_response.results[0].flagged
+  return flagged
+
+def process_user_input(text, thread_id, chat_history, storage):
+  print("User input: ", text)
+  is_flagged = check_moderation_flag(text)
+  print("Check is flagged:", is_flagged)
+  if is_flagged:
+    chat_history[-1][1] = "Your request contains some inappropriate information. We cannot proceed with it."
+    yield [chat_history, storage]
+  else:
+    add_message_to_openai(initial_message, thread_id)
+    for response in handle_events(thread_id, chat_history, storage):
+      yield response
+
 def initiate_chatting(chat_history, storage):
   threadId = storage["threadId"]
   chat_history = [[None, ""]]
-  add_message_to_openai(initial_message, threadId)
-  for response in handle_events(threadId, chat_history, storage):
+  for response in process_user_input(initial_message, threadId, chat_history, storage):
     yield response
 
 def respond_on_user_msg(chat_history, storage):
@@ -118,8 +135,7 @@ def respond_on_user_msg(chat_history, storage):
   threadId = storage["threadId"]
   print("Responding for threadId: ", threadId)
   chat_history[-1][1] = ""
-  add_message_to_openai(message, threadId)
-  for response in handle_events(threadId, chat_history, storage):
+  for response in process_user_input(message, threadId, chat_history, storage):
     yield response
 
 def create_chat_tab():