app.py

"""
Arabic Fashion Marketing AI - Hugging Face Space
mT5 + LoRA → 3 Arabic marketing post variants
"""

import gradio as gr
import torch
import re
import random
from transformers import MT5ForConditionalGeneration, MT5Tokenizer
from peft import PeftModel

# ─── Load Model ───────────────────────────────────────────────────────────────
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
device = DEVICE

print("===== Application Startup =====")
print("Loading model from Space files...")
base_model_name = "google/mt5-base"
tokenizer = MT5Tokenizer.from_pretrained(".")
base_model = MT5ForConditionalGeneration.from_pretrained(base_model_name)
model = PeftModel.from_pretrained(base_model, ".").to(DEVICE)
model.eval()
print(f"✅ Model loaded on: {DEVICE}")

# ─── Config ───────────────────────────────────────────────────────────────────
EMOJIS = ['✨', '💫', '🌟', '💎', '🔥', '🤍', '💛', '🎉', '💖']

PLATFORM_CONFIG = {
    'Instagram':   {'hashtags': 8, 'style': 'formal_marketing', 'max_sentences': 4},
    'TikTok':      {'hashtags': 5, 'style': 'formal_marketing', 'max_sentences': 3},
    'Facebook':    {'hashtags': 4, 'style': 'formal_marketing', 'max_sentences': 4},
    'Website/Ads': {'hashtags': 3, 'style': 'formal_marketing', 'max_sentences': 4},
}

AR_MAP = {
    'platform':    {'Instagram': 'إنستجرام', 'TikTok': 'تيك توك', 'Facebook': 'فيسبوك', 'Website/Ads': 'الموقع والإعلانات'},
    'subcategory': {'Dresses': 'فستان', 'Bags': 'حقيبة', 'Jewelry': 'مجوهرات', 'Loungewear': 'طقم منزلي',
                    'Outerwear': 'معطف', 'Sandals': 'صندل', 'Heels': 'حذاء بكعب', 'Pants': 'بنطال',
                    'Abayas': 'عباية', 'Sneakers': 'حذاء رياضي', 'Tops': 'بلوزة', 'Boots': 'بوت',
                    'Scarves': 'وشاح', 'Flats': 'حذاء مسطح', 'Activewear': 'ملابس رياضية',
                    'Sunglasses': 'نظارة شمسية', 'Belts/Others': 'حزام', 'Belts / Others': 'حزام'},
    'material':    {'Leather': 'الجلد', 'Genuine Leather': 'الجلد الطبيعي', 'Cotton': 'القطن',
                    'Silk': 'الحرير', 'Viscose': 'الفيسكوز', 'Chiffon': 'الشيفون', 'Denim': 'الدنيم',
                    'Suede': 'الشمواه', 'Jersey': 'الجيرسي', 'Polyester': 'البوليستر', 'Linen': 'الكتان',
                    'Crepe': 'الكريب', 'Canvas': 'الكانفاس', 'Mesh': 'الشبك', 'Rubber': 'المطاط',
                    'Synthetic': 'الخامات الصناعية', 'Modal': 'المودال', 'Bamboo': 'ألياف الخيزران',
                    'Gold-plated': 'المعدن المطلي بالذهب'},
    'season':      {'Summer': 'الصيف', 'Winter': 'الشتاء', 'Spring': 'الربيع', 'Autumn': 'الخريف',
                    'All-Season': 'كل المواسم', 'All Seasons': 'كل المواسم'},
    'style':       {'Casual': 'كاجوال', 'Trendy': 'عصري', 'Formal': 'رسمي', 'Sporty': 'رياضي',
                    'Modest': 'محتشم', 'Luxury': 'فاخر'},
    'occasion':    {'Daily': 'الإطلالات اليومية', 'Work': 'أيام العمل', 'Party': 'السهرات والمناسبات',
                    'Travel': 'السفر', 'Ramadan/Eid': 'رمضان والعيد', 'Gym/Beach': 'الرياضة والشاطئ',
                    'Wedding': 'الأفراح', 'Sports': 'الرياضة'},
    'goal':        {'Sales': 'المبيعات', 'Brand Awareness': 'الوعي بالعلامة', 'Engagement': 'التفاعل', 'Launch': 'الإطلاق'},
    'tone':        {'Friendly': 'ودية', 'Bold': 'جريئة', 'Luxury': 'فاخرة', 'Minimal': 'راقية',
                    'Casual': 'خفيفة', 'Formal': 'رسمية', 'Confidence': 'واثقة'},
    'emotion':     {'Comfort': 'الراحة', 'Confidence': 'الثقة', 'Excitement': 'الحماس',
                    'Trust': 'الثقة', 'Elegance': 'الأناقة', 'Desire': 'الرغبة', 'Joy': 'البهجة', 'Pride': 'الفخر'},
    'cta':         {'Discover More': 'استكشفي المزيد', 'Shop Now': 'اطلبيه الآن',
                    'Limited Offer': 'استفيدي من العرض الآن', 'Explore Collection': 'استكشفي المجموعة',
                    'Buy Today': 'اطلبيه اليوم'},
    'usp':         {'Premium embroidery': 'التطريز الفاخر', 'Performance meets style': 'الأداء الذي يجمع بين العملية والأناقة',
                    'Timeless design': 'التصميم الخالد', 'Breathable fabric': 'الخامة المريحة للارتداء'},
}

