Update app.py

app.py

@@ -20,7 +20,7 @@ def get_top_chunks(query):
 
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
 
-def respond(message, history, name, char_class, char_alignment):
+def respond(message, history, name, char_class, char_alignment, char_race):
     response = ""
     top_chunks = get_top_chunks(message)
     context = "\n".join(top_chunks)
@@ -30,7 +30,7 @@ def respond(message, history, name, char_class, char_alignment):
             "role": "system",
             "content": (
                 f"You are a chatbot that helps users create characters for role-playing games."
-                f"The user might also provide the following information, please include their inputs in your response: \n Character name: {name}, Character Class: {char_class}, Character Alignment: {char_alignment}"
+                f"The user might also provide the following information, please use the context of their inputs in your response: \n Character name: {name}, Character Class: {char_class}, Character Race: {char_race}, Character Alignment: {char_alignment}. If they don't include all the information, try to create a character that best fits the needs of their campaign party."
                 f"Use the following knowledge to inform your answers:\n\n{context}\n\n"
                 "Keep your answers under 300 words."
             )
@@ -92,11 +92,20 @@ with gr.Blocks(theme=theme) as chatbot:
             
             character_name = gr.Textbox(label = "Character Name", placeholder="Type your name here…", info ="optional")
             character_class = gr.Dropdown(['Barbarian', 'Bard', 'Cleric', 'Druid', 'Fighter', 'Monk', 'Paladin', 'Ranger', 'Rogue', 'Sorcerer', 'Warlock', 'Wizard'], label="Character Class", info="Choose one or more", multiselect=True)
-            character_alignment = gr.Radio(["Lawful Good", "Neutral Good", "Chaotic Good", "Lawful Neutral", "True Neutral", "Chaotic Neutral", "Lawful Evil", "Neutral Evil", "Chaotic Evil"], elem_classes="alignment_radio")
+            character_race = gr.Dropdown([
+        "Aarakocra", "Aasimar", "Bugbear", "Centaur", "Changeling", "Dragonborn", "Dwarf", "Elf", 
+        "Fairy", "Firbolg", "Genasi", "Githyanki", "Githzerai", "Gnome", "Goblin", "Goliath", 
+        "Half-Elf", "Half-Orc", "Halfling", "Hobgoblin", "Human", "Kenku", "Kobold", "Leonin", 
+        "Lizardfolk", "Minotaur", "Orc", "Owlin", "Satyr", "Shadar-kai", "Shifter", "Tabaxi", 
+        "Tiefling", "Tortle", "Triton", "Warforged", "Yuan-ti"
+    ], multiselect=True)
+            character_alignment = gr.Dropdown(["Lawful Good", "Neutral Good", "Chaotic Good", "Lawful Neutral", "True Neutral", "Chaotic Neutral", "Lawful Evil", "Neutral Evil", "Chaotic Evil"], elem_classes="alignment_radio")
+
+            
         with gr.Column(scale=2):
             gr.ChatInterface(
                 fn=respond,
-                additional_inputs=[character_name, character_class, character_alignment],  # Pass name into function!
+                additional_inputs=[character_name, character_class, character_race, character_alignment],  # Pass name into function!
                 type="messages",
                 examples=None
             )