Create app.py

app.py

@@ -1,19 +1,20 @@
 import streamlit as st
-import re
-import randomquests
+import requests
 import os
 import json
 import base64
 from io import BytesIO
 from PIL import Image
 import time
+import re
+import random
 
 # Set page config
 st.set_page_config(
-    page_title="Marketing Graphic Generator",
+    page_title="Lead Magnet Generator",
     page_icon="🎨",
     layout="wide",
-    initial_sidebar_state="expanded"
+    initial_sidebar_state="collapsed"
 )
 
 # Custom CSS
@@ -22,6 +23,8 @@ st.markdown("""
     .main .block-container {
         padding-top: 2rem;
         padding-bottom: 2rem;
+        max-width: 1000px;
+        margin: 0 auto;
     }
     .stApp {
         background-color: #f8f9fa;
@@ -51,6 +54,18 @@ st.markdown("""
         color: #6c757d;
         margin-top: 2rem;
     }
+    .settings-section {
+        background-color: #f8f9fa;
+        padding: 15px;
+        border-radius: 5px;
+        margin-bottom: 20px;
+    }
+    .examples-section {
+        background-color: #e9f7fe;
+        padding: 15px;
+        border-radius: 5px;
+        margin-bottom: 20px;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -63,8 +78,6 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Initialize session state
-if "api_key" not in st.session_state:
-    st.session_state["api_key"] = ""
 if "generated_image" not in st.session_state:
     st.session_state["generated_image"] = None
 if "prompt_history" not in st.session_state:
@@ -72,8 +85,12 @@ if "prompt_history" not in st.session_state:
 if "image_history" not in st.session_state:
     st.session_state["image_history"] = []
 
+# IMPORTANT: In a real application, this would be securely stored in environment variables or secrets management
+# For demonstration purposes only, we're using a placeholder value
+OPENAI_API_KEY = st.secrets.get("OPENAI_API_KEY", "YOUR_API_KEY_STORED_IN_SECRETS")
+
 # Functions
-def generate_image_from_prompt(prompt, api_key, model="gpt-4o"):
+def generate_image_from_prompt(prompt, api_key=OPENAI_API_KEY, model="gpt-4o"):
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {api_key}"
@@ -181,74 +198,12 @@ def save_image(image_data):
     
     return filename
 
-# Sidebar - Configuration
-with st.sidebar:
-    st.markdown('<div class="custom-subheader">API Configuration</div>', unsafe_allow_html=True)
-    
-    api_key = st.text_input(
-        "Enter your OpenAI API Key", 
-        value=st.session_state["api_key"],
-        type="password",
-        help="Your API key is required to generate images. It will not be stored permanently."
-    )
-    st.session_state["api_key"] = api_key
-    
-    # Display a warning if no API key is provided
-    if not api_key:
-        st.warning("⚠️ Please enter your OpenAI API key to use this tool")
-    
-    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 = st.select_slider(
-        "Image Quality",
-        options=["Standard", "High", "Maximum"],
-        value="High",
-        help="Higher quality takes slightly longer to generate"
-    )
-    
-    # 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"]:
-        st.markdown('<div class="custom-subheader">Prompt History</div>', unsafe_allow_html=True)
-        for i, prompt in enumerate(st.session_state["prompt_history"][-5:]):
-            if st.button(f"Reuse: {prompt[:30]}...", key=f"history_{i}"):
-                st.session_state["current_prompt"] = prompt
-
-# Main content
+# Main content - All inputs in a single section
 col1, col2 = st.columns([1, 1])
 
 with col1:
-    st.markdown('<div class="custom-subheader">Design Specifications</div>', unsafe_allow_html=True)
-    
     # Quick Generate Option
-    st.markdown("##### Quick Generate")
+    st.markdown('<div class="custom-subheader">Create Your Graphic</div>', unsafe_allow_html=True)
     
     # Use example prompt if selected
     if "example_prompt" in st.session_state:
@@ -266,37 +221,57 @@ with col1:
         help="Just describe what you want in plain language - our AI will figure out the details!"
     )
     
-    show_advanced = st.checkbox("Show advanced options", value=False)
+    # Simple settings
+    col_s1, col_s2 = st.columns(2)
+    with col_s1:
+        aspect_ratio = st.selectbox(
+            "Image Shape",
+            ["Square (1:1)", "Landscape (16:9)", "Portrait (9:16)", "Social Media (optimized)"],
+            index=0,
+            help="Choose the shape of your image"
+        )
+    
+    with col_s2:
+        image_quality = st.select_slider(
+            "Quality",
+            options=["Standard", "High", "Maximum"],
+            value="High",
+            help="Higher quality takes slightly longer to generate"
+        )
     
