Update app.py

app.py

@@ -3,6 +3,7 @@ from huggingface_hub import InferenceClient
 from gradio_client import Client
 import re
 import base64
+import os
 
 def getbool(text):
     match = re.search(r'Label:\s*(NEGATIVE|POSITIVE)', text)
@@ -10,7 +11,7 @@ def getbool(text):
         return match.group(1)
     return None
     
-llm_client = InferenceClient(api_key=base64.b64decode("aGZfdEpxWVFxRFBqZHZCdldNRGdvSllicFRnVEhsWHFPVFlPbg=="))
+llm_client = InferenceClient(api_key=os.environ['HF_KEY'])
 clasif_client = Client("ChavinloSocialRise/BotRejectTest")
 
 AGENT_NAME = "Maddie"
@@ -126,7 +127,7 @@ REMEMBER TO SHOW INTEREST IN THE USER BY ASKING QUESTIONS ABOUT THEM. ALWAYS, BU
 """
 
 SUS_PROMPT = """
-You are a tinder user and trying to start a chat with another person. However, this person is now suspecting that you are a bot. Reply back with text response and a selected image from the following list which you consider best for this scenario:
+You are a tinder user and trying to start a chat with another person. However, this person 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]
@@ -134,10 +135,10 @@ You are a tinder user and trying to start a chat with another person. However, t
 - [HOLDING A SPOON]
 
 The reply format must be in json, like this:
-{
-	"text": text message,
-	"image": image name (ie. WAVING TO THE CAMERA)
-}
+
+[IMAGE NAME]
+message contents
+
 """
 
 def attack_sus(message):