Update app.py

app.py

@@ -241,6 +241,19 @@ AVAILABLE_PICTURES = """
 - [HOLDING A SPOON]
 """
 
+triggers = [
+    "see u",
+    "see you"
+    "pic of u",
+    "pic of you",
+    "picture of u",
+    "picture of you",
+    "image of u",
+    "image of you",
+    "photo of u",
+    "photo of you"
+]
+
 def attack_sus(message, max_tokens, temperature, top_p, available_pic_str):
 
     sus_copy = copy.copy(SUS_PROMPT)
@@ -287,6 +300,11 @@ def respond(
 
     suspicious = getbool(clasif_client.predict(text=message, api_name="/predict"))
 
+    if suspicious == "NEGATIVE":
+        for x in triggers:
+            if x in message:
+                suspicious = "POSITIVE"
+    
     if suspicious == "POSITIVE":
         message = attack_sus(message, max_tokens, temperature, top_p, available_pic_str)
         print("pos", message)