Update app.py

app.py

@@ -58,11 +58,12 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     )
     cnt=0
     history1=history
-
+    '''
     stats="*******************\n"
     for eac in health:
         stats+=f'{eac}\n'
     stats+="*******************\n"
+    '''
     for ea in history:
         print (ea)
         for l in ea:
@@ -71,16 +72,16 @@ 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, top_p, repetition_penalty)
-    formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,stats=stats,dice=random.randint(1,100))}, {prompt}", history)
+    formatted_prompt = format_prompt(f"{GAME_MASTER.format(history=history1,stats=health,dice=random.randint(1,100))}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
     for response in stream:
         output += response.token.text
         if history:
-            yield "",[(prompt,output)],stats
+            yield "",[(prompt,output)],stats,None
         else:
-            yield "",[(prompt,output)],stats
+            yield "",[(prompt,output)],stats,None
     generate_kwargs2 = dict(
         temperature=temperature,
         max_new_tokens=128,
@@ -99,24 +100,29 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
        
     lines = output.strip().strip("\n").split("\n")
     skills=[]
+    new_stats="*******************"
     for i,line in enumerate(lines):
         if ": " in line:
             try:
                 lab_1 = line.split(": ")[0]
+                
                 skill_1 = line.split(": ")[1].split(" ")[0]
                 skill_1=int(skill_1)
                 skill ={lab_1:skill_1}
                 skills.append(skill)
+                
+                new_stat += f'{lab_1}: {skill_1}'
+                
                 print(skills)
             except Exception as e:
                 print (f'--Error :: {e}')
             print(f'Line:: {line}')
-    stats=skills
+    stats=new_stat
     if history:
         history.append((prompt,output))
-        yield "",history,stats
+        yield "",history,stats,skills
     else:
-        yield "",[(prompt,output)],stats
+        yield "",[(prompt,output)],stats,skills
 
 def clear_fn():
     return None,None
@@ -144,12 +150,12 @@ with gr.Blocks() as app:
     with gr.Row():
         tokens = gr.Slider(label="Max new tokens",value=2096,minimum=0,maximum=1048*10,step=64,interactive=True,info="The maximum numbers of new tokens")
     json_out=gr.JSON(value=base_stats)
-    health=gr.Number(value=100)
+    char_stats=gr.Textbox()
     #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,json_out],[prompt,chatbot,json_out])
+    go=button.click(generate,[prompt,chatbot,tokens,char_stats],[prompt,chatbot,char_stats,json_out])
     stop_button.click(None,None,None,cancels=[go])
 app.launch(show_api=False)