Added more options for postcard customization

pages/1_Generate_Holiday_Postcard.py

@@ -5,6 +5,8 @@ from PIL import Image, ImageDraw, ImageFont
 from io import BytesIO
 import streamlit as st
 import textwrap
+from PIL import ImageColor
+
 
 # Set page configuration
 st.set_page_config(page_title="FLUX Image Generation Tool", page_icon="🎇")
@@ -55,6 +57,29 @@ holiday_prompts = [
     "A lively Independence Day celebration with colorful fireworks lighting up the night sky, flags waving in the breeze, and a peaceful lakeside reflecting the celebration"
 ]
 
+# Example messages corresponding to each holiday prompt
+example_messages = [
+    "Wishing you peace and serenity this winter season. May the snow bring you joy and tranquility!",
+    "May your Christmas be as cozy and warm as a cabin in the woods, filled with holiday lights and love.",
+    "Warm wishes for the holidays! May your hearth and home be filled with love and joy this season.",
+    "Happy Hanukkah! May the light of the menorah shine brightly and bring joy to your heart.",
+    "Wishing you a vibrant New Year's Eve filled with excitement, fireworks, and joy for the year ahead!",
+    "May the magic and mystery of Samhain fill your spirit with wonder. Have a mystical holiday!",
+    "Celebrating the joy of Día de Muertos! May the memories of your loved ones bring peace and happiness.",
+    "Wishing you a Yule season filled with the warmth of the Norse tradition and the beauty of snowy forests.",
+    "Happy Diwali! May the lights of the diyas and the colors of rangoli bring you prosperity and joy.",
+    "Wishing you a reflective and peaceful Kwanzaa. May your home be filled with love, light, and unity.",
+    "Happy Thanksgiving! May your table be filled with gratitude, and your heart with the warmth of togetherness.",
+    "Wishing you a peaceful Winter Solstice night under the full moon. May the season bring you reflection and rest.",
+    "May the luck of the Irish be with you! Wishing you a joyful St. Patrick's Day filled with rainbows and gold.",
+    "Happy Lunar New Year! May the dragon dancers bring joy, and the fireworks light up a prosperous new year.",
+    "Wishing you a peaceful Easter surrounded by blooming flowers and the warmth of springtime sunshine.",
+    "Happy Valentine's Day! May your day be filled with love, glowing candles, and heartwarming moments.",
+    "Wishing you a joyful Holi! May the colors of the festival fill your life with vibrancy and happiness.",
+    "Ramadan Mubarak! May the crescent moon bring you peace, and the lanterns light up your spiritual path.",
+    "Happy Independence Day! May the fireworks light up the sky, and may freedom continue to inspire us all."
+]
+
 
 
 # Function to make requests to the FLUX models
@@ -90,13 +115,23 @@ def wrap_text(text, max_chars):
     return "\n".join(textwrap.fill(line, width=max_chars) for line in text.split("\n"))
 
 # Function to add the holiday message to the image
-def add_custom_message(image, message, font_path, font_size, position_vertical, position_horizontal, max_chars, bg_color, font_color):
+from PIL import ImageColor, ImageDraw, ImageFont, Image
+
+# Function to add the holiday message to the image
+def add_custom_message(image, message, font_path, font_size, position_vertical, position_horizontal, max_chars, bg_color, font_color, alpha):
     try:
         font = ImageFont.truetype(font_path, font_size)
     except IOError:
         font = ImageFont.load_default()
 
-    draw = ImageDraw.Draw(image)
+    # Convert image to RGBA if it's not already
+    if image.mode != "RGBA":
+        image = image.convert("RGBA")
+
+    # Create an overlay image with the same size and transparency (RGBA mode)
+    overlay = Image.new("RGBA", image.size, (255, 255, 255, 0))  # Fully transparent
+
+    draw = ImageDraw.Draw(overlay)
 
     # Wrap the text based on the max_chars input
     message = wrap_text(message, max_chars)