HOOK_BANK = {
    'Instagram': {
        'product':   [('luxury_statement', 'الفخامة الحقيقية تظهر في التفاصيل التي تصنع حضوراً مميزاً'),
                      ('value_statement',  'الأناقة الراقية تبدأ من اختيار القطعة التي تحمل قيمة واضحة')],
        'lifestyle': [('emotional', 'حين تقود الأناقة تفاصيل الإطلالة يصبح الحضور أكثر تأثيراً'),
                      ('question',  'هل تبحثين عن إطلالة تنسجم مع مناسباتك بأسلوب يلفت الأنظار')],
        'campaign':  [('cta_lead',        'قدمي إطلالتك بأسلوب يمنح المحتوى جاذبية أوضح'),
                      ('value_statement', 'اجعلي حضورك الرقمي أكثر أناقة من خلال قطعة تعكس الذوق الرفيع')],
    },
    'TikTok': {
        'product':   [('direct_benefit', 'اختاري قطعة تمنح إطلالتك توازناً بين العملية والأناقة'),
                      ('value_statement','قطعة مصممة لتمنح حضورك اليومي لمسة أكثر ترتيباً')],
        'lifestyle': [('question', 'هل تبحثين عن تصميم يواكب يومك بأسلوب أنيق'),
                      ('emotional','حين تلتقي الثقة مع البساطة تصبح الإطلالة أكثر حضوراً')],
        'campaign':  [('cta_lead',       'امنحي محتواك بداية أقوى تبرز قيمة الإطلالة'),
                      ('direct_benefit', 'قدمي محتوى أسرع جذباً من خلال قطعة تعبّر عن أسلوب واضح')],
    },
    'Facebook': {
        'product':   [('value_statement',  'اختيار عملي يعكس أناقة واضحة وقيمة مستمرة'),
                      ('luxury_statement', 'القطع المميزة هي التي تجمع بين الجودة وأناقة التفاصيل')],
        'lifestyle': [('question', 'هل تبحثين عن قطعة تجمع بين القيمة والأناقة'),
                      ('emotional','الثقة تبدأ من اختيار ينسجم مع حضورك اليومي')],
        'campaign':  [('cta_lead',        'قدمي محتوى يبرز جودة المنتج بأسلوب احترافي'),
                      ('value_statement', 'اجعلي رسالتك أكثر وضوحاً مع قطعة تدعم حضور العلامة')],
    },
    'Website/Ads': {
        'product':   [('value_statement', 'تفاصيل المنتج الواضحة تمنح الاختيار قيمة أكبر')],
        'lifestyle': [('question',        'هل تبحثين عن قطعة تقدم فائدة واضحة وأناقة عملية')],
        'campaign':  [('cta_lead',        'قدمي منتجك بأسلوب يبرز قيمته بصورة مباشرة')],
    },
}

REALISTIC_HASHTAGS = {
    'Instagram':   ['#أناقة_نسائية', '#إطلالة_راقية', '#موضة_راقية', '#إطلالة_السهرات', '#تفاصيل_فاخرة'],
    'TikTok':      ['#ستايل_عملي', '#إطلالة_عصرية', '#موضة_يومية', '#لوك_يومي', '#سنيكرز_نسائي'],
    'Facebook':    ['#إطلالة_العمل', '#أناقة_عملية', '#ثقة_في_الاختيار', '#جودة_راقية', '#تفاصيل_أنيقة'],
    'Website/Ads': ['#تصميم_مميز', '#خامة_راقية', '#اختيار_أنيق'],
}

