Update app.py

app.py

@@ -140,31 +140,38 @@ def generate_poster():
         "Corporate/Professional", "Retro/Vintage", "Modern/Contemporary", "Illustrative/Artistic"
     ])
     
+    description = st.text_area("Enter prompt for Advertisement:")
     header = st.text_input("Enter Header for Advertisement:")
-    placeholder_text = st.text_input("Enter placeholder text:")
-    description = st.text_area("Enter Description for Advertisement:")
+    sub_header = st.text_input("Enter Sub-header for Advertisement:")    
+    # Allow multiple user prompts
+    user_prompts = []
+    num_prompts = st.number_input("Number of Text Descriptions", min_value=1, max_value=80, value=1)
+    for i in range(num_prompts):
+        user_prompt = st.text_area(f"Enter Descriptions to display in the image (Descriptions {i+1}):")
+        user_prompts.append(user_prompt)
+    
     logo = st.file_uploader("Upload Logo (optional)", type=['png', 'jpg', 'jpeg'])
     
     # Add logo position selection
     logo_position = st.selectbox("Select Logo Position", [
-        "Top Left", "Top Middle", "Top Right",
+        "None", "Top Left", "Top Middle", "Top Right",
         "Left Middle", "Right Middle",
         "Bottom Left", "Bottom Middle", "Bottom Right"
     ])
     
-    # Add padding slider
-    padding_percent = st.slider("Logo Padding (%)", 1, 10, 2)
-    
     num_images = st.number_input("Number of Images", min_value=4, max_value=8, value=4)
-    quality_factor = st.number_input("Quality Factor", min_value=1, max_value=4, value=1)
 
     if st.button("Generate Images"):
         if post_type == "Other":
-            full_prompt = f"{header}. A {design_style.lower()} style post with the following elements: {description}, aspect ratio: {aspect_ratio}, dimensions: {width}x{height}, 4k high resolution. Placeholder: {placeholder_text}"
+            full_prompt = f"Create an image with the following text as the header: '{header}'. Use '{sub_header}' as the sub-header. Use a {design_style.lower()} style for the post. Include the following text as the description: {description}. The image should have an aspect ratio of {aspect_ratio} and dimensions of {width}x{height}. Ensure the image is in 4k high resolution."
         else:
-            full_prompt = f"{header}. A {design_style.lower()} style {post_type.lower()} with the following elements: {description}, aspect ratio: {aspect_ratio}, dimensions: {width}x{height}, 4k high resolution. Placeholder: {placeholder_text}"
+            full_prompt = f"Create a {post_type.lower()} with the following text as the header: '{header}'. Use '{sub_header}' as the sub-header. Use a {design_style.lower()} style for the post. Include the following text as the description: {description}. The image should have an aspect ratio of {aspect_ratio} and dimensions of {width}x{height}. Ensure the image is in 4k high resolution."
+
+        # Add user prompts to full_prompt if provided
+        for i, user_prompt in enumerate(user_prompts):
+            if user_prompt:
+                full_prompt += f" Include the following text in the image (Prompt {i+1}): '{user_prompt}'."
 
-        generated_images = []
         for i in range(num_images):
             with st.spinner(f"Generating image {i+1}..."):
                 logger.info(f"Generating image {i+1} with prompt: {full_prompt}")
@@ -174,20 +181,28 @@ def generate_poster():
                 
                 if image_bytes:
                     image = Image.open(io.BytesIO(image_bytes))
-                    if quality_factor > 1:
-                        image = increase_image_quality(image, quality_factor)
                     
-                    # Add logo to the selected position if provided
+                    # Add logo if provided, or place it randomly if not specified
                     if logo:
                         logo_image = Image.open(logo)
                         logo_width = int(image.width * 0.15)  # 15% of the image width
                         logo_height = int(logo_image.height * (logo_width / logo_image.width))
                         logo_image = logo_image.resize((logo_width, logo_height), Image.LANCZOS)
                         
-                        padding = int(image.width * (padding_percent / 100))  # Convert percentage to pixels
+                        padding = int(image.width * 0.02)  # Fixed 2% padding
                         
