“Update”

colab.ipynb

@@ -176,6 +176,24 @@
     "# Attach the button click event handler\n",
     "generate_button.on_click(on_generate_button_clicked)\n",
     "\n",
+    "def save_image(image, model_label, seed, prompt_label, team):\n",
+    "    \"\"\"\n",
+    "    Save the generated image with a timestamped filename.\n",
+    "\n",
+    "    Args:\n",
+    "        image (PIL.Image.Image): The generated image.\n",
+    "        model_label (str): The label of the selected model.\n",
+    "        prompt_label (str): The seed. The label of the selected prompt.\n",
+    "        team (str): The selected team (\"Red\" or \"Blue\").\n",
+    "\n",
+    "    Returns:\n",
+    "        str: The filename of the saved image.\n",
+    "    \"\"\"\n",
+    "    timestamp = datetime.now().strftime(\"%Y%m%d_%H%M%S\")\n",
+    "    output_filename = f\"{timestamp}_{model_label.replace(' ', '_').lower()}_{seed}_{prompt_label.replace(' ', '_').lower()}_{team.lower()}.png\"\n",
+    "    image.save(output_filename)\n",
+    "    return output_filename\n",
+    "\n",
     "# Display the widgets\n",
     "display(prompt_dropdown, team_dropdown, model_dropdown, custom_prompt_input, seed_input, randomize_seed_checkbox, generate_button, output)"
    ]

src/img_gen_colab.py

@@ -74,7 +74,7 @@ def generate_image(prompt, team_color, model_name, height, width, num_inference_
     except Exception as e:
         return f"An error occurred: {e}"
 
-def save_image(image, model_label, prompt_label, team):
+def save_image(image, model_label, seed, prompt_label, team):
     """
     Save the generated image with a timestamped filename.
 
@@ -88,6 +88,6 @@ def save_image(image, model_label, prompt_label, team):
         str: The filename of the saved image.
     """
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    output_filename = f"{timestamp}_{model_label.replace(' ', '_').lower()}_{prompt_label.replace(' ', '_').lower()}_{team.lower()}.png"
+    output_filename = f"{timestamp}_{model_label.replace(' ', '_').lower()}_{seed}_{prompt_label.replace(' ', '_').lower()}_{team.lower()}.png"
     image.save(output_filename)
     return output_filename