app.py

"""
BrandBrain AI Marketing Campaign Generator - Gemini API Version
Updated to use Google's Gemini AI API instead of Ollama

Requirements:
google-generativeai==0.3.2
gradio==4.44.0
requests==2.31.0
"""

import gradio as gr
import google.generativeai as genai
import os
from datetime import datetime


class BrandBrainGeminiGenerator:
    def __init__(self, api_key=None):
        self.api_key = api_key
        if api_key:
            genai.configure(api_key=api_key)
            self.model = genai.GenerativeModel("gemini-2.5-flash")

    def configure_api(self, api_key):
        """Configure Gemini API with provided key"""
        try:
            genai.configure(api_key=api_key)
            self.model = genai.GenerativeModel("gemini-2.5-flash")
            self.api_key = api_key
            return True
        except Exception as e:
            return False

    def generate_campaign_prompt(
        self, brand_name, tone, audience, product_description, primary_goal
    ):
        """Create a comprehensive prompt for campaign generation"""
        prompt = f"""
You are BrandBrain AI, an expert marketing strategist. Generate a complete marketing campaign for the following brand:

BRAND: {brand_name}
TONE: {tone}
TARGET AUDIENCE: {audience}
PRODUCT/OFFER: {product_description}
PRIMARY GOAL: {primary_goal}

Create a comprehensive marketing campaign with the following sections (use EXACT formatting with ALL CAPS headers):

VALUE PROPOSITION
[One compelling line that captures the core value]

ELEVATOR PITCH
[A persuasive paragraph explaining what the brand does, who it serves, and why it matters]

LANDING PAGE HEADLINES
[3 different headline options for landing pages]
1. [Headline 1]
2. [Headline 2]
3. [Headline 3]

LANDING PAGE STRUCTURE
Hero Section: [Description of hero section content]
Features Section: [Key features to highlight]
Social Proof Section: [Types of testimonials/proof needed]
Call-to-Action: [Primary CTA strategy]

SOCIAL POSTS
[6 short social media posts, 3 for Instagram and 3 for LinkedIn, each with relevant hashtags]

Instagram Posts:
1. [Post 1 with hashtags]
2. [Post 2 with hashtags]
3. [Post 3 with hashtags]

LinkedIn Posts:
1. [Post 1 with hashtags]
2. [Post 2 with hashtags]
3. [Post 3 with hashtags]

AD COPY
[3 Facebook/Meta ad variations with different approaches]
1. [Ad variation 1]
2. [Ad variation 2]
3. [Ad variation 3]

EMAIL SEQUENCE
Subject Lines:
1. [Subject line 1]
2. [Subject line 2]
3. [Subject line 3]

Email Bodies:
1. [Short email body 1]
2. [Short email body 2]
3. [Short email body 3]

IMAGE PROMPTS
[3 creative briefs for visual content]
1. [Image prompt 1]
2. [Image prompt 2]
3. [Image prompt 3]

Make everything specific to the brand, tone, and audience provided. Be creative and strategic.
"""
        return prompt

    def generate_campaign(
        self,
        brand_name,
        tone,
        audience,
        product_description,
        primary_goal,
        api_key,
        progress=gr.Progress(),
    ):
        """Generate complete marketing campaign using Gemini API"""

        # Validate inputs
        if not all(
            [
                brand_name.strip(),
                tone.strip(),
                audience.strip(),
                product_description.strip(),
                primary_goal.strip(),
            ]
        ):
            return "Error: Please fill in all fields before generating a campaign."

        if not api_key.strip():
            return "Error: Please provide your Gemini API key."

        # Configure API
        progress(0.1, desc="Configuring Gemini API...")
        if not self.configure_api(api_key):
            return "Error: Invalid API key. Please check your Gemini API key and try again."

        # Generate prompt
        progress(0.3, desc="Creating campaign prompt...")
        prompt = self.generate_campaign_prompt(
            brand_name, tone, audience, product_description, primary_goal
        )

        # Call Gemini API
        progress(0.5, desc="Generating campaign with Gemini AI...")
        try:
            response = self.model.generate_content(
                prompt,
                generation_config=genai.types.GenerationConfig(
                    temperature=0.7,
                    top_p=0.9,
                    max_output_tokens=2048,
                ),
            )
            campaign_output = response.text

        except Exception as e:
            return f"Error calling Gemini API: {str(e)}"

        progress(1.0, desc="Campaign generated!")

        # Add metadata
        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        final_output = f"""# BrandBrain AI Marketing Campaign (Gemini Powered)
Generated on: {timestamp}
Brand: {brand_name}
Tone: {tone}
Audience: {audience}
Goal: {primary_goal}

---

{campaign_output}

---
Generated by BrandBrain AI Marketing Campaign Generator - Powered by Google Gemini
"""

        return final_output


def save_campaign_to_file(campaign_text, brand_name):
    """Save campaign to a downloadable text file"""
    if not campaign_text or campaign_text.startswith("Error:"):
        return None

    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"campaign_{brand_name.replace(' ', '_')}_{timestamp}.txt"

    with open(filename, "w", encoding="utf-8") as f:
        f.write(campaign_text)

    return filename


