gender diversity

app.py

@@ -73,7 +73,9 @@ def npc_randomize():
 
     characterClass=random.choice(classes)
 
-    return name,race,characterClass
+    pronoun=random.choices("he","she","they",weights=[0.45,0.45,0.1],k=1)[0]
+
+    return name,race,characterClass,pronoun
 
 
     
@@ -81,9 +83,9 @@ def npc_randomize():
 
 
 
-def npc_generate(name,race,characterClass):
+def npc_generate(name,race,characterClass,pronoun):
 
-  desc="{name} is a {race} {characterClass}".format(name=name,race=race,characterClass=characterClass)
+  desc="{name} is a {race} {characterClass}, {pronoun}".format(name=name,race=race,characterClass=characterClass,pronoun=pronoun)
 
   p = prompt + "\n"+desc
   print(f"*****Inside poem_generate - Prompt is :{p}")
@@ -142,9 +144,10 @@ with demo:
     )
   with gr.Row():
     
-    input_name = gr.Textbox(placeholder="Drizzt")
-    input_race = gr.Textbox(placeholder="dark elf")
-    input_class = gr.Textbox(placeholder="ranger")
+    input_name = gr.Textbox(label="name",placeholder="Drizzt")
+    input_race = gr.Textbox(label="race",placeholder="dark elf")
+    input_class = gr.Textbox(label="class",placeholder="ranger")
+    input_pronoun = gr.Textbox(label="pronoun",placeholder="he")
 
     poem_txt = gr.Textbox(lines=7)
     output_image = gr.Image(type="filepath", shape=(256,256))
@@ -153,8 +156,8 @@ with demo:
   b1 = gr.Button("Generate NPC")
   b2 = gr.Button("Generate Image")
 
-  b0.click(npc_randomize,inputs=[],outputs=[input_name,input_race,input_class])
-  b1.click(npc_generate, inputs=[ input_name,input_race,input_class], outputs=poem_txt)
+  b0.click(npc_randomize,inputs=[],outputs=[input_name,input_race,input_class,input_pronoun])
+  b1.click(npc_generate, inputs=[ input_name,input_race,input_class,input_pronoun], outputs=poem_txt)
   b2.click(poem_to_image, poem_txt, output_image)
   #examples=examples