-                        # Calculate logo position based on user selection
-                        if logo_position == "Top Left":
+                        if logo_position == "None":
+                            # If "None" is selected, randomly choose a position
+                            random_positions = [
+                                (padding, padding),  # Top Left
+                                ((image.width - logo_width) // 2, padding),  # Top Middle
+                                (image.width - logo_width - padding, padding),  # Top Right
+                                (padding, image.height - logo_height - padding),  # Bottom Left
+                                ((image.width - logo_width) // 2, image.height - logo_height - padding),  # Bottom Middle
+                                (image.width - logo_width - padding, image.height - logo_height - padding)  # Bottom Right
+                            ]
+                            position = random.choice(random_positions)
+                        elif logo_position == "Top Left":
                             position = (padding, padding)
                         elif logo_position == "Top Middle":
                             position = ((image.width - logo_width) // 2, padding)
@@ -203,38 +218,64 @@ def generate_poster():
                             position = ((image.width - logo_width) // 2, image.height - logo_height - padding)
                         else:  # Bottom Right
                             position = (image.width - logo_width - padding, image.height - logo_height - padding)
-                        
-                        # Create a new image with an alpha channel
-                        combined_image = Image.new('RGBA', image.size, (0, 0, 0, 0))
-                        combined_image.paste(image, (0, 0))
-                        
+                    
+                    # Create a new image with an alpha channel
+                    combined_image = Image.new('RGBA', image.size, (0, 0, 0, 0))
+                    combined_image.paste(image, (0, 0))
+                    
+                    if logo:
                         # Convert logo to RGBA if it's not already
                         if logo_image.mode != 'RGBA':
                             logo_image = logo_image.convert('RGBA')
                         
                         combined_image.paste(logo_image, position, logo_image)
+                    
+                    # Convert back to RGB for compatibility
+                    image = combined_image.convert('RGB')
+                    
+                    # Display generated image
+                    col1, col2 = st.columns([3, 1])
+                    
+                    with col1:
+                        st.image(image, caption=f"Generated Poster {i+1}", use_column_width=True)
                         
-                        # Convert back to RGB for compatibility
-                        image = combined_image.convert('RGB')
+                        # Provide download option for the generated image
+                        buf = io.BytesIO()
+                        image.save(buf, format="PNG")
+                        byte_im = buf.getvalue()
+                        st.download_button(
+                            label=f"Download generated poster {i+1}",
+                            data=byte_im,
+                            file_name=f"generated_poster_{i+1}.png",
+                            mime="image/png"
+                        )
                     
-                    generated_images.append(image)
+                    with col2:
+                        st.subheader("Color Options")
+                        colors = ["#FF0000", "#00FF00", "#0000FF", "#FFFF00", "#FF00FF", "#00FFFF", "#FFFFFF"]
+                        color_names = ["Red", "Green", "Blue", "Yellow", "Magenta", "Cyan", "White"]
+                        
+                        for color, name in zip(colors, color_names):
+                            if st.button(name, key=f"{name}_{i}"):
+                                # Create a new image with the selected color
+                                colored_img = Image.new('RGB', image.size, color)
+                                # Blend the original image with the colored image
+                                blended_img = Image.blend(image, colored_img, 0.5)
+                                # Display the blended image
+                                st.image(blended_img, caption=f"Generated Poster {i+1} with {name} tint", use_column_width=True)
+                                
+                                # Provide download option for the blended image
+                                buf = io.BytesIO()
+                                blended_img.save(buf, format="PNG")
+                                byte_im = buf.getvalue()
+                                st.download_button(
+                                    label=f"Download {name} tinted poster {i+1}",
+                                    data=byte_im,
+                                    file_name=f"generated_poster_{i+1}_{name}_tint.png",
+                                    mime="image/png"
+                                )
                 else:
-                    st.error("Failed to generate image")
-
-        # Display generated images
-        for i, img in enumerate(generated_images):
-            st.image(img, caption=f"Generated Poster {i+1}", use_column_width=True)
-            
-            # Provide download option for the generated image
-            buf = io.BytesIO()
-            img.save(buf, format="PNG")
-            byte_im = buf.getvalue()
-            st.download_button(
-                label=f"Download generated poster {i+1}",
-                data=byte_im,
-                file_name=f"generated_poster_{i+1}.png",
-                mime="image/png"
-            )
+                    st.error(f"Failed to generate image {i+1}")
 
 # Content from image_to_image.py
 def encode_image(image):