Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-import requests
+import re
+import randomquests
 import os
 import json
 import base64
@@ -56,8 +57,8 @@ st.markdown("""
 # Title and description
 st.markdown("""
 <div class="title-area">
-    <h1>Marketing Graphic Generator 🎨</h1>
-    <p>Create professional marketing and advertising graphics using AI. Customize text, colors, and style for your business.</p>
+    <h1>Lead Magnet & Marketing Graphic Generator 🎨</h1>
+    <p>Create stunning lead magnets and marketing graphics in seconds using AI. Just describe what you want, and we'll do the rest!</p>
 </div>
 """, unsafe_allow_html=True)
 
@@ -78,26 +79,38 @@ def generate_image_from_prompt(prompt, api_key, model="gpt-4o"):
         "Authorization": f"Bearer {api_key}"
     }
     
-    # Create a system message directing the model to create high-quality marketing graphics
-    system_message = """You are an expert graphic designer specialized in creating marketing and advertising materials. 
-    Create high-quality, professional marketing graphics based on the user's specifications.
-    Pay careful attention to font choices, color schemes, layout, and branding elements.
-    Ensure text is readable and properly positioned.
-    Create images with proper aspect ratios and compositions that would work well for marketing purposes."""
+    # Create a system message directing the model to create high-quality marketing graphics,
+    # with specific emphasis on lead magnets if not specified
+    system_message = """You are an expert graphic designer specialized in creating professional marketing and lead magnet materials.
+    You excel at creating attention-grabbing graphics that convert viewers into leads and customers.
+    
+    When creating lead magnets or marketing graphics:
+    1. Make text highly readable with strong contrast against backgrounds
+    2. Use a clear visual hierarchy to guide the viewer's eye
+    3. Include compelling visual elements that reinforce the value proposition
+    4. Ensure the design conveys professionalism and trustworthiness
+    5. Position call-to-action elements prominently
+    
+    If the user hasn't specified certain design elements, use your expertise to:
+    - Choose appropriate color schemes that evoke the right emotional response for the topic
+    - Select fonts that match the industry and target audience
+    - Add visual elements that enhance the message without cluttering
+    - Create a composition that works well for digital marketing purposes"""
     
     # Enhance the user's prompt with specific guidance for marketing images
-    enhanced_prompt = f"""Create a professional marketing or advertising graphic with the following specifications:
+    enhanced_prompt = f"""Create a professional lead magnet or marketing graphic that will drive conversions based on this description:
 
 {prompt}
 
-Make sure the image has:
-1. High visual appeal suitable for marketing purposes
-2. Clear, readable text with proper spacing and alignment
-3. Professional design elements and composition
-4. Appropriate color harmony and contrast
-5. Balanced layout with focal points that draw attention
+Even if details are minimal, create a complete, professional design making appropriate assumptions.
+Make this graphic immediately usable as a marketing asset with:
+1. Attention-grabbing visual elements
+2. Clear, compelling text hierarchy
+3. Professional design elements that inspire trust
+4. Perfect readability with appropriate contrast and spacing
+5. A composition that drives the viewer toward the call-to-action
 
-The image should look like it was created by a professional graphic designer and be ready to use for marketing campaigns.
+The image should look like it was created by an expert marketing designer specifically to generate leads and conversions.
 """
     
     data = {
@@ -184,20 +197,42 @@ with st.sidebar:
     if not api_key:
         st.warning("⚠️ Please enter your OpenAI API key to use this tool")
     
-    st.markdown('<div class="custom-subheader">Advanced Settings</div>', unsafe_allow_html=True)
+    st.markdown('<div class="custom-subheader">Image Settings</div>', unsafe_allow_html=True)
+    
+    # Simplified image settings
+    aspect_ratio = st.selectbox(
+        "Aspect Ratio",
+        ["Square (1:1)", "Landscape (16:9)", "Portrait (9:16)", "Social Media (optimized)"],
+        index=0,
+        help="Choose the shape of your image. Social Media option will optimize based on content type."
+    )
     
-    # Image quality and size settings
     image_quality = st.select_slider(
         "Image Quality",
         options=["Standard", "High", "Maximum"],
-        value="High"
+        value="High",
+        help="Higher quality takes slightly longer to generate"
     )
     
-    aspect_ratio = st.selectbox(
-        "Aspect Ratio",
-        ["Square (1:1)", "Landscape (16:9)", "Portrait (9:16)", "Facebook (1200x628)", "Instagram (1080x1080)", "Twitter (1200x675)"],
+    # Example prompts for inspiration
+    st.markdown('<div class="custom-subheader">Need Inspiration?</div>', unsafe_allow_html=True)
+    
+    example_prompts = [
+        "Create a lead magnet for my free '7-Day SEO Challenge' with a modern professional look that appeals to small business owners",
+        "Design a graphic for my 'Ultimate Email Marketing Checklist' ebook with blue and orange colors",
+        "Make an eye-catching graphic for my 'Social Media Content Calendar' template with a clean, minimalist style",
+        "Design a Facebook ad for my free webinar '5 Steps to Passive Income' with a professional finance theme",
+        "Create a graphic for my 'Healthy Meal Prep Guide' PDF with fresh, vibrant colors and food imagery"
+    ]
+    
+    selected_example = st.selectbox(
+        "Try an example prompt",
+        ["Select an example..."] + example_prompts,
         index=0
     )
+    
+    if selected_example != "Select an example..." and st.button("Use This Example"):
+        st.session_state["example_prompt"] = selected_example
 
     # History
     if st.session_state["prompt_history"]:
@@ -212,102 +247,165 @@ col1, col2 = st.columns([1, 1])
 with col1:
     st.markdown('<div class="custom-subheader">Design Specifications</div>', unsafe_allow_html=True)
     
-    # Business Information
-    business_name = st.text_input("Business Name", placeholder="e.g., Sunset Cafe")
-    business_type = st.text_input("Business Type/Industry", placeholder="e.g., Coffee Shop, Real Estate, Fitness Studio")
-    location = st.text_input("Location (City/Region)", placeholder="e.g., San Francisco, CA")
+    # Quick Generate Option
+    st.markdown("##### Quick Generate")
     
-    # Marketing Content
-    st.markdown("##### Marketing Content")
+    # Use example prompt if selected
+    if "example_prompt" in st.session_state:
+        quick_description_default = st.session_state["example_prompt"]
+        # Clear it after using once
+        del st.session_state["example_prompt"]
+    else:
+        quick_description_default = ""
     
-    marketing_purpose = st.selectbox(
-        "Purpose of Graphic",
-        ["Advertisement", "Social Media Post", "Flyer", "Banner", "Logo", "Business Card", "Email Header", "Website Hero Image", "Promotional Offer"],
-        index=0
+    quick_description = st.text_area(
+        "Describe your lead magnet or marketing graphic", 
+        value=quick_description_default,
+        placeholder="e.g., Create a lead magnet graphic for my free SEO checklist ebook, with modern colors that appeal to small business owners",
+        height=100,
+        help="Just describe what you want in plain language - our AI will figure out the details!"
     )
     
-    headline = st.text_input("Headline/Main Text", placeholder="e.g., Grand Opening! 50% Off All Drinks")
-    subheading = st.text_input("Subheading/Secondary Text", placeholder="e.g., This weekend only - May 15-17")
-    call_to_action = st.text_input("Call to Action", placeholder="e.g., Visit us today! Call 555-123-4567")
-    
-    # Visual Style
-    st.markdown("##### Visual Style")
-    
-    color_scheme = st.text_input("Color Scheme", placeholder="e.g., Blue and gold, #FF5733 and #33FF57, Corporate colors")
-    font_style = st.text_input("Font Style", placeholder="e.g., Modern sans-serif, Elegant serif, Playful handwritten")
-    image_style = st.text_input("Image Style/Mood", placeholder="e.g., Minimalist, Vibrant, Professional, Vintage")
-    
-    # Additional Elements
-    st.markdown("##### Additional Elements")
-    
-    visual_elements = st.text_area(
-        "Specific Visual Elements to Include", 
-        placeholder="e.g., Coffee cup with steam, Mountain backdrop, Product showcase, People enjoying the service",
-        height=100
-    )
+    show_advanced = st.checkbox("Show advanced options", value=False)
     
-    special_instructions = st.text_area(
-        "Special Instructions", 
-        placeholder="Any additional details or specific requirements for the design",
-        height=100
-    )
+    if show_advanced:
+        # Business Information (all optional)
+        st.markdown("##### Business Information (Optional)")
+        col_bus1, col_bus2 = st.columns(2)
+        with col_bus1:
+            business_name = st.text_input("Business Name", placeholder="e.g., Sunset Cafe")
+        with col_bus2:
+            business_type = st.text_input("Business Type/Industry", placeholder="e.g., Coffee Shop, Real Estate")
+        
+        # Marketing Content (simplified, only headline required)
+        st.markdown("##### Marketing Content (Optional)")
+        headline = st.text_input("Headline/Main Text", placeholder="e.g., Download Our Free Guide")
+        
+        col_m1, col_m2 = st.columns(2)
+        with col_m1:
+            marketing_purpose = st.selectbox(
+                "Purpose",
+                ["Lead Magnet", "Social Media Post", "Advertisement", "Promotional Offer", "Email Header", "Other"],
+                index=0
+            )
+        with col_m2:
+            call_to_action = st.text_input("Call to Action", placeholder="e.g., Sign up today!")
+        
+        # Visual Style (all optional)
+        st.markdown("##### Visual Style (Optional)")
+        col_v1, col_v2 = st.columns(2)
+        with col_v1:
+            color_scheme = st.text_input("Color Scheme", placeholder="e.g., Blue and gold, #FF5733")
+        with col_v2:
+            image_style = st.selectbox(
+                "Style",
+                ["Modern", "Professional", "Casual", "Elegant", "Bold", "Minimalist", "Playful", "Custom"],
+                index=0
+            )
+        
+        if image_style == "Custom":
+            custom_style = st.text_input("Describe your custom style", placeholder="e.g., Vintage watercolor with pastel colors")
+        else:
+            custom_style = ""
+        
+        # Additional Elements (optional)
+        special_instructions = st.text_area(
+            "Any other details or requirements", 
+            placeholder="e.g., Include our brand mascot, make sure text is very readable",
+            height=75
+        )
+    else:
+        # Set default values for advanced options
+        business_name = ""
+        business_type = ""
+        headline = ""
+        marketing_purpose = "Lead Magnet"
+        call_to_action = ""
+        color_scheme = ""
+        image_style = "Modern"
+        custom_style = ""
+        special_instructions = ""
     
     # Build the prompt
     if "current_prompt" in st.session_state:
         complete_prompt = st.session_state["current_prompt"]
         del st.session_state["current_prompt"]
     else:
-        prompt_parts = []
-        
-        # Add business information
-        if business_name:
-            prompt_parts.append(f"Business Name: {business_name}")
-        if business_type:
-            prompt_parts.append(f"Business Type: {business_type}")
-        if location:
-            prompt_parts.append(f"Location: {location}")
-        
-        # Add marketing content
-        prompt_parts.append(f"Create a {marketing_purpose.lower()}")
-        if headline:
-            prompt_parts.append(f"Headline: '{headline}'")
-        if subheading:
-            prompt_parts.append(f"Subheading: '{subheading}'")
-        if call_to_action:
-            prompt_parts.append(f"Call to Action: '{call_to_action}'")
-        
-        # Add visual style
-        if color_scheme:
-            prompt_parts.append(f"Color Scheme: {color_scheme}")
-        if font_style:
-            prompt_parts.append(f"Font Style: {font_style}")
-        if image_style:
-            prompt_parts.append(f"Visual Style: {image_style}")
-        
-        # Add aspect ratio
-        if "Square" in aspect_ratio:
-            prompt_parts.append("Create a square (1:1) image")
-        elif "Landscape" in aspect_ratio:
-            prompt_parts.append("Create a landscape (16:9) format image")
-        elif "Portrait" in aspect_ratio:
-            prompt_parts.append("Create a portrait (9:16) format image")
-        elif "Facebook" in aspect_ratio:
-            prompt_parts.append("Create an image in Facebook post dimensions (1200x628)")
-        elif "Instagram" in aspect_ratio:
-            prompt_parts.append("Create an image in Instagram post dimensions (1080x1080)")
-        elif "Twitter" in aspect_ratio:
-            prompt_parts.append("Create an image in Twitter post dimensions (1200x675)")
-        
-        # Add image quality
-        prompt_parts.append(f"Generate a {image_quality.lower()} quality image")
-        
-        # Add additional elements
-        if visual_elements:
-            prompt_parts.append(f"Include these visual elements: {visual_elements}")
-        if special_instructions:
-            prompt_parts.append(f"Special instructions: {special_instructions}")
-        
-        prompt_parts.append("Make text clear, readable, and properly positioned. Ensure professional design suitable for marketing purposes.")
+        # Check if using quick generate or advanced
+        if quick_description and not show_advanced:
+            # Simple quick generation
+            prompt_parts = [
+                f"Create a professional marketing graphic for a lead magnet or marketing material with this description: {quick_description}",
+                f"Generate a {image_quality.lower()} quality image",
+            ]
+            
+            # Add aspect ratio
+            if "Square" in aspect_ratio:
+                prompt_parts.append("Create a square (1:1) image")
+            elif "Landscape" in aspect_ratio:
+                prompt_parts.append("Create a landscape (16:9) format image")
+            elif "Portrait" in aspect_ratio:
+                prompt_parts.append("Create a portrait (9:16) format image")
+            elif "Facebook" in aspect_ratio:
+                prompt_parts.append("Use Facebook post dimensions (1200x628)")
+            elif "Instagram" in aspect_ratio:
+                prompt_parts.append("Use Instagram post dimensions (1080x1080)")
+            elif "Twitter" in aspect_ratio:
+                prompt_parts.append("Use Twitter post dimensions (1200x675)")
+                
+            prompt_parts.append("Make text clear, readable, and properly positioned. Ensure professional design suitable for marketing purposes.")
+            
+        else:
+            # Advanced generation with all fields
+            prompt_parts = []
+            
+            # Add quick description if it exists
+            if quick_description:
+                prompt_parts.append(f"Create a marketing graphic with this general description: {quick_description}")
+            
+            # Add business information
+            if business_name:
+                prompt_parts.append(f"Business Name: {business_name}")
+            if business_type:
+                prompt_parts.append(f"Business Type: {business_type}")
+            
+            # Add marketing content
+            prompt_parts.append(f"Create a {marketing_purpose.lower()}")
+            if headline:
+                prompt_parts.append(f"Headline: '{headline}'")
+            if call_to_action:
+                prompt_parts.append(f"Call to Action: '{call_to_action}'")
+            
+            # Add visual style
+            if color_scheme:
+                prompt_parts.append(f"Color Scheme: {color_scheme}")
+            if image_style != "Custom":
+                prompt_parts.append(f"Visual Style: {image_style}")
+            else:
+                prompt_parts.append(f"Visual Style: {custom_style}")
+            
+            # Add aspect ratio
+            if "Square" in aspect_ratio:
+                prompt_parts.append("Create a square (1:1) image")
+            elif "Landscape" in aspect_ratio:
+                prompt_parts.append("Create a landscape (16:9) format image")
+            elif "Portrait" in aspect_ratio:
+                prompt_parts.append("Create a portrait (9:16) format image")
+            elif "Facebook" in aspect_ratio:
+                prompt_parts.append("Use Facebook post dimensions (1200x628)")
+            elif "Instagram" in aspect_ratio:
+                prompt_parts.append("Use Instagram post dimensions (1080x1080)")
+            elif "Twitter" in aspect_ratio:
+                prompt_parts.append("Use Twitter post dimensions (1200x675)")
+            
+            # Add image quality
+            prompt_parts.append(f"Generate a {image_quality.lower()} quality image")
+            
+            # Add additional elements
+            if special_instructions:
+                prompt_parts.append(f"Special instructions: {special_instructions}")
+            
+            prompt_parts.append("Make text clear, readable, and properly positioned. Ensure professional design suitable for marketing purposes.")
         
         complete_prompt = ". ".join(prompt_parts)