Update app.py

app.py

@@ -157,10 +157,6 @@ def generate_poster():
     st.header("Content")
     
     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
@@ -169,6 +165,13 @@ def generate_poster():
         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")
+    
+    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'])
         
@@ -196,8 +199,8 @@ def generate_poster():
 
         generated_images = []
         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}")
+            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}})
@@ -262,13 +265,13 @@ def generate_poster():
                     image.save(buf, format="PNG")
                     byte_im = buf.getvalue()
                     st.download_button(
-                        label=f"Download generated poster {i+1}",
+                        label=f"Download generated Graphic {i+1}",
                         data=byte_im,
-                        file_name=f"generated_poster_{i+1}.png",
+                        file_name=f"generated_Graphic_{i+1}.png",
                         mime="image/png"
                     )
                 else:
-                    st.error(f"Failed to generate image {i+1}")
+                    st.error(f"Failed to generate Graphic {i+1}")
 
 # Content from image_to_image.py
 def encode_image(image):
@@ -549,45 +552,34 @@ def main():
     if 'page' not in st.session_state:
         st.session_state.page = "poster_generation"
     
-    # Function to display logo and description
-    def display_logo_and_description(title, description):
-        col1, col2, col3 = st.columns([1, 2, 1])
-        with col2:
-            st.image(logo, width=200)
+    # Function to display title and description
+    def display_title_and_description(title, description):
         st.title(title)
         st.write(description)
     
-    #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("### 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("### 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("### Advertisement Generator")
     if st.sidebar.button("Advertisement Generator"):
         st.session_state.page = "advertisement_generator"
-    #st.sidebar.markdown("Create compelling advertisements with AI assistance.")
 
     if st.session_state.page == "text_to_image":
-        display_logo_and_description("Graphic Generator", "Transform your ideas into stunning visuals.")
+        display_title_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("Graphic Generator", "Enhance and modify your images with powerful tools.")
+        display_title_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("Graphic Generator", "Create eye-catching posters for various platforms.")
+        display_title_and_description("Graphic Generator", "Create eye-catching posters for various platforms.")
         generate_poster()
     elif st.session_state.page == "advertisement_generator":
-        display_logo_and_description("Graphic Generator", "Create compelling advertisements with AI assistance.")
+        display_title_and_description("Graphic Generator", "Create compelling advertisements with AI assistance.")
         advertisement_generator()
 
 def text_to_image_generation():