Update app.py

app.py

@@ -58,14 +58,11 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     )
     cnt=0
     history1=history
-    stats=f'''
-        *******************
-        Health: {health}
-        Power: 1
-        Strength: 20
-        Magic: 24
-        *******************
-    '''
+
+    stats="*******************\n"
+    for eac in health:
+        stats+=f'{eac}\n'
+    stats+="*******************\n"
     for ea in history:
         print (ea)
         for l in ea:
@@ -81,9 +78,9 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     for response in stream:
         output += response.token.text
         if history:
-            yield "",[(prompt,output)],health
+            yield "",[(prompt,output)],stats
         else:
-            yield "",[(prompt,output)],health
+            yield "",[(prompt,output)],stats
     generate_kwargs2 = dict(
         temperature=temperature,
         max_new_tokens=128,
@@ -101,32 +98,36 @@ def generate(prompt, history,max_new_tokens,health,temperature=temperature,top_p
     #    output2 += response.token.text
        
     lines = output.strip().strip("\n").split("\n")
+    skills=[]
     for i,line in enumerate(lines):
-        if "Health: " in line:
+        if ": " in line:
             try:
-                new_health = line.split("Health: ")[1]
-                health=int(new_health)
-                print(health)
+                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)
+                print(skills)
             except Exception as e:
-                print (f'{health}--Error :: {e}')
-        if "Power: " in line:
-            print(line)
-        if line.startswith("3. "):
-            print(line)
-        if line.startswith("4. "):
-            print(line)
-        if line.startswith("5. "):
-            print(line)            
-        else: 
+                print (f'--Error :: {e}')
             print(f'Line:: {line}')
+    stats=skills
     if history:
         history.append((prompt,output))
-        yield "",history,health
+        yield "",history,stats
     else:
-        yield "",[(prompt,output)],health
+        yield "",[(prompt,output)],stats
 
 def clear_fn():
     return None,None
+
+base_stats=[
+    {"Health":100},
+    {"Power":20},
+    {"Strength":24},
+]
+
+    
 with gr.Blocks() as app:
     gr.HTML("""<center><h1>Mixtral 8x7B RPG</h1><h3>Role Playing Game Master</h3>""")
     chatbot = gr.Chatbot(label="Mixtral 8x7B Chatbot",show_copy_button=True)
@@ -142,13 +143,13 @@ with gr.Blocks() as app:
         clear_btn = gr.Button("Clear")
     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()
+    json_out=gr.JSON(value=base_stats)
     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,health],[prompt,chatbot,health])
+    go=button.click(generate,[prompt,chatbot,tokens,json_out],[prompt,chatbot,json_out])
     stop_button.click(None,None,None,cancels=[go])
 app.launch(show_api=False)