added new function to grab traits for bard stats

app.py

@@ -41,7 +41,7 @@ def char_description(prompt):
 
     prompt1 = f"{DESC_SYSTEM_PROMPT}\n\nCharacter Description: {prompt}\n\nDescribe:"
     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:"
+    prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {prompt}\n\nBackstory:"
     output2 = pipe(prompt2, return_full_text=False, max_new_tokens=256)
     return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
 
@@ -52,7 +52,7 @@ def rand_char_description():
     rand_race = random.choice(races)
     prompt1 = f"{DESC_SYSTEM_PROMPT}\n\nCharacter Description: {rand_race}\n\nDescribe:"
     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:"
+    prompt2 = f"{BACK_SYSTEM_PROMPT}\n\n{output1}\n\nUser prompt: {rand_race}\n\nBackstory:"
     output2 = pipe(prompt2, return_full_text=False, max_new_tokens=256)
     return output1[0]["generated_text"].strip(),output2[0]["generated_text"].strip()
 
@@ -80,19 +80,23 @@ def stats_gen_2():
     speed = '25 ft'
     return ac, hp, speed
 
+def traits_picker(filepath):
+    with open(filepath, 'r') as f:
+        items = [line.strip() for line in f if line.strip()]
+    count = random.randint(1, min(3, len(items)))
+    selected = random.sample(items,count)
+    return ', '.join(selected)
+
 def bard_stats_gen():
     global smash
     smash = random.randint(1,30)
-    input_file = open('traits.txt', 'r')
-    traits = input_file.readline()
     global pref_traits
-    picks = int(random.randint(1,3))
-    pref_traits = random.choice(traits, k=picks)
+    pref_traits = traits_picker('traits.txt')
     global ick_traits
-    picks = int(random.randint(1,3))
-    ick_traits = random.choice(traits, k=picks)
+    ick_traits = traits_picker('traits.txt')
     return smash, pref_traits, ick_traits
 
+
 ##UI is fine as-is
 with gr.Blocks() as demo:
     title = gr.HTML("<center><h1>NPC Generator</h1></center>")