BANNED_PHRASES = [
    'إحساساً من', 'أسلوب فاخرة', 'أسلوب جريئة', 'حضوراً رسمية', 'رسالة بصرية',
    'حضور رقمي', 'يمنح المحتوى', 'تفصيل يعكس', 'بداية تبرز قيمة المنتج',
    'كما يمنح', 'رسالة تسويقية', 'جاذبية المحتوى',
    'حيث نركز بشكل أساسي', 'لنضمن أن الجودة'
]

CTA_PHRASES = list(AR_MAP['cta'].values())

# ─── Helpers ──────────────────────────────────────────────────────────────────
def ar(category, value):
    return AR_MAP.get(category, {}).get(str(value).strip(), str(value).strip())

def normalize_spaces(text):
    return re.sub(r'\s+', ' ', str(text)).strip()

def clean_terminal_punctuation(text):
    text = normalize_spaces(text)
    text = re.sub(r'[.؟!]+$', '', text)
    return text.strip('، ')

def to_sentence(text, emoji=None, ending='.'):
    text = clean_terminal_punctuation(text)
    if not text: return ''
    return f"{text}{emoji}{ending}" if emoji else f"{text}{ending}"

def hook_to_sentence(text, hook_type):
    return to_sentence(text, random.choice(EMOJIS), ending='؟' if hook_type == 'question' else '.')

def apply_grammar_fixes(text):
    fixes = {
        'إحساساً من الأناقة': 'إحساساً بالأناقة',
        'إحساساً من الثقة': 'إحساساً بالثقة',
        'أسلوب فاخرة': 'أسلوب فاخر',
        'أسلوب جريئة': 'أسلوب جريء',
        'حضوراً رسمية': 'حضوراً رسمياً',
        'تفصيل يعكس': 'تفاصيل تعكس',
        'هذه القطعة هو': 'هذه القطعة هي',
        'هذه الحقيبة هو': 'هذه الحقيبة هي',
        'هذه الحقيبة الذي': 'هذه الحقيبة التي',
        'هذه القطعة الذي': 'هذه القطعة التي',
        'فـ هذه': 'فهذه',
        'يليق بككِ': 'يليق بكِ',
        'يعزز جاذبية العرض وتدعم': 'يعزز جاذبية العرض ويدعم',
        'أسلوب فاخرة يدعم': 'أسلوب فاخر يدعم',
        'أسلوب جريئة يلفت': 'أسلوب جريء يلفت',
    }
    for bad, good in fixes.items():
        text = text.replace(bad, good)
    text = re.sub(r'\s+([🔥✨💫🌟💎🤍💛🎉💖])', r'\1', text)
    return normalize_spaces(text)

def contains_non_arabic_wording(text, product_name=''):
    tmp = text.replace(product_name, '')
    tmp = re.sub(r'[#_]', ' ', tmp)
    return bool(re.search(r'[A-Za-z]{2,}', tmp))

def strip_cta_phrases(text):
    t = str(text)
    for cta in CTA_PHRASES:
        if cta in t and normalize_spaces(t) != cta:
            t = t.replace(f'، {cta}', '').replace(f'،{cta}', '').replace(cta, '')
    return normalize_spaces(t.strip(' ،.'))

def clean_generated_line(text, product_name=''):
    if not text: return ''
    text = str(text).split('الهاشتاجات:')[0]
    text = re.sub(r'[#].*', '', text)
    text = normalize_spaces(text)
    text = strip_cta_phrases(text)
    if contains_non_arabic_wording(text, product_name): return ''
    for bad in BANNED_PHRASES:
        if bad in text: return ''
    text = apply_grammar_fixes(text)
    if len(text.split()) < 5: return ''
    return text

def choose_hook(platform, focus):
    bank = HOOK_BANK.get(platform, HOOK_BANK['Instagram'])
    return random.choice(bank.get(focus, bank['product']))

