Update app.py

app.py

@@ -83,7 +83,7 @@ def generate_advertisement_prompt(description):
     return response.content
 
 def advertisement_generator():
-    st.title("Advertisement Post Generator")
+    #st.title("Advertisement Post Generator")
 
     post_description = st.text_input("Enter a brief description for your advertisement post:")
 
@@ -121,144 +121,162 @@ def increase_image_quality(image, scale_factor):
     return image.resize(new_size, Image.LANCZOS)
 
 def generate_poster():
-    st.header("Generate Social Media Post")
-    post_type = st.selectbox("Select Post Type", ["Instagram advertisement post", "Facebook advertisement post", "Twitter advertisement post", "Other"])
-    aspect_ratio = st.selectbox("Select Image Aspect Ratio", ["1:1", "16:9", "4:5", "9:16"])
-    
-    # Add dimension selection based on aspect ratio
-    if aspect_ratio == "1:1":
-        dimensions = st.selectbox("Select Image Dimensions", ["1080x1080", "1200x1200", "1500x1500"])
-    elif aspect_ratio == "16:9":
-        dimensions = st.selectbox("Select Image Dimensions", ["1920x1080", "1280x720", "2560x1440"])
-    elif aspect_ratio == "4:5":
-        dimensions = st.selectbox("Select Image Dimensions", ["1080x1350", "1200x1500", "1600x2000"])
-    elif aspect_ratio == "9:16":
-        dimensions = st.selectbox("Select Image Dimensions", ["1080x1920", "1350x2400", "1600x2844"])
-    
-    # Extract width and height from the selected dimensions
-    width, height = map(int, dimensions.split('x'))
-    
-    design_style = st.selectbox("Select Design Style", [
-        "Minimalistic", "Bold/Graphic", "Elegant", "Playful/Fun",
-        "Corporate/Professional", "Retro/Vintage", "Modern/Contemporary", "Illustrative/Artistic"
-    ])
-    
-    # Add color selection with predefined options
-    color_options = ["None", "Black", "White", "Red", "Blue", "Green", "Yellow", "Purple"]
-    selected_color = st.selectbox("Choose a dominant color for the image", color_options)
-    
-    description = st.text_area("Enter prompt for Advertisement:")
-    if st.button("Use Advertisement Generator", key="ad_generator_button"):
+    #st.header("Generate Social Media Post")
+    description = st.text_input("Enter prompt for Advertisement:")
+    if st.button("✨", 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
-    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)
+
+    st.header("Graphic Type")
+    col1, col2 = st.columns(2)
+    with col1:
+        post_type = st.selectbox("Select Post Type", ["Instagram advertisement post", "Facebook advertisement post", "Twitter advertisement post", "Other"])
+        aspect_ratio = st.selectbox("Select Image Aspect Ratio", ["1:1", "16:9", "4:5", "9:16"])
+    with col2:
+        # Add dimension selection based on aspect ratio
+        if aspect_ratio == "1:1":
+            dimensions = st.selectbox("Select Image Dimensions", ["1080x1080", "1200x1200", "1500x1500"])
+        elif aspect_ratio == "16:9":
+            dimensions = st.selectbox("Select Image Dimensions", ["1920x1080", "1280x720", "2560x1440"])
+        elif aspect_ratio == "4:5":
+            dimensions = st.selectbox("Select Image Dimensions", ["1080x1350", "1200x1500", "1600x2000"])
+        elif aspect_ratio == "9:16":
+            dimensions = st.selectbox("Select Image Dimensions", ["1080x1920", "1350x2400", "1600x2844"])
+        
+        design_style = st.selectbox("Select Design Style", [
+            "Minimalistic", "Bold/Graphic", "Elegant", "Playful/Fun",
+            "Corporate/Professional", "Retro/Vintage", "Modern/Contemporary", "Illustrative/Artistic"
+        ])
+
+    # Extract width and height from the selected dimensions
+    width, height = map(int, dimensions.split('x'))
+
+    st.header("Content")
     
-    logo = st.file_uploader("Upload Logo (optional)", type=['png', 'jpg', 'jpeg'])
+    with st.expander("Content Details", expanded=False):
+        header = st.text_input("Enter Header 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)
+
+    st.header("Branding")
     
-    # Add logo position selection
-    logo_position = st.selectbox("Select Logo Position", [
-        "None", "Top Left", "Top Middle", "Top Right",
-        "Left Middle", "Right Middle",
-        "Bottom Left", "Bottom Middle", "Bottom Right"
-    ])
+    with st.expander("Branding Details", expanded=False):
+        # Add color selection with predefined options
+        color_options = ["None", "Black", "White", "Red", "Blue", "Green", "Yellow", "Purple"]
+        selected_color = st.selectbox("Choose a dominant color for the image", color_options)
+        
+        logo = st.file_uploader("Upload Logo (optional)", type=['png', 'jpg', 'jpeg'])
+        
+        # Add logo position selection
+        logo_position = st.selectbox("Select Logo Position", [
+            "None", "Top Left", "Top Middle", "Top Right",
+            "Left Middle", "Right Middle",
+            "Bottom Left", "Bottom Middle", "Bottom Right"
+        ])
     
-    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 design style and go with this {design_style.lower()} style for the post. Include the following text as the description for the advertisement: {description}. The image should have an aspect ratio of {aspect_ratio} and dimensions of {width}x{height}. Use {selected_color.lower()} as the dominant color in the image. Ensure the image is in 4k high resolution. And do not include any extra text in the image only instruction text included.And try to fill the vacant part with releveng images. Make the design clean and elegant, focusing on simplicity and realism."
-
-        if post_type == "Other":
-            full_prompt = f"{base_prompt} The image should be suitable for general use across various platforms."
-        else:
-            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):
-            if user_prompt:
-                full_prompt += f" Include the following text in the image (Prompt {i+1}): '{user_prompt}'."
-
-        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}")
-                # Add a random seed to ensure different images
-                seed = random.randint(1, 1000000)
-                image_bytes = query({"inputs": full_prompt, "parameters": {"seed": seed, "width": width, "height": height}})
-                
-                if image_bytes:
-                    image = Image.open(io.BytesIO(image_bytes))
+    num_images = st.number_input("Number of Images", min_value=4, max_value=100, value=4)
+
+    # Create a column layout to center the button
+    col1, col2, col3 = st.columns([1, 2, 1])
+
+    # Place the button in the middle column and increase its width
+    with col2:
+        if st.button("Generate Graphics", use_container_width=True):
+            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 design style and go with this {design_style.lower()} style for the post. Include the following text as the description for the advertisement: {description}. The image should have an aspect ratio of {aspect_ratio} and dimensions of {width}x{height}. Use {selected_color.lower()} as the dominant color in the image. Ensure the image is in 4k high resolution. And do not include any extra text in the image only instruction text included.And try to fill the vacant part with releveng images. Make the design clean and elegant, focusing on simplicity and realism."
+
+            if post_type == "Other":
+                full_prompt = f"{base_prompt} The image should be suitable for general use across various platforms."
+            else:
+                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):
+                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 Graphic {i+1}..."):
+                    logger.info(f"Generating Graphic {i+1} with prompt: {full_prompt}")
+                    # Add a random seed to ensure different images
+                    seed = random.randint(1, 1000000)
+                    image_bytes = query({"inputs": full_prompt, "parameters": {"seed": seed, "width": width, "height": height}})
                     
