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,7 +121,32 @@ def increase_image_quality(image, scale_factor):
     return image.resize(new_size, Image.LANCZOS)
 
 def generate_poster():
-    #st.header("Generate Social Media Post")
+    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.session_state.show_ad_generator = True
@@ -129,60 +154,25 @@ def generate_poster():
     if st.session_state.get('show_ad_generator', False):
         with st.expander("Advertisement Generator :arrow_right:", expanded=True):
             advertisement_generator()
-
-    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")
+    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)
     
-    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")
+    logo = st.file_uploader("Upload Logo (optional)", type=['png', 'jpg', 'jpeg'])
     
-    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"
-        ])
+    # 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=100, value=4)
+    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."
@@ -197,10 +187,9 @@ def generate_poster():
             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}")
+            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}})
@@ -255,8 +244,6 @@ def generate_poster():
                         # 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)
                     
@@ -265,13 +252,13 @@ def generate_poster():
                     image.save(buf, format="PNG")
                     byte_im = buf.getvalue()
                     st.download_button(
-                        label=f"Download generated Graphic {i+1}",
+                        label=f"Download generated poster {i+1}",
                         data=byte_im,
-                        file_name=f"generated_Graphic_{i+1}.png",
+                        file_name=f"generated_poster_{i+1}.png",
                         mime="image/png"
                     )
                 else:
-                    st.error(f"Failed to generate Graphic {i+1}")
+                    st.error(f"Failed to generate image {i+1}")
 
 # Content from image_to_image.py
 def encode_image(image):
@@ -541,49 +528,23 @@ def remove_background(image):
     return Image.fromarray(result)
 
 # Main Streamlit App
-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"
+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"])
 
-    if st.session_state.page == "text_to_image":
-        display_title_and_description("Graphic Generator", "Transform your ideas into stunning visuals.")
+    if page == "Text to Image Generation":
         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.")
+    elif page == "Image Editing":
         image_editing()
-    elif st.session_state.page == "poster_generation":
-        display_title_and_description("Graphic Generator", "Create eye-catching posters for various platforms.")
+    elif page == "Poster Generation":
         generate_poster()
-    elif st.session_state.page == "advertisement_generator":
-        display_title_and_description("Graphic Generator", "Create compelling advertisements with AI assistance.")
+    elif page == "Advertisement Generator":
         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")
@@ -633,7 +594,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")