Update app.py

app.py

@@ -123,20 +123,26 @@ def increase_image_quality(image, scale_factor):
 def generate_poster():
     #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
+    col1, col2 = st.columns(2)  # Equal width columns
+    with col1:
+        if st.button("✨Enhance My Prompt", key="ad_generator_button", use_container_width=True):
+            st.session_state.show_ad_generator = True
+    with col2:
+        generate_button = st.button("Generate Graphics", use_container_width=True)
+
+    st.markdown("<div style='height: 30px;'></div>", unsafe_allow_html=True)
 
     if st.session_state.get('show_ad_generator', False):
         with st.expander("Advertisement Generator :arrow_right:", expanded=True):
             advertisement_generator()
 
-    st.header("Graphic Type")
+    st.markdown("<div style='height: 30px;'></div>", unsafe_allow_html=True)
+
     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":
@@ -151,12 +157,12 @@ def generate_poster():
             "Corporate/Professional", "Retro/Vintage", "Modern/Contemporary", "Illustrative/Artistic"
         ])
 
+    st.markdown("<div style='height: 30px;'></div>", unsafe_allow_html=True)
+
     # Extract width and height from the selected dimensions
     width, height = map(int, dimensions.split('x'))
-
-    st.header("Content")
     
-    with st.expander("Content Details", expanded=False):
+    with st.expander("Add Content : Header, Sub-header and Descriptions", expanded=False):
         header = st.text_input("Enter Header for Advertisement:")
         sub_header = st.text_input("Enter Sub-header for Advertisement:")    
         # Allow multiple user prompts
@@ -166,9 +172,9 @@ def generate_poster():
             user_prompt = st.text_area(f"Enter Descriptions to display in the image (Descriptions {i+1}):")
             user_prompts.append(user_prompt)
 
-    st.header("Branding")
+    st.markdown("<div style='height: 30px;'></div>", unsafe_allow_html=True)
     
-    with st.expander("Branding Details", expanded=False):
+    with st.expander("Add Branding : Logo and Color", 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)
@@ -181,102 +187,104 @@ def generate_poster():
             "Left Middle", "Right Middle",
             "Bottom Left", "Bottom Middle", "Bottom Right"
         ])
-    
-    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])
+    st.markdown("<div style='height: 30px;'></div>", unsafe_allow_html=True)
 
-    # 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 number of images to generate (hidden from user)
+    #num_images = st.number_input("Number of images to generate", min_value=4, max_value=100, value=1)
+
+    num_images = 4
+
+    if generate_button:
+        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}})
+                
+                if image_bytes:
+                    image = Image.open(io.BytesIO(image_bytes))
                     
-                    if image_bytes:
-                        image = Image.open(io.BytesIO(image_bytes))
+                    # 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)
                         
-                        # 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)
+                        padding = int(image.width * 0.02)  # Fixed 2% 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))
-                            
-                            # 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')
+                        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))
                         
-                        generated_images.append(image)
+                        # Convert logo to RGBA if it's not already
+                        if logo_image.mode != 'RGBA':
+                            logo_image = logo_image.convert('RGBA')
                         
-                        # Display generated image
-                        st.image(image, caption=f"Generated Poster {i+1}", use_column_width=True)
+                        combined_image.paste(logo_image, position, logo_image)
                         
-                        # 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}")
+                        # Convert back to RGB for compatibility
+                        image = combined_image.convert('RGB')
+                    
+                    generated_images.append(image)
+                    
+                    # 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 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}")
+
+        st.markdown("<div style='height: 15px;'></div>", unsafe_allow_html=True)
 
 # Content from image_to_image.py
 def encode_image(image):
@@ -289,7 +297,7 @@ def generate_image_prompt(image):
     
     prompt_parts = [
         {"mime_type": "image/png", "data": base64.b64decode(encoded_image)},
-        "Analyze this image and generate a detailed prompt that could be used to recreate this image using an AI image generation model. Include key visual elements, style, composition, and any other relevant details but do not include any kind of text elements."
+        "Analyze this image and generate a detailed prompt that could be used to recreate this image using an AI image generation model. Include key visual elements, style, composition,text element, and any other relevant details."
     ]
 
     response = model.generate_content(prompt_parts)