def cta_line(data):
    platform = data.get('Platform', 'Instagram')
    base = ar('cta', data.get('CTA', 'Explore Collection'))
    return to_sentence(base, random.choice(EMOJIS) if platform != 'TikTok' else None)

def parse_product_details(text):
    data = {}
    for line in str(text).splitlines():
        if ':' in line:
            k, v = line.split(':', 1)
            data[k.strip()] = v.strip()
    if 'Category' in data and ' - ' in data['Category']:
        main, sub = data['Category'].split(' - ', 1)
        data['_main_category'] = main.strip()
        data['_subcategory'] = sub.strip()
    return data

def format_model_input(data):
    return f"""Generate polished formal Arabic marketing copy and Arabic hashtags.
Strict rules:
- Keep ONLY the product name in English.
- Write all other text in polished formal Arabic with correct grammar.
- Avoid literal repetition, weak generic phrases, and repeated CTAs.
- Product-focused MUST mention product name, material, and USP clearly.
- Lifestyle-focused MUST mention occasion or season and fit the target audience naturally.
- Campaign-focused MUST mention the platform in Arabic once, then continue naturally.
- Use realistic fashion-marketing hashtags in Arabic only.

Category: {data.get('Category', '')}
Product: {data.get('Product', '')}
Description: {data.get('Description', '')}
Material: {data.get('Material', '')}
Season: {data.get('Season', '')}
Style: {data.get('Style', '')}
Occasion: {data.get('Occasion', '')}
Target: {data.get('Target', '')}
Goal: {data.get('Goal', '')}
Platform: {data.get('Platform', '')}
Tone: {data.get('Tone', '')}
Emotion: {data.get('Emotion', '')}
CTA: {data.get('CTA', '')}
USP: {data.get('USP', '')}"""

# ─── Generation ───────────────────────────────────────────────────────────────
def generate_candidates(product_details, num_return_sequences=6, max_len=220):
    data = parse_product_details(product_details)
    inputs = tokenizer(format_model_input(data), return_tensors="pt", max_length=512, truncation=True).to(device)
    with torch.no_grad():
        outputs = model.generate(
            **inputs, do_sample=True, temperature=0.9, top_p=0.92, top_k=50,
            num_beams=1, early_stopping=False, no_repeat_ngram_size=3,
            repetition_penalty=1.15, max_length=max_len, num_return_sequences=num_return_sequences,
        )
    decoded = [tokenizer.decode(o, skip_special_tokens=True) for o in outputs]
    return list(dict.fromkeys(decoded))

def product_core_line(data):
    p = data.get('Product', 'المنتج')
    material = ar('material', data.get('Material', ''))
    usp = ar('usp', data.get('USP', ''))
    return f"يتألق {p} بخامة {material} وتفاصيل تعكس {usp} بأسلوب يبرز قيمة الإطلالة."

def product_support_line(data):
    material = ar('material', data.get('Material', ''))
    usp = ar('usp', data.get('USP', ''))
    platform = data.get('Platform', 'Instagram')
    options = {
        'Instagram': [
            f"وتمنح خامة {material} مع {usp} التصميم حضوراً راقياً يلفت الأنظار بأسلوب متوازن.",
            f"وتبرز تفاصيل {usp} مع خامة {material} قيمة التصميم وتمنحه طابعاً أكثر فخامة."
        ],
        'TikTok': [
            f"وتضيف خامة {material} مع {usp} لمسة عملية تجعل الإطلالة أكثر ترتيباً وسهولة في التنسيق.",
            f"ويمنح الجمع بين {material} و{usp} التصميم حضوراً واضحاً يناسب الإطلالة اليومية."
        ],
        'Facebook': [
            f"وتؤكد خامة {material} مع {usp} قيمة التصميم وتمنحه حضوراً أنيقاً يدوم أثره.",
            f"كما تعكس تفاصيل {usp} مع خامة {material} جودة الاختيار وتبرز أناقة التصميم."
        ],
    }
    return random.choice(options.get(platform, options['Instagram']))

def lifestyle_core_line(data):
    p = data.get('Product', 'المنتج')
    occ    = ar('occasion', data.get('Occasion', ''))
    season = ar('season',   data.get('Season', ''))
    style  = ar('style',    data.get('Style', ''))
    target = str(data.get('Target', 'الفئة المستهدفة')).split('(')[0].strip()
    target_ar = 'المرأة' if 'women' in target.lower() or 'نساء' in target else target
    return f"يلائم {p} {occ} في {season} بأسلوب {style} يناسب {target_ar}."

