Update app.py

app.py

@@ -122,31 +122,6 @@ def increase_image_quality(image, scale_factor):
 
 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.session_state.show_ad_generator = True
@@ -154,25 +129,57 @@ def generate_poster():
     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)
-    
-    logo = st.file_uploader("Upload Logo (optional)", type=['png', 'jpg', 'jpeg'])
+
+    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")
     
-    # 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("Content 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)
+        
+        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)
+        
+        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)
+    num_images = st.number_input("Number of Images", min_value=4, max_value=100, 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."
@@ -533,7 +540,7 @@ def remove_background(image):
 # Main Streamlit App
 def main():    
     # Add logo to the middle of the sidebar
-    logo = Image.open("logo.jpg")  # Replace with your logo path
+    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)
@@ -550,37 +557,37 @@ def main():
         st.title(title)
         st.write(description)
     
-    st.sidebar.markdown("### Poster Generation")
-    if st.sidebar.button("Go to Poster Generation"):
+    #st.sidebar.markdown("### Poster Generation")
+    if st.sidebar.button("Poster Generation"):
         st.session_state.page = "poster_generation"
-    st.sidebar.markdown("Create eye-catching posters for various platforms.")
+    #st.sidebar.markdown("Create eye-catching posters for various platforms.")
     
-    st.sidebar.markdown("### Image to Image Generation")
-    if st.sidebar.button("Go to Image to Image Generation"):
+   # st.sidebar.markdown("### Image to Image Generation")
+    if st.sidebar.button("Image to Image Generation"):
         st.session_state.page = "text_to_image"
-    st.sidebar.markdown("Transform your ideas into stunning visuals.")
+    #st.sidebar.markdown("Transform your ideas into stunning visuals.")
     
-    st.sidebar.markdown("### Image Editing")
-    if st.sidebar.button("Go to Image Editing"):
+    #st.sidebar.markdown("### Image Editing")
+    if st.sidebar.button("Image Editing"):
         st.session_state.page = "image_editing"
-    st.sidebar.markdown("Enhance and modify your images with powerful tools.")
+    #st.sidebar.markdown("Enhance and modify your images with powerful tools.")
     
-    st.sidebar.markdown("### Advertisement Generator")
-    if st.sidebar.button("Go to Advertisement Generator"):
+    #st.sidebar.markdown("### Advertisement Generator")
+    if st.sidebar.button("Advertisement Generator"):
         st.session_state.page = "advertisement_generator"
-    st.sidebar.markdown("Create compelling advertisements with AI assistance.")
+    #st.sidebar.markdown("Create compelling advertisements with AI assistance.")
 
     if st.session_state.page == "text_to_image":
-        display_logo_and_description("Marketing Reimagined With AI", "Transform your ideas into stunning visuals.")
+        display_logo_and_description("Graphic Generator", "Transform your ideas into stunning visuals.")
         text_to_image_generation()
     elif st.session_state.page == "image_editing":
-        display_logo_and_description("Marketing Reimagined With AI", "Enhance and modify your images with powerful tools.")
+        display_logo_and_description("Graphic Generator", "Enhance and modify your images with powerful tools.")
         image_editing()
     elif st.session_state.page == "poster_generation":
-        display_logo_and_description("Marketing Reimagined With AI", "Create eye-catching posters for various platforms.")
+        display_logo_and_description("Graphic Generator", "Create eye-catching posters for various platforms.")
         generate_poster()
     elif st.session_state.page == "advertisement_generator":
-        display_logo_and_description("Marketing Reimagined With AI", "Create compelling advertisements with AI assistance.")
+        display_logo_and_description("Graphic Generator", "Create compelling advertisements with AI assistance.")
         advertisement_generator()
 
 def text_to_image_generation():