Update app.py

app.py

@@ -4,12 +4,14 @@ from typing import Optional
 import openai
 import streamlit as st
 from langchain_core.messages import AIMessage, HumanMessage
+from openai import OpenAI
 
 from rag.runnable_and_memory import get_runnable_and_memory
 from utils.error_message_template import ERROR_MESSAGE
 
 logging.basicConfig(level=logging.ERROR)
 
+openai_client = OpenAI()
 
 # Streamlit page configuration
 st.set_page_config(
@@ -19,7 +21,7 @@ st.set_page_config(
     initial_sidebar_state="collapsed",
 )
 
-# CSS
+# Streamlit CSS configuration
 with open("styles/styles.css") as css:
     st.markdown(f"<style>{css.read()}</style>", unsafe_allow_html=True)
 
@@ -113,38 +115,52 @@ if __name__ == "__main__":
 
     # Chat interface
     if user_input and user_input.strip():
-        st.session_state["chat_history"].append(HumanMessage(content=user_input))
-
-        # Append the location to the user input (important!)
-        user_query_location = (
-            f"{user_input}\nLocation: {st.session_state.selected_location}."
+        # OPENAI moderator api
+        moderator = openai_client.moderations.create(
+            model="omni-moderation-latest",
+            input=user_input,
         )
 
-        # Render the user input
-        with st.chat_message("Human", avatar=avatars["Human"]):
-            st.write(user_input)
-
-        # Render the AI response
-        with st.chat_message("AI", avatar=avatars["AI"]):
-            try:
-                with st.spinner(" "):
-                    response = st.write_stream(
-                        st.session_state["runnable"].stream(
-                            {"user_query": user_query_location}
+        if moderator.results[0].flagged:
+            response = """Sorry, I can't process your message because it doesn't follow our content guidelines. Please revise it and try again.
+            If you need assistance, please contact our clinic directly [here](https://www.talltreehealth.ca/contact-us).
+            """
+
+            with st.chat_message("AI", avatar=avatars["AI"]):
+                st.warning(response, icon="⚠️")
+
+        else:
+            st.session_state["chat_history"].append(HumanMessage(content=user_input))
+            # Append the location to the user input (important!)
+            user_query_location = (
+                f"{user_input}\nLocation: {st.session_state.selected_location}."
+            )
+
+            # Render the user input
+            with st.chat_message("Human", avatar=avatars["Human"]):
+                st.write(user_input)
+
+            # Render the AI response
+            with st.chat_message("AI", avatar=avatars["AI"]):
+                try:
+                    with st.spinner(" "):
+                        response = st.write_stream(
+                            st.session_state["runnable"].stream(
+                                {"user_query": user_query_location}
+                            )
                         )
-                    )
-            except openai.BadRequestError:
-                handle_errors()
-            except Exception:
-                handle_errors()
-
-        # Add AI response to the message history
-        st.session_state["chat_history"].append(AIMessage(content=response))
-
-        # Update runnable memory
-        st.session_state["memory"].chat_memory.add_user_message(
-            HumanMessage(content=user_query_location)
-        )
-        st.session_state["memory"].chat_memory.add_ai_message(
-            AIMessage(content=response)
-        )
+                except openai.BadRequestError:
+                    handle_errors()
+                except Exception:
+                    handle_errors()
+
+            # Add AI response to the message history
+            st.session_state["chat_history"].append(AIMessage(content=response))
+
+            # Update runnable memory
+            st.session_state["memory"].chat_memory.add_user_message(
+                HumanMessage(content=user_query_location)
+            )
+            st.session_state["memory"].chat_memory.add_ai_message(
+                AIMessage(content=response)
+            )