Add flag for backstory generator

app.py

@@ -8,7 +8,7 @@ filename = "mistral-7b-instruct-v0.1.Q4_K_M.gguf"
 model_file_path = hf_hub_download(repo_id=repo_id, filename=filename)
 
 
-def generate_characters():
+def generate_characters(generate_backstories=False):
     class_list = ["Barbarian", "Bard", "Cleric", "Druid", "Fighter",
                   "Monk", "Paladin", "Ranger", "Rogue", "Sorcerer", "Warlock", "Wizard"]
     sub_class_list = ["Barbarian - Beserker", "Barbarian - Wildheart", "Barbarian - Wild Magic", "Bard - College of Lore", "Bard - College of Valour", "Bard - College of Swords", "Cleric - Life Domain", "Cleric -  Light Domain", "Cleric - Trickery Domain", "Cleric - Knowledge Domain", "Cleric - Nature Domain", "Cleric - Tempest Domain", "Cleric - War Domain", "Druid - Circle of the moon", "Druid - Circle of the Land", "Druid - Circle of the Spores", "Fighter - Battle Master", "Fighter - Eldritch Knight", "Fighter - Champion", "Monk - Way of the open hand", "Monk - Way of shadow", "Monk - Way of the four elements",
@@ -44,22 +44,23 @@ def generate_characters():
 
     characters = [generate_character() for _ in range(2)]
     characters.append(generate_multiclass_character())
-    
-    def create_character_output(character):
+
+    def create_character_output(character, generate_backstory=False):
         output = ""
         for key, value in character.items():
             output += f"- **{key}:** {value}\n"
         output += "\n"
-        backstory = generate_backstory(character)
-        # Remove everything up to and including `[/INST]\n`
-        output += backstory[backstory.find("[/INST]\n") + 8:]
-        
+        if generate_backstory:
+            backstory = generate_backstory(character)
+            # Remove everything up to and including `[/INST]\n`
+            output += backstory[backstory.find("[/INST]\n") + 8:]
+
         return output
 
     output = ""
     for idx, char in enumerate(characters, 1):
         output += f"### Character {idx}:\n"
-        output += create_character_output(char)
+        output += create_character_output(char, generate_backstories)
 
     return output
 
@@ -95,6 +96,8 @@ def generate_backstory(character):
 if __name__ == "__main__":
     iface = gr.Interface(fn=generate_characters,
                          title="D&D Character Generator",
+                         inputs=[gr.inputs.Checkbox(
+                             label="Generate Backstories?")],
                          description="Generates 3 random D&D characters.",
-                         inputs=[], outputs="markdown")
+                         outputs="markdown")
     iface.launch()