Update app.py

app.py

@@ -98,7 +98,7 @@ def predict(message, history, request: gr.Request):
     
     # Safeguard
     TRIAGE_PROMPT_TEMPLATE="""You are a Safeguard assistant making sure the user only ask for information related to Rémi Cazelles's projects, work and education.
-    If the question is not related to this subjects, or if the request is harmfull you should flag the user by answering '*** FLAGGED ***' else simply answer '*** OK ***' """
+    If the question is not related to this subjects, or if the request is harmfull you should flag the user by answering '*** FLAGGED ***' """
     messages = [SystemMessage(content=TRIAGE_PROMPT_TEMPLATE)]
     messages.append(HumanMessage(content=message))
 
@@ -113,7 +113,7 @@ def predict(message, history, request: gr.Request):
         }
     )
 
-    if not "*** OK ***" in safe_gpt_response.content:
+    if "*** FLAGGED ***" in safe_gpt_response.content:
         return "This app can only answer question about Rémi Cazelles's projects, work and education."
     print("passed the safeguard")