def lifestyle_support_line(data):
    occ    = ar('occasion', data.get('Occasion', ''))
    season = ar('season',   data.get('Season', ''))
    style  = ar('style',    data.get('Style', ''))
    target = str(data.get('Target', 'الفئة المستهدفة')).split('(')[0].strip()
    target_ar = 'المرأة' if 'women' in target.lower() or 'نساء' in target else target
    return random.choice([
        f"ويمنح {occ} حضوراً ينسجم مع {season} ويعكس أسلوباً {style} يليق بـ {target_ar}.",
        f"ليجعل {occ} في {season} أكثر انسجاماً مع إطلالة {style} تعبر عن {target_ar}.",
    ])

def campaign_core_line(data):
    p        = data.get('Product', 'المنتج')
    platform = ar('platform', data.get('Platform', 'Instagram'))
    usp      = ar('usp', data.get('USP', ''))
    openers = {
        'إنستجرام':          f"عند ظهوره على {platform}، يمنح {p} الظهور طابعاً أكثر لفتاً للانتباه عبر إبراز {usp}.",
        'تيك توك':           f"عند ظهوره على {platform}، يمنح {p} المحتوى حضوراً أسرع لفتاً للانتباه من خلال إبراز {usp}.",
        'فيسبوك':            f"عند ظهوره على {platform}، يمنح {p} الحضور طابعاً أكثر وضوحاً عبر إبراز {usp}.",
        'الموقع والإعلانات': f"عند تقديمه عبر {platform}، يمنح {p} الرسالة حضوراً أوضح من خلال إبراز {usp}.",
    }
    return openers.get(platform, openers['إنستجرام'])

def campaign_support_line(data):
    tone = ar('tone', data.get('Tone', 'رسمية'))
    goal = data.get('Goal', '')
    goal_map = {'Sales': 'يعزز جاذبية العرض بصورة أكثر إقناعاً', 'Engagement': 'يدعم التفاعل بصورة أوضح',
                'Brand Awareness': 'يقوي حضور العلامة بصورة أكثر احترافية', 'Launch': 'يبرز انطلاقة المنتج بصورة لافتة'}
    tone_map = {'فاخرة': 'بأسلوب فاخر', 'جريئة': 'بأسلوب جريء', 'رسمية': 'بأسلوب رسمي'}
    return f"ويساعد على إبراز الرسالة بصياغة متوازنة {goal_map.get(goal, 'يعزز حضور الحملة بصورة أوضح')} {tone_map.get(tone, 'بأسلوب أنيق')}."

def dedupe_lines(lines):
    out, seen = [], []
    for ln in lines:
        if not ln: continue
        key = re.sub(r'[^\wء-ي]+', ' ', ln)
        if any(key[:25] == re.sub(r'[^\wء-ي]+', ' ', x)[:25] for x in seen): continue
        out.append(ln); seen.append(ln)
    return out

def hashtagify(text):
    text = re.sub(r'\s+', '_', str(text).strip())
    text = re.sub(r'[^ء-ي0-9_]+', '', text)
    text = re.sub(r'_+', '_', text).strip('_')
    return f"#{text}" if text else ''

def hashtags_for_variant(data, focus):
    platform = data.get('Platform', 'Instagram')
    count = PLATFORM_CONFIG.get(platform, PLATFORM_CONFIG['Instagram'])['hashtags']
    product_tags  = [ar('subcategory', data.get('_subcategory', '')), ar('material', data.get('Material', '')), ar('usp', data.get('USP', ''))]
    lifestyle_tags = [ar('occasion', data.get('Occasion', '')), ar('season', data.get('Season', '')),
                      ar('style', data.get('Style', '')),
                      'أناقة_نسائية' if platform == 'Instagram' else 'ستايل_عملي' if platform == 'TikTok' else 'أناقة_عملية']
    campaign_tags = REALISTIC_HASHTAGS.get(platform, [])
    seeds = product_tags if focus == 'product' else lifestyle_tags if focus == 'lifestyle' else campaign_tags
    tags = []
    for t in seeds:
        h = hashtagify(t)
        if h and h not in tags and len(h) <= 28: tags.append(h)
    for t in REALISTIC_HASHTAGS.get(platform, []):
        h = hashtagify(t.replace('#', ''))
        if h and h not in tags and len(h) <= 28 and len(tags) < count: tags.append(h)
    return ' '.join(tags[:count])

