import gradio as gr from transformers import pipeline import re from fpdf import FPDF import tempfile # -------------------------------------- # Load Model # -------------------------------------- generator = pipeline("text2text-generation", model="google/flan-t5-xl") # -------------------------------------- # Brand Memory (simple key-value storage) # -------------------------------------- brand_memory = { "tone": "Professional, modern, emotionally resonant", "style": "Story-driven, cinematic, customer-first", } # -------------------------------------- # Core Campaign Generation # -------------------------------------- def generate_campaign(brand_name, product_type, goal, tone, audience, region): combined_tone = f"{tone}. The brand historically sounds like this: {brand_memory['tone']}." prompt = f""" You are an expert creative strategist hired to design a world-class campaign. Your output must read like a premium brand deck. Brand Name: {brand_name} Product Type: {product_type} Goal: {goal} Tone: {combined_tone} Target Audience: {audience} Region: {region} Write: **Creative Direction** – 3–5 line vision. **Emotional Tone** – emotional personality + brand resonance. **Ad Script** – cinematic script with Intro, Body, CTA. **Poster Prompt** – visually detailed description for an AI image generator. **Taglines** – two powerful options. Write in rich natural language, with clear section titles. """ llm_output = generator(prompt, max_new_tokens=650, temperature=0.8)[0]["generated_text"] # SECTION EXTRACTION (Regex) def extract(section): pattern = rf"{section}([\s\S]*?)(?=\n[A-Z][a-zA-Z ]+:|$)" match = re.search(pattern, llm_output, re.IGNORECASE) return match.group(1).strip() if match else "Not found." creative_direction = extract("Creative Direction") emotional_tone = extract("Emotional Tone") ad_script = extract("Ad Script") poster_prompt = extract("Poster Prompt") taglines = extract("Taglines") # FORMATTED MARKDOWN OUTPUT md = f""" # 🎨 **Enterprise Creative Campaign Output** --- ## ✨ Brand: **{brand_name}** ### 🌍 Region: {region} --- ## 🎯 **Creative Direction** {creative_direction} --- ## ❤️ **Emotional Tone** {emotional_tone} --- ## 🎬 **Ad Script** {ad_script} --- ## 🖼 **Poster Prompt** {poster_prompt} --- ## 🔥 **Taglines** {taglines} --- """ return md, creative_direction, emotional_tone, ad_script, poster_prompt, taglines # -------------------------------------- # POSTER PROMPT ONLY # -------------------------------------- def generate_poster_prompt(brand, mood, theme, audience): prompt = f""" Create a cinematic AI poster prompt for Stable Diffusion. Brand: {brand} Mood: {mood} Theme: {theme} Audience: {audience} Describe lighting, color palette, environment, camera angle, people, typography. """ result = generator(prompt, max_new_tokens=300, temperature=0.8)[0]["generated_text"] return result # -------------------------------------- # UPDATE BRAND MEMORY # -------------------------------------- def update_brand_memory(new_tone, new_style): if new_tone: brand_memory["tone"] = new_tone if new_style: brand_memory["style"] = new_style return f"✅ Brand memory updated successfully!", str(brand_memory) # -------------------------------------- # Export as PDF # -------------------------------------- def export_pdf(text): pdf = FPDF() pdf.add_page() pdf.set_auto_page_break(auto=True, margin=15) pdf.set_font("Arial", size=12) for line in text.split("\n"): pdf.multi_cell(0, 10, line) tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") pdf.output(tmp.name) return tmp.name # -------------------------------------- # Gradio UI # -------------------------------------- with gr.Blocks(theme=gr.themes.Monochrome()) as demo: gr.HTML("""

🚀 Enterprise Creative Campaign Suite

Powered by HuggingFace + Rookus Reasoning Core

""") with gr.Tabs(): # ---------------------------- TAB 1: Campaign Generator ---------------------------- with gr.Tab("🎨 Campaign Generator"): with gr.Row(): brand_name = gr.Textbox(label="Brand Name") product_type = gr.Textbox(label="Product Type") goal = gr.Textbox(label="Campaign Goal") tone = gr.Textbox(label="Tone / Voice") audience = gr.Textbox(label="Target Audience") region = gr.Textbox(label="Region") run_btn = gr.Button("🚀 Generate Full Campaign", variant="primary") output_md = gr.Markdown("Your campaign will appear here…") # Hidden values for export cd = gr.Textbox(visible=False) et = gr.Textbox(visible=False) ad = gr.Textbox(visible=False) pp = gr.Textbox(visible=False) tg = gr.Textbox(visible=False) run_btn.click( fn=generate_campaign, inputs=[brand_name, product_type, goal, tone, audience, region], outputs=[output_md, cd, et, ad, pp, tg] ) # ---------------------------- TAB 2: Poster Prompt ---------------------------- with gr.Tab("🖼 Poster Prompt Generator"): brandp = gr.Textbox(label="Brand Name") mood = gr.Textbox(label="Mood & Emotion") themep = gr.Textbox(label="Theme") audp = gr.Textbox(label="Audience") gen_poster_btn = gr.Button("🎨 Generate Poster Prompt") poster_out = gr.Textbox(label="Poster Prompt", lines=8) gen_poster_btn.click( generate_poster_prompt, [brandp, mood, themep, audp], poster_out ) # ---------------------------- TAB 3: Brand Memory ---------------------------- with gr.Tab("🧠 Brand Tone Memory"): gr.Markdown("### Update your persistent brand tone + style memory") new_tone = gr.Textbox(label="New Tone") new_style = gr.Textbox(label="New Style") upd_btn = gr.Button("💾 Update Memory") msg = gr.Textbox(label="Status") mem = gr.Textbox(label="Current Memory", lines=5) upd_btn.click(update_brand_memory, [new_tone, new_style], [msg, mem]) # ---------------------------- TAB 4: Export Center ---------------------------- with gr.Tab("📦 Export Center"): gr.Markdown("### Export your full campaign as PDF") export_btn = gr.Button("📄 Export Campaign as PDF") export_file = gr.File() export_btn.click(export_pdf, inputs=output_md, outputs=export_file) demo.launch()