Update app.py

app.py

@@ -65,7 +65,7 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     print(f'cnt:: {cnt}')
     if cnt > MAX_HISTORY:
         history1 = compress_history(str(history), temperature, max_new_tokens, top_p, repetition_penalty)
-    formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,health=health)}, {prompt}", history)
+    formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1)}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
@@ -76,13 +76,19 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
         else:
             yield "",[(prompt,output)],health
 
-    lines = output.strip().strip("\n").split("\n")
     #history=""
+    #formatted_prompt2 = format_prompt(f"{ADJUST_STATS.format(history=output,health=health)}, {prompt}", history)
+    stream2 = client.text_generation(f"{ADJUST_STATS.format(history=output,health=health)}", **generate_kwargs, stream=True, details=True, return_full_text=False)  
+    output2=""
+    for response in stream2:
+        output2 += response.token.text
+       
+    lines = output2.strip().strip("\n").split("\n")
     for i,line in enumerate(lines):
         if line.startswith("Health: "):
             try:
                 new_health = line.split("Health: ")[1]
-                health+=int(new_health)
+                health=int(new_health)
                 print(health)
             except Exception as e:
                 print (f'{health}--Error :: {e}')
@@ -94,7 +100,8 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
             print(line)
         if line.startswith("5. "):
             print(line)            
-    
+        else: 
+            print(f'Line:: {line}')
     if history:
         history.append((prompt,output))
         yield "",history,health