@@ -105,7 +140,7 @@ def add_custom_message(image, message, font_path, font_size, position_vertical,
 
     # Calculate the text bounding box dynamically
     text_lines = message.split("\n")
-    line_height = draw.textbbox((0, 0), "A", font=font)[3]  # Get height of one line of text (bbox returns top, left, bottom, right)
+    line_height = draw.textbbox((0, 0), "A", font=font)[3]  # Get height of one line of text
     total_text_height = line_height * len(text_lines)
     text_width = max([draw.textbbox((0, 0), line, font=font)[2] for line in text_lines])
 
@@ -128,14 +163,23 @@ def add_custom_message(image, message, font_path, font_size, position_vertical,
     # Define the padding for the background box behind the text
     padding = 10
 
-    # Draw the background rectangle (the text box) with user-specified background color
-    draw.rectangle([x_pos - padding, y_pos - padding, x_pos + text_width + padding, y_pos + total_text_height + padding], fill=bg_color)
+    # Convert the user-specified background color to RGBA and apply the alpha transparency value
+    bg_color_rgba = (*ImageColor.getrgb(bg_color), alpha)  # Apply user-controlled transparency
+
+    # Draw the semi-transparent background rectangle on the overlay
+    draw.rectangle([x_pos - padding, y_pos - padding, x_pos + text_width + padding, y_pos + total_text_height + padding], fill=bg_color_rgba)
 
     # Draw the text line by line with the user-specified font color
     for i, line in enumerate(text_lines):
         draw.text((x_pos, y_pos + i * line_height), line, font=font, fill=font_color)
 
-    return image
+    # Composite the overlay with the original image
+    combined = Image.alpha_composite(image, overlay)
+
+    return combined.convert("RGB")  # Convert back to RGB for saving/display
+
+
+
 
 # Streamlit UI
 st.title("FLUX Image Generation")
@@ -147,10 +191,13 @@ custom_prompt = st.text_input("Enter your custom prompt") if selected_prompt ==
 prompt = custom_prompt if selected_prompt == "Custom" else selected_prompt
 
 # Dropdown to select the model
-model_choice = st.selectbox("Select the model:", ["flux-1-dev", "flux-1-schnell"])
+model_choice = st.selectbox("Select the model:", ["flux-1-schnell", "flux-1-dev"])
+
+# Dropdown to select an example message or write a custom one
+selected_message = st.selectbox("Choose an example message or write your own:", options=["Custom"] + example_messages)
+message = st.text_input("Enter a holiday message to add:", value=selected_message if selected_message != "Custom" else "")
 
 # Additional inputs for customizing the message
-message = st.text_input("Enter a holiday message to add:")
 font_choice = st.selectbox("Select a font:", list(fonts.keys()))
 font_size = st.slider("Select font size:", 10, 100, 40)
 max_chars = st.slider("Max characters per line:", 10, 100, 40)  # Slider to select character wrap limit
@@ -159,10 +206,14 @@ max_chars = st.slider("Max characters per line:", 10, 100, 40)  # Slider to sele
 bg_color = st.color_picker("Pick a background color for the text box:", "#FFFFFF")
 font_color = st.color_picker("Pick a font color:", "#000000")
 
+# **Add a transparency slider for controlling the alpha channel**
+alpha = st.slider("Select transparency level for the background (0: fully transparent, 255: fully opaque)", 0, 255, 220)
+
 # Position options for the message (vertical and horizontal)
 position_vertical = st.radio("Select message vertical position:", ["Top", "Center", "Bottom"])
 position_horizontal = st.radio("Select message horizontal position:", ["Left", "Center", "Right"])
 
+
 # Button to generate images
 if st.button("Generate Image"):
     if not prompt.strip():
@@ -182,14 +233,14 @@ if st.button("Generate Image"):
                 # Add the holiday message to the generated image
                 image_with_message = add_custom_message(
                     generated_image.copy(), message, font_path, font_size, 
-                    position_vertical, position_horizontal, max_chars, bg_color, font_color
+                    position_vertical, position_horizontal, max_chars, bg_color, font_color, alpha
                 )
 
                 # Display the image with the message
                 st.image(image_with_message, caption=f"Generated using {model_choice} with custom message", use_column_width=True)
 
                 # Preview the placement
-                st.write(f"Message preview (vertical position: {position_vertical}, horizontal position: {position_horizontal}, font: {font_choice}, size: {font_size}, max chars: {max_chars}, bg color: {bg_color}, font color: {font_color})")
+                st.write(f"Message preview (vertical position: {position_vertical}, horizontal position: {position_horizontal}, font: {font_choice}, size: {font_size}, max chars: {max_chars}, bg color: {bg_color}, font color: {font_color}, transparency: {alpha})")
                 
         except Exception as e:
             st.error(f"An error occurred: {e}")