-    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")
+    # Examples for inspiration
+    with st.expander("Need inspiration? Click here for example prompts", expanded=False):
+        st.markdown('<div class="examples-section">', unsafe_allow_html=True)
         
-        # 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")
+        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"
+        ]
         
-        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!")
+        for i, example in enumerate(example_prompts):
+            if st.button(f"Example {i+1}", key=f"example_{i}"):
+                st.session_state["example_prompt"] = example
+                st.experimental_rerun()
+        
+        st.markdown('</div>', unsafe_allow_html=True)
+    
+    # Advanced options
+    with st.expander("Advanced Options (optional)", expanded=False):
+        st.markdown('<div class="settings-section">', unsafe_allow_html=True)
+        
+        col_a1, col_a2 = st.columns(2)
+        with col_a1:
+            business_name = st.text_input("Business Name", placeholder="e.g., Growth Experts")
+        with col_a2:
+            business_type = st.text_input("Business Type/Industry", placeholder="e.g., Marketing Agency")
         
-        # Visual Style (all optional)
-        st.markdown("##### Visual Style (Optional)")
-        col_v1, col_v2 = st.columns(2)
-        with col_v1:
+        col_a3, col_a4 = st.columns(2)
+        with col_a3:
             color_scheme = st.text_input("Color Scheme", placeholder="e.g., Blue and gold, #FF5733")
-        with col_v2:
+        with col_a4:
             image_style = st.selectbox(
                 "Style",
                 ["Modern", "Professional", "Casual", "Elegant", "Bold", "Minimalist", "Playful", "Custom"],
@@ -307,83 +282,38 @@ with col1:
             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 = ""
+        
+        st.markdown('</div>', unsafe_allow_html=True)
     
     # Build the prompt
     if "current_prompt" in st.session_state:
         complete_prompt = st.session_state["current_prompt"]
         del st.session_state["current_prompt"]
     else:
-        # 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.")
+        # Simplified prompt building
+        if quick_description:
+            prompt_parts = [quick_description]
             
-        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:
+            # Add business info if provided
+            if 'business_name' in locals() and business_name:
                 prompt_parts.append(f"Business Name: {business_name}")
-            if business_type:
+            if 'business_type' in locals() and 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:
+                
+            # Add style info if provided
+            if 'color_scheme' in locals() and 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}")
-            
+            if 'image_style' in locals() and image_style == "Custom" and 'custom_style' in locals() and custom_style:
+                prompt_parts.append(f"Style: {custom_style}")
+            elif 'image_style' in locals() and image_style != "Custom":
+                prompt_parts.append(f"Style: {image_style}")
+                
             # Add aspect ratio
             if "Square" in aspect_ratio:
                 prompt_parts.append("Create a square (1:1) image")
@@ -391,48 +321,41 @@ with col1:
                 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)")
+            elif "Social Media" in aspect_ratio:
+                prompt_parts.append("Optimize dimensions for social media sharing")
             
-            # 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)
-    
-    # Display the complete prompt with ability to edit
-    final_prompt = st.text_area(
-        "Complete Prompt (Edit as needed)",
-        value=complete_prompt,
-        height=150
-    )
+            # Add special instructions if provided
+            if 'special_instructions' in locals() and special_instructions:
+                prompt_parts.append(special_instructions)
+                
+            complete_prompt = ". ".join(prompt_parts)
+        else:
+            complete_prompt = ""
     
     # Generate button
     generate_button = st.button(
-        "🎨 Generate Marketing Graphic",
-        disabled=not api_key,
+        "🎨 Generate My Graphic",
+        disabled=not quick_description,
         use_container_width=True
     )
     
+    if not quick_description:
+        st.info("Please describe what type of lead magnet or marketing graphic you want to create")
+
+# Display generated image
+with col2:
+    st.markdown('<div class="custom-subheader">Generated Graphic</div>', unsafe_allow_html=True)
+    
     if generate_button:
-        with st.spinner("Generating your marketing graphic... This may take up to 60 seconds"):
+        with st.spinner("Creating your graphic... This may take up to 60 seconds"):
             # Save the prompt to history
-            if final_prompt not in st.session_state["prompt_history"]:
-                st.session_state["prompt_history"].insert(0, final_prompt)
+            if complete_prompt not in st.session_state["prompt_history"]:
+                st.session_state["prompt_history"].insert(0, complete_prompt)
                 # Keep only the last 10 prompts
                 st.session_state["prompt_history"] = st.session_state["prompt_history"][:10]
             
             # Call API to generate image
-            result = generate_image_from_prompt(final_prompt, api_key)
+            result = generate_image_from_prompt(complete_prompt)
             
             if result["success"]:
                 # Handle image URL or base64 data
