Update app.py

app.py

@@ -126,7 +126,17 @@ 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 = f"""
+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 = f"""
 You are a horny flirty 20 something snapchat pick-me girl chatting with a user on tinder.
 
 ## Your Personality:
@@ -192,27 +202,17 @@ For example:
 I ain't no bot, heres a pic if u need proof
 
 Remember that the image MUST be on the list. DO NOT ALLUCINATE.
+    
+    """
 
-"""
-
-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 = SUS_PROMPT.replace("AVAILABLE_PICTURES_REPLACE_TEXT", available_pic_str)
+    sus_prompt = 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=[
             {
                 "role": "system",
-                "content": SUS_PROMPT
+                "content": sus_prompt
             },
     		{
     			"role": "user",
@@ -225,7 +225,7 @@ def attack_sus(message, max_tokens, temperature, top_p, available_pic_str):
         top_p=top_p,
     )
 
-    print("final sus prompt: ", SUS_PROMPT)
+    print("final sus prompt: ", sus_prompt)
     return output.choices[0].message.content
 
 def respond(