“Update”

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"Winning Team Text: {winning_team_text}")
+    print(f"Team Color: {team_color}")
 
     prompt = prompt.format(team_color=team_color, enemy_color=enemy_color)
 
@@ -74,20 +74,3 @@ 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, seed, prompt_label, team):
-    """
-    Save the generated image with a timestamped filename.
-
-    Args:
-        image (PIL.Image.Image): The generated image.
-        model_label (str): The label of the selected model.
-        prompt_label (str): The label of the selected prompt.
-        team (str): The selected team ("Red" or "Blue").
-
-    Returns:
-        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()}_{seed}_{prompt_label.replace(' ', '_').lower()}_{team.lower()}.png"
-    image.save(output_filename)
-    return output_filename