Update app.py

app.py

@@ -28,6 +28,33 @@ def model_inference(
     if text == "" and images:
         gr.Error("Please input a text query along the image(s).")
 
+    # Check if the input is related to marketing
+    marketing_keywords = [
+        "product", "brand", "advertisement", "marketing", "strategy", "comparison",
+        "analysis", "trend", "audience", "segmentation", "targeting", "demographics",
+        "psychographics", "behavioral", "content", "SEO", "SEM", "PPC", "email",
+        "marketing", "automation", "CRM", "lead", "nurturing", "conversion",
+        "optimization", "funnel", "landing", "page", "CTA", "banner", "ads",
+        "display", "retargeting", "remarketing", "social", "media", "platform",
+        "influencer", "marketing", "KOL", "brand", "ambassador", "hashtag",
+        "campaign", "ROI", "KPI", "metrics", "analytics", "Google", "Analytics",
+        "Facebook", "Ads", "Instagram", "Twitter", "LinkedIn", "YouTube", "TikTok",
+        "Snapchat", "Pinterest", "Quora", "Reddit", "Forums", "blog", "podcast",
+        "webinar", "event", "offline", "activation", "loyalty", "program", "referral",
+        "program", "affiliate", "marketing", "partnership", "collaboration", "influencer",
+        "relationship", "management", "brand", "voice", "tone", "messaging", "storytelling",
+        "visual", "design", "UX", "UI", "wireframe", "prototype", "user", "testing",
+        "A/B", "testing", "heatmaps", "surveys", "feedback", "customer", "journey",
+        "customer", "experience", "CX", "voice", "of", "customer", "VOC", "net",
+        "promoter", "score", "NPS", "CSAT", "customer", "support", "customer", "service",
+        "customer", "success", "customer", "acquisition", "customer", "retention",
+        "churn", "rate", "lifetime", "value", "CLV", "cohort", "analysis", "funnel",
+        "analysis", "conversion", "rate", "bounce", "rate", "engagement", "rate",
+        "time", "on", "site", "page", "views", "unique", "visitors", "session", "duration"
+    ]
+    if not any(keyword in text.lower() for keyword in marketing_keywords):
+        return "Your question is not in the marketing area. Please upload another image or try again."
+
     if isinstance(images, Image.Image):
         images = [images]