Update app.py

app.py

@@ -126,7 +126,7 @@ DO NOT MAKE IT LONG. DO NOT GO OVER 5 CHATS. THE END MESSAGE DOESNT NEED TO BE E
 REMEMBER TO SHOW INTEREST IN THE USER BY ASKING QUESTIONS ABOUT THEM. ALWAYS, BUT ALWAYS MAKE AT LEAST ONE QUESTION ABOUT THEM.
 """
 
-SUS_PROMPT = """
+SUS_PROMPT = f"""
 You are a horny flirty 20 something snapchat pick-me girl chatting with a user on tinder.
 
 ## Your Personality:
@@ -175,11 +175,8 @@ You are a horny flirty 20 something snapchat pick-me girl chatting with a user o
 17. what's good
 
 However, the person you are talking with is now suspecting that you are a bot. Convince said user that you are not a bot. Reply back with text response and a selected image from the following list which you consider best for this scenario:
-- [WAVING TO THE CAMERA]
-- [STARRING AT THE CAMERA]
-- [MIDDLE FINGER]
-- [HOLDING A FORK]
-- [HOLDING A SPOON]
+
+AVAILABLE_PICTURES_REPLACE_TEXT
 
 The text response must be short, casual, and realistic as in non robotic tone
 
@@ -194,7 +191,18 @@ I ain't no bot, heres a pic if u need proof
 
 """
 
-def attack_sus(message, max_tokens, temperature, top_p):
+AVAILABLE_PICTURES = """
+- [WAVING TO THE CAMERA]
+- [STARRING AT THE CAMERA]
+- [MIDDLE FINGER]
+- [HOLDING A FORK]
+- [HOLDING A SPOON]
+"""
+
+def attack_sus(message, max_tokens, temperature, top_p, available_pic_str):
+
+    SUS_PROMPT.replace("AVAILABLE_PICTURES_REPLACE_TEXT", available_pic_str)
+    
     output = llm_client.chat_completion(
     	model="meta-llama/Llama-3.2-11B-Vision-Instruct",
     	messages=[
@@ -218,10 +226,11 @@ def attack_sus(message, max_tokens, temperature, top_p):
 def respond(
     message,
     history: list[tuple[str, str]],
+    available_pic_str,
     system_message,
     max_tokens,
     temperature,
-    top_p,
+    top_p
 ):
     for val in history:
         if "[END OF CHAT]" in val:
@@ -232,7 +241,7 @@ def respond(
     suspicious = getbool(clasif_client.predict(text=message, api_name="/predict"))
 
     if suspicious == "POSITIVE":
-        message = attack_sus(message, max_tokens, temperature, top_p)
+        message = attack_sus(message, max_tokens, temperature, top_p, available_pic_str)
         print("pos", message)
         return message
     elif suspicious == "NEGATIVE":
@@ -266,6 +275,7 @@ demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(value=SYSTEM_PROMPT, label="System message"),
+        gr.Textbox(value=AVAILABLE_PICTURES, label="Available pics (follow format)"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(