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, character_type):
+def respond(message, history, name, char_class, char_alignemnt):
     response = ""
     top_chunks = get_top_chunks(message)
     context = "\n".join(top_chunks)
@@ -30,7 +30,7 @@ def respond(message, history, name, character_type):
             "role": "system",
             "content": (
                 f"You are a chatbot that helps users create characters for role-playing games."
-                f"The user wants their character to be called {name} with the character type {character_type}."
+                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"Use the following knowledge to inform your answers:\n\n{context}\n\n"
                 "Keep your answers under 300 words."
             )
@@ -57,6 +57,12 @@ def respond(message, history, name, character_type):
 
 # === GUI ===
 
+alignment_css="""
+#alignment_radio{
+    color: blue;
+}
+"""
+
 with gr.Blocks() as chatbot:
     with gr.Row(scale=1):
         with gr.Column(scale=1):  
@@ -67,12 +73,13 @@ with gr.Blocks() as chatbot:
                 show_download_button=False
             )
         with gr.Column(scale=1):
-            character_name = gr.Textbox(placeholder="Type your name here…")
-            charType = gr.CheckboxGroup(["dwarf", "paladin", "bard"], label="Character Type", info="What is their Character Type?")
+            character_name = gr.Textbox(label = "Character Name", placeholder="Type your name here…", info ="optional")
+            character_class = gr.CheckboxGroup(['Barbarian', 'Bard', 'Cleric', 'Druid', 'Fighter', 'Monk', 'Paladin', 'Ranger', 'Rogue', 'Sorcerer', 'Warlock', 'Wizard'], label="Character Class", info="Choose one or more")
+            character_alignment = gr.Radio(["Lawful Good", "Neutral Good", "Chaotic Good", "Lawful Neutral", "True Neutral", "Chaotic Neutral", "Lawful Evil", "Neutral Evil", "Chaotic Evil"], css=alignment_css, elem_id="alignment_radio")
 
             gr.ChatInterface(
                 fn=respond,
-                additional_inputs=[character_name, charType],  # Pass name into function!
+                additional_inputs=[character_name, character_class, character_alignment],  # Pass name into function!
                 type="messages",
                 examples=None,
                 title="Character Creator",