added bard stats and changed regular stats to random generation. Intern forgot that the model wont consistently give just an integer.

app.py

@@ -31,6 +31,12 @@ constitution = ''
 intelligence = ''
 wisdom = ''
 charisma = ''
+ac = ''
+hp = ''
+smash = ''
+pref_traits = ''
+ick_traits = ''
+
     
 
 @spaces.GPU
@@ -41,23 +47,31 @@ def char_description(prompt):
     prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {prompt}\n\nDescribe:"
     output2 = pipe(prompt2, return_full_text=False, max_new_tokens=128)
     global strength
-    strength_prompt = f"{STAT_SYSTEM_PROMPT} Strength for {output1}"
-    strength = pipe(strength_prompt, return_full_text=False)
+    strength = random.randint[2,12]
     global dexterity
-    dexterity_prompt = f"{STAT_SYSTEM_PROMPT} Dexterity for {output1}"
-    dexterity = int(pipe(dexterity_prompt, return_full_text=False))
+    dexterity = random.randint[2,12]
     global constitution
-    constitution_prompt = f"{STAT_SYSTEM_PROMPT} Constitution for {output1}"
-    constitution = int(pipe(constitution_prompt, return_full_text=False))
+    constitution = random.randint[2,12]
     global intelligence
-    intelligence_prompt = f"{STAT_SYSTEM_PROMPT} Intelligence for {output1}"
-    intelligence = int(pipe(intelligence_prompt, return_full_text=False))
+    intelligence = random.randint[2,12]
     global wisdom
-    wisdom_prompt = f"{STAT_SYSTEM_PROMPT} Wisdom for {output1}"
-    wisdom = int(pipe(wisdom_prompt, return_full_text=False))
+    wisdom = random.randint[2,12]
     global charisma
-    charisma_prompt = f"{STAT_SYSTEM_PROMPT} Charisma for {output1}"
-    charisma = int(pipe(charisma_prompt, return_full_text=False))
+    charisma = random.randint[2,12]
+    global ac
+    ac = random.randint[2,14]
+    global hp
+    hp = constitution + random.randint[1,6]
+    global smash
+    smash = random.randint[1,30]
+    input_file = open('traits.txt', 'r')
+    traits = input_file.readline()
+    global pref_traits
+    picks = random.randint[1,3]
+    pref_traits = random.choice(traits, k=picks)
+    global ick_traits
+    picks = random.randint[1,3]
+    ick_traits = random.choice(traits, k=picks)
     return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
 
 @spaces.GPU
@@ -70,23 +84,17 @@ def rand_char_description():
     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=128)
     global strength
-    strength_prompt = f"{STAT_SYSTEM_PROMPT} Strength for {output1}"
-    strength = pipe(strength_prompt, return_full_text=False)
+    strength = random.randint[2,12]
     global dexterity
-    dexterity_prompt = f"{STAT_SYSTEM_PROMPT} Dexterity for {output1}"
-    dexterity = int(pipe(dexterity_prompt, return_full_text=False))
+    dexterity = random.randint[2,12]
     global constitution
-    constitution_prompt = f"{STAT_SYSTEM_PROMPT} Constitution for {output1}"
-    constitution = int(pipe(constitution_prompt, return_full_text=False))
+    constitution = random.randint[2,12]
     global intelligence
-    intelligence_prompt = f"{STAT_SYSTEM_PROMPT} Intelligence for {output1}"
-    intelligence = int(pipe(intelligence_prompt, return_full_text=False))
+    intelligence = random.randint[2,12]
     global wisdom
-    wisdom_prompt = f"{STAT_SYSTEM_PROMPT} Wisdom for {output1}"
-    wisdom = int(pipe(wisdom_prompt, return_full_text=False))
+    wisdom = random.randint[2,12]
     global charisma
-    charisma_prompt = f"{STAT_SYSTEM_PROMPT} Charisma for {output1}"
-    charisma = int(pipe(charisma_prompt, return_full_text=False))
+    charisma = random.randint[2,12]
     return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
 
 
@@ -132,7 +140,7 @@ with gr.Blocks() as demo:
         
         with gr.Accordion("Bard Stats", open = False):
             with gr.Row():
-                dcs = gr.Textbox(label = "DC to Smash")
+                dcs = gr.Textbox(value = smash, label = "DC to Smash")
                 prefs = gr.Textbox(label = "Advantage if:")
                 icks = gr.Textbox(label = "Disdvantage if:")
 

traits.txt

@@ -0,0 +1,56 @@
+Dragonborn
+Dwarf
+Eladrin
+Elf
+Fairy
+Firbolg
+Gnome
+Goblin
+Half-elf
+Half-orc
+Halfling
+Human
+Orc
+Tiefling
+Male
+Female
+Nonbinary
+Gender ambiguous
+Blue eyes
+Green eyes
+Brown eyes
+Yellow eyes
+Pink eyes
+Purple eyes
+Red eyes
+Orange eyes
+dexterity under 10
+constitution under 10
+intelligence under 10
+wisdom under 10
+charisma under 10
+strength over 10
+dexterity over 10
+constitution over 10
+intelligence over 10
+wisdom over 10
+charisma over 10
+Black hair
+Red hair
+Blonde hair
+Blue hair
+Green hair
+Pink hair
+Facial hair
+Clean shaven
+Gold over 100
+Gold under 100
+Married
+Unmarried
+Height under 4ft
+Height under 6ft
+Height over 4ft
+Height over 6ft
+strength under 10
+Under 30 years old
+Over 30 years old