def create_interface():
    """Create the Gradio interface"""

    # Initialize the generator
    generator = BrandBrainGeminiGenerator()

    with gr.Blocks(
        title="BrandBrain AI Marketing Campaign Generator - Gemini Powered",
        theme=gr.themes.Soft(),
    ) as interface:

        # Header
        gr.Markdown(
            """
        # 🧠 BrandBrain AI Marketing Campaign Generator
        ### Powered by Google Gemini AI
        
        Create comprehensive marketing campaigns in minutes using Google's advanced Gemini AI model.
        """
        )

        with gr.Row():
            # Input Section
            with gr.Column(scale=1):
                gr.Markdown("## 🔑 API Configuration")

                api_key = gr.Textbox(
                    label="Gemini API Key",
                    placeholder="Enter your Google Gemini API key",
                    type="password",
                    info="Get your API key from: https://aistudio.google.com/app/apikey",
                )

                gr.Markdown("## 📝 Campaign Details")

                brand_name = gr.Textbox(
                    label="Brand Name", placeholder="Enter your brand or company name"
                )

                tone = gr.Dropdown(
                    label="Brand Tone",
                    choices=[
                        "Professional & Authoritative",
                        "Friendly & Conversational",
                        "Bold & Energetic",
                        "Luxury & Sophisticated",
                        "Fun & Playful",
                        "Trustworthy & Reliable",
                        "Innovative & Tech-Forward",
                        "Warm & Caring",
                    ],
                    value="Professional & Authoritative",
                )

                audience = gr.Textbox(
                    label="Target Audience",
                    placeholder="e.g., Small business owners, Tech professionals, Parents",
                    lines=2,
                )

                product_description = gr.Textbox(
                    label="Product/Offer Description",
                    placeholder="Describe what you're selling or promoting",
                    lines=3,
                )

                primary_goal = gr.Dropdown(
                    label="Primary Campaign Goal",
                    choices=[
                        "Increase Sales",
                        "Build Brand Awareness",
                        "Generate Leads",
                        "Drive Website Traffic",
                        "Boost App Downloads",
                        "Promote Event Registration",
                        "Increase Newsletter Signups",
                        "Launch New Product",
                    ],
                    value="Increase Sales",
                )

                # Generate button
                generate_btn = gr.Button(
                    "🚀 Generate Campaign with Gemini", variant="primary", size="lg"
                )

            # Output Section
            with gr.Column(scale=1):
                gr.Markdown("## 📊 Generated Campaign")

                campaign_output = gr.Textbox(
                    label="Marketing Campaign",
                    lines=25,
                    max_lines=50,
                    show_copy_button=True,
                    placeholder="Your Gemini-powered marketing campaign will appear here...",
                )

                with gr.Row():
                    download_btn = gr.DownloadButton(
                        "💾 Download as TXT",
                        size="sm",
                        variant="secondary",
                        visible=False,
                    )

        # Footer
        gr.Markdown(
            """
        ---
        **Getting Started:**
        1. Get your free Gemini API key from [Google AI Studio](https://aistudio.google.com/app/apikey)
        2. Enter your API key above (it's stored securely in your session)
        3. Fill in your campaign details
        4. Generate your comprehensive marketing campaign!
        
        **Features:**
        - ✅ Advanced AI-powered campaign generation
        - ✅ 8 different marketing asset types
        - ✅ Customized to your brand and audience
        - ✅ Professional copywriting quality
        
        *Powered by Google Gemini AI*
        """
        )

        # Event handlers
        def generate_and_prepare_download(
            brand_name,
            tone,
            audience,
            product_description,
            primary_goal,
            api_key,
            progress=gr.Progress(),
        ):
            # Generate campaign
            campaign = generator.generate_campaign(
                brand_name,
                tone,
                audience,
                product_description,
                primary_goal,
                api_key,
                progress,
            )

            # Prepare download file
            if not campaign.startswith("Error:") and brand_name.strip():
                filename = save_campaign_to_file(campaign, brand_name)
                return campaign, gr.DownloadButton(
                    "💾 Download as TXT", value=filename, visible=True
                )
            else:
                return campaign, gr.DownloadButton("💾 Download as TXT", visible=False)

        # Wire up the generate button
        generate_btn.click(
            fn=generate_and_prepare_download,
            inputs=[
                brand_name,
                tone,
                audience,
                product_description,
                primary_goal,
                api_key,
            ],
            outputs=[campaign_output, download_btn],
            show_progress=True,
        )

    return interface


def main():
    """Main function to run the application"""
    print("🧠 Starting BrandBrain AI Marketing Campaign Generator...")
    print("✨ Powered by Google Gemini AI")
    print("🔑 Get your API key: https://aistudio.google.com/app/apikey")
    print()

    # Create and launch interface
    interface = create_interface()

    # Launch with corrected settings
    interface.launch(server_port=7860, share=False, show_error=True)


if __name__ == "__main__":
    main()