“Update”

config/prompts.py

@@ -7,7 +7,7 @@ prompts = [
     },
     {
         "alias": "Forest Battle",
-        "text": "A fierce battle between two armies in a dense forest, with knights wielding swords and axes, horses rearing, and the ground covered in mud and blood. The winning army is dressed in {team_color} armor and banners. The enemy army, dressed in {enemy_color} armor, is locked in brutal combat, with soldiers fighting hand-to-hand amidst the trees. Arrows whiz past, and the sounds of clashing steel echo through the forest. Injured soldiers scream in pain, and the forest is littered with broken weapons and shields. Unreal Engine render style, photorealistic, realistic fantasy style." 
+        "text": "A fierce battle between two armies in a dense forest, with knights wielding swords and axes, horses rearing, and the ground covered in mud and blood.  The winning army is dressed in {team_color} armor and banners.. The enemy army, dressed in {enemy_color} armor, is locked in brutal combat, with soldiers fighting hand-to-hand amidst the trees. Arrows whiz past, and the sounds of clashing steel echo through the forest. Injured soldiers scream in pain, and the forest is littered with broken weapons and shields. Unreal Engine render style, photorealistic, realistic fantasy style." 
     },
     {
         "alias": "Boiling Oil Defense",

src/img_gen_colab.py

@@ -36,7 +36,7 @@ def generate_image(prompt, team_color, model_name, height, width, num_inference_
     print("Original Prompt:")
     print(prompt)
     print(f"Enemy Color: {enemy_color}")
-    print(f"Team Color: {team_color}")
+    print(f"Team Color: {team_color.lower()}")
 
     prompt = prompt.format(team_color=team_color.lower(), enemy_color=enemy_color)