Update app.py

app.py

@@ -141,6 +141,12 @@ def generate_poster():
     ])
     
     description = st.text_area("Enter prompt for Advertisement:")
+    if st.button("Use Advertisement Generator", key="ad_generator_button"):
+        st.session_state.show_ad_generator = True
+
+    if st.session_state.get('show_ad_generator', False):
+        with st.expander("Advertisement Generator :arrow_right:", expanded=True):
+            advertisement_generator()
     header = st.text_input("Enter Header for Advertisement:")
     sub_header = st.text_input("Enter Sub-header for Advertisement:")    
     # Allow multiple user prompts
@@ -162,10 +168,12 @@ def generate_poster():
     num_images = st.number_input("Number of Images", min_value=4, max_value=8, value=4)
 
     if st.button("Generate Images"):
+        base_prompt = f"Create a minimalistic and realistic poster image with the following text as the header: '{header}'. Use '{sub_header}' as the sub-header. Use this designe style and go with thi {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. Make the design clean and elegant, focusing on simplicity and realism rather than vibrant colors."
+
         if post_type == "Other":
-            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."
+            full_prompt = f"{base_prompt} The image should be suitable for general use across various platforms."
         else:
-            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."
+            full_prompt = f"{base_prompt} This image is specifically for a {post_type.lower()}."
 
         # Add user prompts to full_prompt if provided
         for i, user_prompt in enumerate(user_prompts):
@@ -182,7 +190,7 @@ def generate_poster():
                 if image_bytes:
                     image = Image.open(io.BytesIO(image_bytes))
                     
-                    # Add logo if provided, or place it randomly if not specified
+                    # Add logo if provided
                     if logo:
                         logo_image = Image.open(logo)
                         logo_width = int(image.width * 0.15)  # 15% of the image width
@@ -192,88 +200,56 @@ def generate_poster():
                         padding = int(image.width * 0.02)  # Fixed 2% padding
                         
                         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)
-                        elif logo_position == "Top Right":
-                            position = (image.width - logo_width - padding, padding)
-                        elif logo_position == "Left Middle":
-                            position = (padding, (image.height - logo_height) // 2)
-                        elif logo_position == "Right Middle":
-                            position = (image.width - logo_width - padding, (image.height - logo_height) // 2)
-                        elif logo_position == "Bottom Left":
-                            position = (padding, image.height - logo_height - padding)
-                        elif logo_position == "Bottom Middle":
-                            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))
+                            # Randomly choose a corner for logo placement
+                            corner = random.choice(["Top Left", "Top Right", "Bottom Left", "Bottom Right"])
+                            if corner == "Top Left":
+                                position = (padding, padding)
+                            elif corner == "Top Right":
+                                position = (image.width - logo_width - padding, padding)
+                            elif corner == "Bottom Left":
+                                position = (padding, image.height - logo_height - padding)
+                            else:  # Bottom Right
+                                position = (image.width - logo_width - padding, image.height - logo_height - padding)
+                        else:
+                            if logo_position == "Top Left":
+                                position = (padding, padding)
+                            elif logo_position == "Top Middle":
+                                position = ((image.width - logo_width) // 2, padding)
+                            elif logo_position == "Top Right":
+                                position = (image.width - logo_width - padding, padding)
+                            elif logo_position == "Bottom Left":
+                                position = (padding, image.height - logo_height - padding)
+                            elif logo_position == "Bottom Middle":
+                                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)
                     
-                    if logo:
+                        # 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))
+                        
                         # 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')
+                        
+                        # Convert back to RGB for compatibility
+                        image = combined_image.convert('RGB')
                     
                     # Display generated image
-                    col1, col2 = st.columns([3, 1])
+                    st.image(image, caption=f"Generated Poster {i+1}", use_column_width=True)
                     
-                    with col1:
-                        st.image(image, caption=f"Generated Poster {i+1}", use_column_width=True)
-                        
-                        # 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"
-                        )
-                    
-                    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"
-                                )
+                    # 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"
+                    )
                 else:
                     st.error(f"Failed to generate image {i+1}")