Update app.py

app.py

@@ -6,13 +6,13 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 from prompts import GAME_MASTER, COMPRESS_HISTORY, ADJUST_STATS
 def format_prompt(message, history):
     prompt=""
-    '''
+
     prompt = "<s>"
     
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
         prompt += f" {bot_response}</s> "
-    '''
+
     prompt += f"[INST] {message} [/INST]"
     return prompt
 
@@ -60,10 +60,10 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     history1=history
     stats=f'''
         *******************
-        "Health": {health}
-        "Power": 1
-        "Strength": 20
-        "Magic": 24
+        Health: {health}
+        Power: 1
+        Strength: 20
+        Magic: 24
         *******************
     '''
     for ea in history:
@@ -102,14 +102,14 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
        
     lines = output.strip().strip("\n").split("\n")
     for i,line in enumerate(lines):
-        if line.startswith("Health: "):
+        if "Health: " in line:
             try:
                 new_health = line.split("Health: ")[1]
                 health=int(new_health)
                 print(health)
             except Exception as e:
                 print (f'{health}--Error :: {e}')
-        if line.startswith("Power: "):
+        if "Power: " in line:
             print(line)
         if line.startswith("3. "):
             print(line)