def build_variant(data, focus_key, candidates):
    platform = data.get('Platform', 'Instagram')
    hook_type, hook_text = choose_hook(platform, focus_key)

    if focus_key == 'product':
        line2, line3 = product_core_line(data), product_support_line(data)
    elif focus_key == 'lifestyle':
        line2, line3 = lifestyle_core_line(data), lifestyle_support_line(data)
    else:
        line2, line3 = campaign_core_line(data), campaign_support_line(data)

    hook_line = apply_grammar_fixes(hook_to_sentence(hook_text, hook_type))
    line2 = apply_grammar_fixes(to_sentence(strip_cta_phrases(line2), random.choice(EMOJIS)))
    line3 = apply_grammar_fixes(to_sentence(strip_cta_phrases(line3), random.choice(EMOJIS)))
    cta   = apply_grammar_fixes(cta_line(data))

    final_lines = dedupe_lines([hook_line, line2, line3, cta])
    cleaned_lines = []
    for idx, ln in enumerate(final_lines):
        ln = normalize_spaces(ln)
        if idx < len(final_lines) - 1:
            ln = strip_cta_phrases(ln)
            ln = apply_grammar_fixes(ln)
            if not ln: continue
            ending = '؟' if idx == 0 and hook_type == 'question' else '.'
            ln = re.sub(r'[.؟!]+$', '', ln).strip() + ending
        cleaned_lines.append(ln)

    if not cleaned_lines or not any(cta_p in cleaned_lines[-1] for cta_p in CTA_PHRASES):
        cleaned_lines.append(cta)

    caption = '\n'.join(cleaned_lines[:PLATFORM_CONFIG.get(platform, PLATFORM_CONFIG['Instagram'])['max_sentences']])
    return caption, hashtags_for_variant(data, focus_key)

def generate_3_variants(product_details):
    data = parse_product_details(product_details)
    candidates = generate_candidates(product_details)
    results = []
    for focus in ['product', 'lifestyle', 'campaign']:
        cap, tags = build_variant(data, focus, candidates)
        results.extend([cap, tags])
    return results

# ─── Gradio UI ────────────────────────────────────────────────────────────────
CATEGORY_MAP = {
    "Apparel":     ["Dresses", "Abayas", "Tops", "Pants", "Outerwear", "Activewear", "Loungewear"],
    "Footwear":    ["Sneakers", "Sandals", "Heels", "Flats", "Boots"],
    "Accessories": ["Bags", "Jewelry", "Scarves", "Sunglasses", "Belts / Others"],
}
ALL_SUBCATEGORIES = [s for subs in CATEGORY_MAP.values() for s in subs]

def update_subcategory(cat):
    choices = CATEGORY_MAP.get(cat, [])
    return gr.Dropdown(choices=choices, value=choices[0] if choices else None)

def generate_posts(category, subcategory, product_name, description, material,
                   season, style, occasion, gender, target_audience,
                   campaign_goal, platform, tone, emotion, cta, usp):
    product_details = f"""Category: {category} - {subcategory}
Product: {product_name}
Description: {description}
Material: {material}
Season: {season}
Style: {style}
Occasion: {occasion}
Target: {target_audience} ({gender})
Goal: {campaign_goal}
Platform: {platform}
Tone: {tone}
Emotion: {emotion}
CTA: {cta}
USP: {usp}"""
    return generate_3_variants(product_details)

