refining stat generation

app.py

@@ -42,18 +42,6 @@ def char_description(prompt):
     output1 = pipe(prompt1, return_full_text=False, max_new_tokens=256)
     prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {prompt}\n\nDescribe:"
     output2 = pipe(prompt2, return_full_text=False, max_new_tokens=256)
-    global strength
-    strength = random.randint(2,12)
-    global dexterity
-    dexterity = random.randint(2,12)
-    global constitution
-    constitution = random.randint(2,12)
-    global intelligence
-    intelligence = random.randint(2,12)
-    global wisdom
-    wisdom = random.randint(2,12)
-    global charisma
-    charisma = random.randint(2,12)
     global ac
     ac = random.randint(2,14)
     global hp
@@ -79,6 +67,9 @@ def rand_char_description():
     output1 = pipe(prompt1, return_full_text=False, max_new_tokens=256)
     prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {rand_race}\n\nDescribe:"
     output2 = pipe(prompt2, return_full_text=False, max_new_tokens=256)
+    return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
+
+def stats_gen():
     global strength
     strength = random.randint(2,12)
     global dexterity
@@ -91,10 +82,11 @@ def rand_char_description():
     wisdom = random.randint(2,12)
     global charisma
     charisma = random.randint(2,12)
-    return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
+    return strength, dexterity, constitution, intelligence, wisdom, charisma
+
 
 ##UI is fine as-is
-with gr.Blocks(live = True) as demo:
+with gr.Blocks() as demo:
     title = gr.HTML("<center><h1>NPC Generator</h1></center>")
     description = gr.HTML("<center>For tabletop GMs when players ask for a description of random people in the tavern.</center>")
     tags = gr.HTML("<center>| Backyard AI | Tiny Titan |</center>")
@@ -115,6 +107,7 @@ with gr.Blocks(live = True) as demo:
             intl_box = gr.Textbox(value = intelligence, label = "Intelligence", interactive = False)
             wis_box = gr.Textbox(value = wisdom, label = "Wisdom", interactive = False)
             char_box = gr.Textbox(value = charisma, label = "Charisma", interactive = False)
+            btn.click(fn = stats_gen, outputs = [stre_box, dex_box, con_box, intl_box, wis_box, char_box])
 
         with gr.Column(scale = 2):
             text_out1 = gr.Textbox(label = "Description")
@@ -136,6 +129,7 @@ with gr.Blocks(live = True) as demo:
                 dcs = gr.Textbox(value = smash, label = "DC to Smash")
                 prefs = gr.Textbox(value = pref_traits, label = "Advantage if:")
                 icks = gr.Textbox(value = ick_traits, label = "Disdvantage if:")
+                
 
 
 # Need to add export feature.