Update app.py

app.py

@@ -40,7 +40,7 @@ def compress_history(history,temperature=temperature,top_p=top_p,repetition_pena
 
 MAX_HISTORY=100
 
-def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
+def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
     temperature = float(temperature)
     if temperature < 1e-2:
         temperature = 1e-2
@@ -65,7 +65,7 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=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)}, {prompt}", history)
+    formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,health=health)}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
@@ -79,8 +79,13 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,
     lines = output.strip().strip("\n").split("\n")
     #history=""
     for i,line in enumerate(lines):
-        if line.startswith("1. "):
-            print(line)
+        if line.startswith("Health: "):
+            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("2. "):
             print(line)
         if line.startswith("3. "):
@@ -92,9 +97,9 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,
     
     if history:
         history.append((prompt,output))
-        yield "",history
+        yield "",history,health
     else:
-        yield "",[(prompt,output)]
+        yield "",[(prompt,output)],health
 
 def clear_fn():
     return None,None
@@ -114,12 +119,12 @@ with gr.Blocks() as app:
     with gr.Row():
         tokens = gr.Slider(label="Max new tokens",value=1048,minimum=0,maximum=1048*10,step=64,interactive=True,info="The maximum numbers of new tokens")
     json_out=gr.JSON()
-    e_box=gr.Textbox()
+    health=gr.Number(value=100)
     #text=gr.JSON()
     #inp_query.change(search_models,inp_query,models_dd)
     #test_b=test_btn.click(itt,url,e_box)
     clear_btn.click(clear_fn,None,[prompt,chatbot])
-    go=button.click(generate,[prompt,chatbot,tokens],[prompt,chatbot])
+    go=button.click(generate,[prompt,chatbot,tokens,health],[prompt,chatbot,health])
     stop_button.click(None,None,None,cancels=[go])
 app.launch(show_api=False)