with gr.Blocks(title="Arabic Fashion Marketing AI") as demo:
    gr.Markdown("""
    # 👗 Arabic Fashion Marketing AI
    ### توليد 3 بوستات تسويقية مختلفة باللغة العربية
    """)

    with gr.Row():
        with gr.Column():
            gr.Markdown("### 📦 معلومات المنتج")
            category     = gr.Dropdown(label="Category",     choices=list(CATEGORY_MAP.keys()), value="Apparel")
            subcategory  = gr.Dropdown(label="Sub-Category", choices=ALL_SUBCATEGORIES, value="Dresses")
            category.change(fn=update_subcategory, inputs=category, outputs=subcategory)
            product_name = gr.Textbox(label="Product Name",  placeholder="e.g. Flowy Midi Dress")
            description  = gr.Textbox(label="Description",   placeholder="e.g. Lightweight chiffon dress")
            material     = gr.Textbox(label="Material",      placeholder="e.g. Chiffon, Cotton, Leather...")
            season       = gr.Dropdown(label="Season",       choices=["Summer","Winter","Spring","Autumn","All-Season"], value="Summer")
            style        = gr.Dropdown(label="Style",        choices=["Casual","Formal","Sporty","Luxury","Modest","Trendy"], value="Casual")
            occasion     = gr.Dropdown(label="Occasion",     choices=["Daily","Work","Party","Wedding","Travel","Sports","Ramadan/Eid","Gym/Beach"], value="Daily")

        with gr.Column():
            gr.Markdown("### 🎯 تفاصيل الحملة")
            gender          = gr.Dropdown(label="Gender",         choices=["Women","Men","Unisex"], value="Women")
            target_audience = gr.Textbox(label="Target Audience", placeholder="e.g. Modern women", value="Modern women")
            campaign_goal   = gr.Dropdown(label="Campaign Goal",  choices=["Sales","Engagement","Brand Awareness","Launch"], value="Sales")
            platform        = gr.Dropdown(label="Platform",       choices=["Instagram","Facebook","TikTok","Website/Ads"], value="Instagram")
            tone            = gr.Dropdown(label="Tone",           choices=["Friendly","Luxury","Bold","Formal","Casual","Minimal","Confidence"], value="Friendly")
            emotion         = gr.Dropdown(label="Emotion",        choices=["Comfort","Confidence","Elegance","Joy","Excitement","Trust","Desire","Pride"], value="Comfort")
            cta             = gr.Dropdown(label="Call to Action", choices=["Shop Now","Discover More","Explore Collection","Buy Today","Limited Offer"], value="Shop Now")
            usp             = gr.Textbox(label="USP",             placeholder="e.g. Breathable fabric", value="Breathable fabric")

    generate_btn = gr.Button("✨ Generate 3 Arabic Posts", variant="primary", size="lg")

    gr.Markdown("---")
    gr.Markdown("### 📝 البوست الأول — تركيز على المنتج")
    with gr.Row():
        cap1  = gr.Textbox(label="النص",       lines=5, rtl=True)
        hash1 = gr.Textbox(label="الهاشتاجات", lines=2, rtl=True)

    gr.Markdown("### 🌿 البوست الثاني — تركيز على اللايف ستايل")
    with gr.Row():
        cap2  = gr.Textbox(label="النص",       lines=5, rtl=True)
        hash2 = gr.Textbox(label="الهاشتاجات", lines=2, rtl=True)

    gr.Markdown("### 🎯 البوست الثالث — تركيز على الحملة")
    with gr.Row():
        cap3  = gr.Textbox(label="النص",       lines=5, rtl=True)
        hash3 = gr.Textbox(label="الهاشتاجات", lines=2, rtl=True)

    generate_btn.click(
        fn=generate_posts,
        inputs=[category, subcategory, product_name, description, material,
                season, style, occasion, gender, target_audience,
                campaign_goal, platform, tone, emotion, cta, usp],
        outputs=[cap1, hash1, cap2, hash2, cap3, hash3]
    )

    gr.Examples(
        examples=[
            ["Apparel","Dresses","Flowy Midi Dress","Lightweight chiffon dress","Chiffon","Summer","Casual","Daily","Women","Modern women","Sales","Instagram","Friendly","Comfort","Shop Now","Breathable fabric"],
            ["Footwear","Sneakers","Classic Sneakers","Leather sneakers","Leather","All-Season","Sporty","Daily","Women","Active women","Engagement","TikTok","Bold","Confidence","Discover More","Performance meets style"],
            ["Accessories","Bags","Shoulder Bag","Leather bag with compartments","Leather","Winter","Luxury","Work","Women","Working women","Brand Awareness","Facebook","Luxury","Elegance","Explore Collection","Timeless design"],
        ],
        inputs=[category, subcategory, product_name, description, material, season, style, occasion,
                gender, target_audience, campaign_goal, platform, tone, emotion, cta, usp],
        run_on_click=False,
    )

if __name__ == "__main__":
    demo.launch(theme=gr.themes.Soft())