@@ -453,60 +376,61 @@ with col1:
             else:
                 st.error(result["message"])
                 if "API key" in result["message"]:
-                    st.warning("Please check that your API key is valid and has access to the GPT-4o model")
-
-# Display generated image
-with col2:
-    st.markdown('<div class="custom-subheader">Generated Marketing Graphic</div>', unsafe_allow_html=True)
+                    st.warning("There was an issue with the API key. Please try again or contact support.")
     
     if st.session_state["generated_image"] is not None:
         st.markdown('<div class="generated-image">', unsafe_allow_html=True)
         st.image(st.session_state["generated_image"], use_column_width=True)
         st.markdown('</div>', unsafe_allow_html=True)
         
-        # Download button
-        img_buffer = BytesIO()
-        st.session_state["generated_image"].save(img_buffer, format="PNG")
-        img_bytes = img_buffer.getvalue()
+        # Download and regenerate buttons in columns
+        col_b1, col_b2 = st.columns(2)
         
-        st.download_button(
-            label="💾 Download Image",
-            data=img_bytes,
-            file_name=f"marketing_graphic_{int(time.time())}.png",
-            mime="image/png",
-            use_container_width=True
-        )
+        with col_b1:
+            # Download button
+            img_buffer = BytesIO()
+            st.session_state["generated_image"].save(img_buffer, format="PNG")
+            img_bytes = img_buffer.getvalue()
+            
+            st.download_button(
+                label="💾 Download Image",
+                data=img_bytes,
+                file_name=f"lead_magnet_{int(time.time())}.png",
+                mime="image/png",
+                use_container_width=True
+            )
         
-        # Regenerate button
-        if st.button("🔄 Regenerate with Same Prompt", use_container_width=True):
-            with st.spinner("Regenerating image... This may take up to 60 seconds"):
-                result = generate_image_from_prompt(final_prompt, api_key)
-                
-                if result["success"]:
-                    if "image_url" in result:
-                        try:
-                            response = requests.get(result["image_url"])
-                            img = Image.open(BytesIO(response.content))
-                            st.session_state["generated_image"] = img
-                        except Exception as e:
-                            st.error(f"Error loading image: {str(e)}")
-                    elif "image_data" in result:
-                        try:
-                            img_data = base64.b64decode(result["image_data"])
-                            img = Image.open(BytesIO(img_data))
-                            st.session_state["generated_image"] = img
-                        except Exception as e:
-                            st.error(f"Error decoding image: {str(e)}")
-                else:
-                    st.error(result["message"])
+        with col_b2:
+            # Regenerate button
+            if st.button("🔄 Regenerate", use_container_width=True):
+                with st.spinner("Regenerating image... This may take up to 60 seconds"):
+                    result = generate_image_from_prompt(complete_prompt)
+                    
+                    if result["success"]:
+                        if "image_url" in result:
+                            try:
+                                response = requests.get(result["image_url"])
+                                img = Image.open(BytesIO(response.content))
+                                st.session_state["generated_image"] = img
+                            except Exception as e:
+                                st.error(f"Error loading image: {str(e)}")
+                        elif "image_data" in result:
+                            try:
+                                img_data = base64.b64decode(result["image_data"])
+                                img = Image.open(BytesIO(img_data))
+                                st.session_state["generated_image"] = img
+                            except Exception as e:
+                                st.error(f"Error decoding image: {str(e)}")
+                    else:
+                        st.error(result["message"])
     else:
-        st.info("Your generated marketing graphic will appear here. Fill in the form and click 'Generate Marketing Graphic' to create your image.")
-        st.image("https://via.placeholder.com/800x800.png?text=Marketing+Graphic+Preview", use_column_width=True)
+        st.info("Your generated graphic will appear here. Describe what you want in the text area and click 'Generate My Graphic'.")
+        # Placeholder image
+        st.image("https://via.placeholder.com/800x800.png?text=Your+Graphic+Will+Appear+Here", use_column_width=True)
 
-# Footer with disclaimer
+# Footer with simplified disclaimer
 st.markdown("""
 <div class="footnote">
-    <p><strong>Disclaimer:</strong> This tool uses OpenAI's 4o Image Generation to create marketing graphics. The quality and accuracy of the generated images depend on the model's capabilities and your prompt. For professional marketing materials, consider reviewing and refining the generated images with a graphic designer.</p>
-    <p>Images generated using this tool comply with OpenAI's usage policies. You are responsible for ensuring that your use of the generated images complies with applicable laws and regulations.</p>
+    <p><strong>Note:</strong> This tool uses OpenAI's latest 4o Image Generation technology to create high-quality marketing graphics and lead magnets. For best results, be specific about what type of lead magnet or marketing content you're creating.</p>
 </div>
 """, unsafe_allow_html=True)