randomize character creation

app.py

@@ -1,7 +1,9 @@
+from asyncio import constants
 import gradio as gr
 import requests
 import os 
 import re
+import random
 
 # GPT-J-6B API
 API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
@@ -17,6 +19,64 @@ Bremen is a human wizard, he wears a blue robe and carries a wand.
 examples = [["river"], ["night"], ["trees"],["table"],["laughs"]]
 
 
+def npc_randomize():
+    #name is a random combination of syllables
+    vowels = "aeiou".split()
+    constants = "bcdfghjklmnpqrstvwxyz".split()
+    seperators="-'".split()
+    name =""
+    for i in range(random.randint(2,4)):
+        name += random.choice(constants)
+        name += random.choice(vowels)
+        if random.random()<0.5:
+            name += random.choice(constants)
+        if random.random()<0.1:
+            name += random.choice(seperators)
+    
+    races="""
+    Dwarf
+    Elf
+    Halfling
+    Human
+    Dragonborn
+    Gnome
+    Half-elf
+    Half-orc
+    Tiefling
+    Aarakocra
+    Genasi
+    Goliath"""
+
+    races=[x.strip() for x in races]
+
+    race=random.choice(races)
+
+
+    classes="""Barbarian
+    Bard
+    Cleric
+    Druid
+    Fighter
+    Monk
+    Paladin
+    Ranger
+    Rogue
+    Sorcerer
+    Warlock
+    Wizard""".split("\n")
+    
+    classes=[x.strip() for x in classes]
+
+    characterClass=random.choice(classes)
+
+    return name,race,characterClass
+
+
+    
+
+
+
+
 def npc_generate(name,race,characterClass):
 
   desc="{name} is a {race} {characterClass}".format(name=name,race=race,characterClass=characterClass)
@@ -85,9 +145,11 @@ with demo:
     poem_txt = gr.Textbox(lines=7)
     output_image = gr.Image(type="filepath", shape=(256,256))
   
+  b0 = gr.Button("Randomize name,race and class")
   b1 = gr.Button("Generate NPC")
   b2 = gr.Button("Generate Image")
 
+  b0.click(npc_randomize,outputs=[input_name,input_race,input_class])
   b1.click(npc_generate, inputs=[ input_name,input_race,input_class], outputs=poem_txt)
   b2.click(poem_to_image, poem_txt, output_image)
   #examples=examples