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,7 @@ 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"])
     
@@ -187,6 +187,7 @@ 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 image {i+1}..."):
                 logger.info(f"Generating image {i+1} with prompt: {full_prompt}")
@@ -244,6 +245,8 @@ 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)
                     
@@ -528,23 +531,60 @@ 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("logo.jpg")  # 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 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)
+        st.title(title)
+        st.write(description)
+    
+    st.sidebar.markdown("### Poster Generation")
+    if st.sidebar.button("Go to 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("Go to 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("Go to 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("Go to Advertisement Generator"):
+        st.session_state.page = "advertisement_generator"
+    st.sidebar.markdown("Create compelling advertisements with AI assistance.")
 
-    if page == "Text to Image Generation":
+    if st.session_state.page == "text_to_image":
+        display_logo_and_description("Marketing Reimagined With AI", "Transform your ideas into stunning visuals.")
         text_to_image_generation()
-    elif page == "Image Editing":
+    elif st.session_state.page == "image_editing":
+        display_logo_and_description("Marketing Reimagined With AI", "Enhance and modify your images with powerful tools.")
         image_editing()
-    elif page == "Poster Generation":
+    elif st.session_state.page == "poster_generation":
+        display_logo_and_description("Marketing Reimagined With AI", "Create eye-catching posters for various platforms.")
         generate_poster()
-    elif page == "Advertisement Generator":
+    elif st.session_state.page == "advertisement_generator":
+        display_logo_and_description("Marketing Reimagined With AI", "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 +634,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")