-                    # Add logo if provided
-                    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)
+                    if image_bytes:
+                        image = Image.open(io.BytesIO(image_bytes))
                         
-                        padding = int(image.width * 0.02)  # Fixed 2% padding
+                        # Add logo if provided
+                        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 * 0.02)  # Fixed 2% padding
+                            
+                            if logo_position == "None":
+                                # 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_position == "None":
-                            # 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)
-                    
-                        # 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))
+                            
+                            # 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 logo to RGBA if it's not already
-                        if logo_image.mode != 'RGBA':
-                            logo_image = logo_image.convert('RGBA')
+                        generated_images.append(image)
                         
-                        combined_image.paste(logo_image, position, logo_image)
+                        # Display generated image
+                        st.image(image, caption=f"Generated Poster {i+1}", use_column_width=True)
                         
-                        # Convert back to RGB for compatibility
-                        image = combined_image.convert('RGB')
-                    
-                    # Display generated image
-                    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"
-                    )
-                else:
-                    st.error(f"Failed to generate image {i+1}")
+                        # 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 Graphic {i+1}",
+                            data=byte_im,
+                            file_name=f"generated_Graphic_{i+1}.png",
+                            mime="image/png"
+                        )
+                    else:
+                        st.error(f"Failed to generate Graphic {i+1}")
 
 # Content from image_to_image.py
 def encode_image(image):
@@ -528,23 +546,49 @@ def remove_background(image):
     return Image.fromarray(result)
 
 # Main Streamlit App
-def main():
-    st.title("Image Generation and Editing App")
-
-    # Sidebar for navigation
-    page = st.sidebar.selectbox("Choose a function", ["Text to Image Generation", "Image Editing", "Poster Generation", "Advertisement Generator"])
+def main():    
+    # Add logo to the middle of the sidebar
+    logo = Image.open("Mark8 AI.png")  # Replace with your logo path
+    col1, col2, col3 = st.sidebar.columns([1, 2, 1])
+    with col2:
+        st.image(logo, width=150)
+    
+    # Initialize session state for page
+    if 'page' not in st.session_state:
+        st.session_state.page = "poster_generation"
+    
+    # Function to display title and description
+    def display_title_and_description(title, description):
+        st.title(title)
+        st.write(description)
+    
+    if st.sidebar.button("Poster Generation"):
+        st.session_state.page = "poster_generation"
+    
+    if st.sidebar.button("Image to Image Generation"):
+        st.session_state.page = "text_to_image"
+    
+    if st.sidebar.button("Image Editing"):
+        st.session_state.page = "image_editing"
+    
+    if st.sidebar.button("Advertisement Generator"):
+        st.session_state.page = "advertisement_generator"
 
-    if page == "Text to Image Generation":
+    if st.session_state.page == "text_to_image":
+        display_title_and_description("Graphic Generator", "Transform your ideas into stunning visuals.")
         text_to_image_generation()
-    elif page == "Image Editing":
+    elif st.session_state.page == "image_editing":
+        display_title_and_description("Graphic Generator", "Enhance and modify your images with powerful tools.")
         image_editing()
-    elif page == "Poster Generation":
+    elif st.session_state.page == "poster_generation":
+        display_title_and_description("Graphic Generator", "Create eye-catching posters for various platforms.")
         generate_poster()
-    elif page == "Advertisement Generator":
+    elif st.session_state.page == "advertisement_generator":
+        display_title_and_description("Graphic Generator", "Create compelling advertisements with AI assistance.")
         advertisement_generator()
 
 def text_to_image_generation():
-    st.header("Text to Image Generation")
+   # st.header("Text to Image Generation")
 
     # Image to Image Generation
     st.subheader("Image to Image Generation")
@@ -594,7 +638,7 @@ def text_to_image_generation():
                             st.image(new_image, caption=f"Generated Image {i+1}", use_column_width=True)
 
 def image_editing():
-    st.header("Image Editing")
+    #st.header("Image Editing")
 
     # Background Removal
     st.subheader("Background Removal")