@@ -547,11 +555,24 @@ def remove_background(image):
 
 # Main Streamlit App
 def main():    
-    # Add logo to the middle of the sidebar
+    # Add logo to the center 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)
+    st.sidebar.markdown(
+        """
+        <style>
+        .sidebar-logo {
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            padding: 1rem 0;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+    st.sidebar.markdown('<div class="sidebar-logo">', unsafe_allow_html=True)
+    st.sidebar.image(logo, width=150)
+    st.sidebar.markdown('</div>', unsafe_allow_html=True)
     
     # Initialize session state for page
     if 'page' not in st.session_state:
@@ -562,7 +583,19 @@ def main():
         st.title(title)
         st.write(description)
     
-    if st.sidebar.button("Poster Generation"):
+    # Create even-shaped buttons in the sidebar
+    button_style = """
+        <style>
+        div.stButton > button {
+            width: 100%;
+            height: 3em;
+            margin-bottom: 10px;
+        }
+        </style>
+    """
+    st.sidebar.markdown(button_style, unsafe_allow_html=True)
+    
+    if st.sidebar.button("Designer"):
         st.session_state.page = "poster_generation"
     
     if st.sidebar.button("Image to Image Generation"):
@@ -575,16 +608,16 @@ def main():
         st.session_state.page = "advertisement_generator"
 
     if st.session_state.page == "text_to_image":
-        display_title_and_description("Graphic Generator", "Transform your ideas into stunning visuals.")
+        display_title_and_description("Mark8 Designer", "Transform your ideas into stunning visuals.")
         text_to_image_generation()
     elif st.session_state.page == "image_editing":
-        display_title_and_description("Graphic Generator", "Enhance and modify your images with powerful tools.")
+        display_title_and_description("Mark8 Designer", "Enhance and modify your images with powerful tools.")
         image_editing()
     elif st.session_state.page == "poster_generation":
-        display_title_and_description("Graphic Generator", "Create eye-catching posters for various platforms.")
+        display_title_and_description("Mark8 Designer", "Create eye-catching posters for various platforms.")
         generate_poster()
     elif st.session_state.page == "advertisement_generator":
-        display_title_and_description("Graphic Generator", "Create compelling advertisements with AI assistance.")
+        display_title_and_description("Mark8 Designer", "Create compelling advertisements with AI assistance.")
         advertisement_generator()
 
 def text_to_image_generation():
@@ -604,23 +637,13 @@ def text_to_image_generation():
                 st.text_area(label="", value=generated_prompt.strip(), height=200, key="generated_prompt", disabled=True)
                 st.session_state['saved_prompt'] = generated_prompt.strip()
 
-        # Add multiple text elements
-        st.subheader("Add Text Elements")
-        num_text_elements = st.number_input("Number of text elements to add:", min_value=1, max_value=5, value=1)
-        text_elements = []
-        for i in range(num_text_elements):
-            text_element = st.text_input(f"Enter text {i+1} to add to the image:")
-            if text_element:
-                text_elements.append(f'text on image: "{text_element}"')
-        text_element_prompt = ", ".join(text_elements)
-
         # User input prompt
         st.subheader("Additional Prompt")
         user_prompt = st.text_input("Enter additional prompt details:")
 
         # Combine prompts
         saved_prompt = st.session_state.get('saved_prompt', '')
-        final_prompt = f"{saved_prompt}, {text_element_prompt}, {user_prompt}".strip()
+        final_prompt = f"{saved_prompt}, {user_prompt}".strip()
         
         st.subheader("Final Prompt")
         final_prompt_area = st.text_area("Final prompt for image generation:", value=final_prompt, height=150, key="final_prompt")