Update app.py

app.py

@@ -28,33 +28,6 @@ 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]
 
@@ -98,9 +71,43 @@ def model_inference(
     generated_texts = processor.batch_decode(generated_ids[:, generation_args["input_ids"].size(1):], skip_special_tokens=True)
     return generated_texts[0]
 
-with gr.Blocks(fill_height=True) as demo:
-    gr.Markdown("## Marketing Vision App 📈")
-    gr.Markdown("This app uses the [HuggingFaceM4/Idefics3-8B-Llama3](https://huggingface.co/HuggingFaceM4/Idefics3-8B-Llama3) model to answer questions related to various topics. Upload an image and a text query.")
+css = """
+#app-container {
+    background-color: #f8f9fa;
+    padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+}
+
+#app-header {
+    text-align: center;
+    margin-bottom: 20px;
+}
+
+#app-description {
+    text-align: center;
+    margin-bottom: 20px;
+}
+
+#app-footer {
+    text-align: center;
+    margin-top: 20px;
+    font-size: 0.9em;
+}
+
+#app-footer a {
+    color: #007bff;
+    text-decoration: none;
+}
+
+#app-footer a:hover {
+    text-decoration: underline;
+}
+"""
+
+with gr.Blocks(css=css, theme='JohnSmith9982/small_and_pretty') as demo:
+    gr.HTML("<h1 id='app-header'>Marketing Vision App 📈</h1>")
+    gr.HTML("<p id='app-description'>This app uses the <a href='https://huggingface.co/HuggingFaceM4/Idefics3-8B-Llama3' target='_blank'>HuggingFaceM4/Idefics3-8B-Llama3</a> model to answer questions related to various topics. Upload an image and a text query.</p>")
     gr.Markdown("**Disclaimer:** This app may not consistently follow prompts or handle complex tasks. However, adding a prefix to the assistant's response can significantly improve the output. You could also play with the parameters such as the temperature in non-greedy mode.")
     with gr.Column():
         image_input = gr.Image(label="Upload your Image", type="pil", scale=1)
@@ -185,10 +192,10 @@ with gr.Blocks(fill_height=True) as demo:
                                                       max_new_tokens, repetition_penalty, top_p], outputs=output)
 
     footer = """
-    <div style="text-align: center; margin-top: 20px;">
-        <a href="https://www.linkedin.com/in/pejman-ebrahimi-4a60151a7/" target="_blank">LinkedIn</a> |
-        <a href="https://github.com/arad1367" target="_blank">GitHub</a> |
-        <a href="https://arad1367.pythonanywhere.com/" target="_blank">Live demo of my PhD defense</a>
+    <div id='app-footer'>
+        <a href='https://www.linkedin.com/in/pejman-ebrahimi-4a60151a7/' target='_blank'>LinkedIn</a> |
+        <a href='https://github.com/arad1367' target='_blank'>GitHub</a> |
+        <a href='https://arad1367.pythonanywhere.com/' target='_blank'>Live demo of my PhD defense</a>
         <br>
         Made with 💖 by Pejman Ebrahimi
     </div>