Update app.py

app.py

@@ -16,6 +16,7 @@ from fontTools.ttLib import TTFont
 from langchain_groq import ChatGroq
 import logging
 
+
 # Load environment variables
 from dotenv import load_dotenv
 load_dotenv()
@@ -56,22 +57,24 @@ def generate_advertisement_prompt(description):
     Description: "{description}"
 
     Generate a detailed image prompt for an AI image generation model, incorporating these elements:
+    1. Header: Give one header based on the description.
+    2. Sub Header: Give one sub header based on the description.
 
-    1. Subject: Describe the main subject or product in the advertisement, including its key features and visual characteristics.
+    3. Subject: Describe the main subject or product in the advertisement, including its key features and visual characteristics.
 
-    2. Composition: Explain how the subject should be positioned within the frame, any specific angles or perspectives to highlight its best features.
+    4. Composition: Explain how the subject should be positioned within the frame, any specific angles or perspectives to highlight its best features.
 
-    3. Background: Detail the setting or environment that complements the subject and reinforces the advertisement's message.
+    5. Background: Detail the setting or environment that complements the subject and reinforces the advertisement's message.
 
-    4. Text Elements: text elemeny should be adverticement purpose like based on the description generate adverticement text.
+    6. Text Elements: text elemeny should be adverticement purpose like based on the description generate adverticement text.
 
-    5. Style: Describe the overall visual style, color scheme, and mood that best represents the brand and appeals to the target audience.
+    7. Style: Describe the overall visual style, color scheme, and mood that best represents the brand and appeals to the target audience.
 
-    6. Additional Elements: List any supporting visual elements, such as logos, icons, or graphics that should be included to enhance the advertisement's impact.
+    8. Additional Elements: List any supporting visual elements, such as logos, icons, or graphics that should be included to enhance the advertisement's impact.
 
-    7. Target Audience: Briefly mention the intended audience to ensure the image resonates with them.
+    9. Target Audience: Briefly mention the intended audience to ensure the image resonates with them.
 
-    8. Think in the basis of adverticment designer and combined all 7 points and make a final prompt.
+    10. Think in the basis of adverticment designer and combined all 9 points and make a final prompt.
 
     Please provide a cohesive image prompt that incorporates all these elements into a striking, attention-grabbing advertisement poster, based on the given description. The prompt should be detailed enough to generate a compelling and effective advertisement image.
     """
@@ -140,6 +143,10 @@ def generate_poster():
         "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
@@ -168,7 +175,7 @@ def generate_poster():
     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 designe style and go with thi {design_style.lower()} style for the post. Include the following text as the description: {description}. The image should have an aspect ratio of {aspect_ratio} and dimensions of {width}x{height}. Ensure the image is in 4k high resolution. Make the design clean and elegant, focusing on simplicity and realism rather than vibrant colors."
+        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."