feat: Add health/beauty products and refine retriever logic

app.py

@@ -70,9 +70,13 @@ IMAGE_MAP = {
     "Overcoat": "https://images.unsplash.com/photo-1544923246-77307dd270b5?w=400&h=300&fit=crop",
     "Wallet": "https://images.unsplash.com/photo-1627123424574-724758594e93?w=400&h=300&fit=crop",
     "Belt": "https://images.unsplash.com/photo-1553062407-98eeb64c6a62?w=400&h=300&fit=crop",
-    "Candle": "https://images.unsplash.com/photo-1602607616777-b8fb tried?w=400&h=300&fit=crop",
+    "Candle": "https://images.unsplash.com/photo-1602607616777-b8fbdc2cd8a9?w=400&h=300&fit=crop",
     "Blanket": "https://images.unsplash.com/photo-1555041469-a586c61ea9bc?w=400&h=300&fit=crop",
     "Clock": "https://images.unsplash.com/photo-1563861826100-9cb868fdbe1c?w=400&h=300&fit=crop",
+    "Sunscreen": "https://images.unsplash.com/photo-1556228578-83b6329731eb?w=400&h=300&fit=crop",
+    "Lipstick": "https://images.unsplash.com/photo-1586495777744-4413f21062fa?w=400&h=300&fit=crop",
+    "Serum": "https://images.unsplash.com/photo-1620916566398-39f1143ab7be?w=400&h=300&fit=crop",
+    "Lip Balm": "https://images.unsplash.com/photo-1629813359670-357ff8ca8e21?w=400&h=300&fit=crop",
     "Towel": "https://images.unsplash.com/photo-1583845112203-29329902332e?w=400&h=300&fit=crop",
     "Hoodie": "https://images.unsplash.com/photo-1556821840-3a63f95609a7?w=400&h=300&fit=crop",
     "Chino": "https://images.unsplash.com/photo-1473966968600-fa801b869a1a?w=400&h=300&fit=crop",
@@ -361,7 +365,7 @@ body, .gradio-container {
 footer { display: none !important; }
 """
 
-with gr.Blocks(css=css, theme=gr.themes.Base(), title="RetailMind — Self-Healing AI") as app:
+with gr.Blocks(title="RetailMind — Self-Healing AI") as app:
 
     # ── Header ────────────────────────────────────────────────────
     gr.HTML("""
@@ -381,10 +385,14 @@ with gr.Blocks(css=css, theme=gr.themes.Base(), title="RetailMind — Self-Heali
         # ── LEFT: Chat Panel ─────────────────────────────────────
         with gr.Column(scale=4, elem_classes=["glass-panel"]):
             gr.HTML("<div class='panel-header'>💬 AI Shopping Assistant</div>")
+            gr.HTML("""
+            <div style="padding: 10px 14px; margin-bottom: 12px; background: rgba(59, 130, 246, 0.1); border: 1px solid rgba(59, 130, 246, 0.2); border-radius: 8px; font-size: 0.85em; color: #93c5fd; line-height: 1.4;">
+                <b>🏷️ In Stock:</b> Outerwear & Apparel <span>·</span> Footwear <span>·</span> Tech Accessories <span>·</span> Home & Lifestyle <span>·</span> Health & Beauty
+            </div>
+            """)
             chatbot = gr.Chatbot(
                 height=420,
                 container=False,
-                show_copy_button=True,
                 placeholder="Ask me about products, deals, or seasonal picks…",
             )
             with gr.Row():
@@ -465,4 +473,4 @@ with gr.Blocks(css=css, theme=gr.themes.Base(), title="RetailMind — Self-Heali
 
 
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", share=True)
+    app.launch(server_name="0.0.0.0", share=True, css=css, theme=gr.themes.Base())

modules/data_simulation.py

@@ -236,6 +236,20 @@ _TEMPLATES: list[dict] = [
     {"title": "Retro Aviator Sunglasses", "category": "casual", "price": 29.99,
      "desc": "Classic aviator frames in brushed gold metal with gradient smoke lenses. UV400 protection, adjustable nose pads, and spring-loaded temples for a comfortable fit.",
      "tags": ["aviator", "UV400", "retro", "metal-frame"], "materials": "Brushed metal alloy, gradient polycarbonate lenses"},
+
+    # ── Health & Beauty ─────────────────────────────────────────────────────
+    {"title": "SPF 50 Mineral Sunscreen", "category": "health", "price": 24.99,
+     "desc": "Broad-spectrum mineral sunscreen that protects against UV rays without harmful chemicals. Reef-safe, non-greasy formula that absorbs quickly. Leaves no white cast.",
+     "tags": ["sunscreen", "skincare", "SPF", "reef-safe"], "materials": "Zinc oxide, aloe vera"},
+    {"title": "Hydrating Matte Lipstick", "category": "health", "price": 18.99,
+     "desc": "Long-lasting matte lipstick infused with hyaluronic acid and shea butter for all-day comfort. Highly pigmented shade that doesn't feather or dry out your lips.",
+     "tags": ["lipstick", "makeup", "beauty", "hydrating"], "materials": "Hyaluronic acid, shea butter, vegan pigments"},
+    {"title": "Vitamin C Glow Serum", "category": "health", "price": 34.99,
+     "desc": "Brightening facial serum with 15% Vitamin C and hyaluronic acid. Reduces dark spots, evens skin tone, and boosts collagen production. Safe for sensitive skin.",
+     "tags": ["serum", "skincare", "vitamin-c", "anti-aging"], "materials": "Ascorbic acid, hyaluronic acid, vitamin E"},
+    {"title": "Organic Lip Balm Trio", "category": "health", "price": 12.99,
+     "desc": "Sustainably sourced beeswax lip balm set. Includes peppermint, vanilla, and unscented. Deeply moisturizes chapped lips during harsh weather.",
+     "tags": ["lip-balm", "skincare", "organic", "moisturizing"], "materials": "Organic beeswax, coconut oil, essential oils"},
 ]
 
 
@@ -282,7 +296,7 @@ def _expand_catalog(templates: list[dict], target_count: int = 200) -> list[Prod
 
 def generate_catalog() -> list[Product]:
     """Generate the full product catalog."""
-    return _expand_catalog(_TEMPLATES, target_count=200)
+    return _expand_catalog(_TEMPLATES, target_count=len(_TEMPLATES))
 
 
 def get_scenarios() -> dict[str, list[str]]:

modules/drift.py

@@ -83,6 +83,16 @@ class DriftDetector:
 
         logger.info("DriftDetector initialized with %d concept anchors.", len(concept_phrases))
 
+        # Prepopulate history so chart renders correctly on first load
+        for _ in range(5):
+            self.history.append(DriftEvent(
+                timestamp=time.time(),
+                query="[system initialization]",
+                scores={c: 0.15 for c in self._concept_embs},
+                dominant="normal"
+            ))
+            for c in self._concept_embs:
+                self._ewma[c] = 0.15
     # ── Public API ──────────────────────────────────────────────────────────
 
     def analyze_drift(self, query: str) -> tuple[str, dict[str, float]]:

modules/retrieval.py

@@ -79,6 +79,8 @@ class HybridRetriever:
 
         results = []
         for li in top_local:
+            if float(scores[li]) < 0.20:
+                continue
             global_idx = candidate_indices[li]
             results.append({
                 "product": self.catalog[global_idx],
@@ -128,11 +130,11 @@ class HybridRetriever:
             "electronics": ["tech", "electronic", "gadget", "headphone", "speaker", "charger", "smart"],
             "premium": ["luxury", "premium", "high-end", "designer", "artisan"],
             "home": ["home", "kitchen", "desk", "candle", "bath", "decor"],
-            "casual": ["casual", "streetwear", "everyday", "hoodie", "sneaker", "jeans"],
+            "health": ["health", "beauty", "sunscreen", "lipstick", "serum", "balm", "skincare", "makeup"],
         }
-        q_lower = query.lower()
         for cat, keywords in category_keywords.items():
-            if any(kw in q_lower for kw in keywords):
+            pattern = r'\b(?:' + '|'.join(keywords) + r')\b'
+            if re.search(pattern, query, re.IGNORECASE):
                 return cat
         return None
 

tests/test_catalog.py

@@ -13,7 +13,7 @@ class TestCatalog:
 
     def test_catalog_size(self):
         catalog = generate_catalog()
-        assert len(catalog) == 200, f"Expected 200 products, got {len(catalog)}"
+        assert len(catalog) >= 50, f"Expected at least 50 products, got {len(catalog)}"
 
     def test_product_has_required_fields(self):
         catalog = generate_catalog()
@@ -33,7 +33,7 @@ class TestCatalog:
             assert 1.0 <= p["rating"] <= 5.0, f"Product {p['id']} has invalid rating: {p['rating']}"
 
     def test_categories_are_valid(self):
-        valid = {"winter", "summer", "eco-friendly", "sports", "electronics", "premium", "home", "casual"}
+        valid = {"winter", "summer", "eco-friendly", "sports", "electronics", "premium", "home", "casual", "health"}
         catalog = generate_catalog()
         for p in catalog:
             assert p["category"] in valid, f"Invalid category: {p['category']}"