added tabs to frontend by changing file to storefront_tabs_app.py

README.md

@@ -5,7 +5,7 @@ colorFrom: indigo
 colorTo: blue
 sdk: gradio
 sdk_version: "4.38.0"
-app_file: storefront_app.py
+app_file: storefront_tabs_app.py
 pinned: false
 license: mit
 short_description: Test for the project front-end.

storefront_tabs_app.py

@@ -0,0 +1,225 @@
+import os, json
+import gradio as gr
+from transformers import pipeline
+
+# ---------------- Model (same pipeline you used) ----------------
+MODEL_NAME = os.getenv("HF_MODEL_GENERATION", "distilgpt2")
+_pipe = None
+def _get_pipe():
+    global _pipe
+    if _pipe is None:
+        _pipe = pipeline("text-generation", model=MODEL_NAME)
+    return _pipe
+
+def model_generate(message, max_new_tokens=128, temperature=0.8, top_p=0.95):
+    out = _get_pipe()(
+        message,
+        max_new_tokens=int(max_new_tokens),
+        do_sample=True,
+        temperature=float(temperature),
+        top_p=float(top_p),
+        pad_token_id=50256,
+    )
+    return out[0]["generated_text"]
+
+
+# ---------------- Storefront knowledge (helper module preferred) ----------------
+STORE_DATA, USE_HELPERS = None, False
+try:
+    # If you keep these under agenticcore/
+    from agenticcore.storefront_rules import (
+        load_storefront, answer_faq, get_parking_rules, get_venue_rules, search_products
+    )
+    STORE_DATA = load_storefront()
+    USE_HELPERS = True
+except Exception:
+    # Fallback: look for JSON near this file or under agenticcore/
+    for p in [
+        os.path.join(os.path.dirname(__file__), "storefront_data.json"),
+        os.path.join(os.path.dirname(__file__), "agenticcore", "storefront_data.json"),
+    ]:
+        if os.path.exists(p):
+            with open(p, "r", encoding="utf-8") as f:
+                STORE_DATA = json.load(f)
+            break
+
+# Defaults if JSON is missing
+DEFAULT_PRODUCTS = [
+    {"SKU": "CG-SET", "Name": "Cap & Gown Set", "Price": 59.00, "Notes": "Tassel included; ship until 10 days before event"},
+    {"SKU": "PK-1",   "Name": "Parking Pass",   "Price": 10.00, "Notes": "Multiple passes allowed per student"},
+]
+DEFAULT_PARKING = ["No double parking.", "Vehicles parked in handicap will be towed."]
+DEFAULT_VENUE   = ["Formal attire recommended (not required).", "No muscle shirts.", "No sagging pants."]
+
+if STORE_DATA:
+    try:
+        DEFAULT_PRODUCTS = [{
+            "SKU": p.get("sku",""),
+            "Name": p.get("name",""),
+            "Price": p.get("price_usd",""),
+            "Notes": (p.get("description") or "")[:120],
+        } for p in STORE_DATA.get("products", [])]
+        DEFAULT_PARKING = STORE_DATA.get("policies", {}).get("parking_rules", DEFAULT_PARKING)
+        DEFAULT_VENUE   = STORE_DATA.get("policies", {}).get("venue_rules", DEFAULT_VENUE)
+    except Exception:
+        pass
+
+
+# ---------------- Routing: storefront first, then LLM ----------------
+def storefront_qna(text: str) -> str | None:
+    t = (text or "").lower().strip()
+    if not t:
+        return None
+
+    # Prefer your helper functions if available
+    if USE_HELPERS and STORE_DATA:
+        a = answer_faq(STORE_DATA, t)
+        if a:
+            return a
+        if "parking" in t and "rule" in t:
+            r = get_parking_rules(STORE_DATA)
+            if r:
+                return "Parking rules:\n- " + "\n- ".join(r)
+        if "venue" in t and "rule" in t or "attire" in t or "dress code" in t:
+            r = get_venue_rules(STORE_DATA)
+            if r:
+                return "Venue rules:\n- " + "\n- ".join(r)
+        # Extra: arrival timing for parking lots
+        if "parking" in t and ("hours" in t or "time" in t):
+            return "Parking lots open 2 hours before the ceremony."
+        hits = search_products(STORE_DATA, t)
+        if hits:
+            return "\n".join(
+                f"{p.get('name','Item')} — ${p.get('price_usd',0):.2f}: {p.get('description','')}"
+                for p in hits
+            )
+        return None
+
+    # Fallback rules (no helper module)
+    if "parking" in t and ("multiple" in t or "more than one" in t or "extra" in t):
+        return "Yes, multiple parking passes are allowed per student."
+    if "parking" in t and "rule" in t:
+        return "Parking rules:\n- " + "\n- ".join(DEFAULT_PARKING)
+    if "parking" in t and ("hours" in t or "time" in t):
+        return "Parking lots open 2 hours before the ceremony."
+    if "attire" in t or "dress code" in t or ("venue" in t and "rule" in t):
+        return "Venue rules:\n- " + "\n- ".join(DEFAULT_VENUE)
+    if "cap" in t or "gown" in t:
+        return "\n".join(f"{p['Name']} — ${p['Price']:.2f}: {p['Notes']}" for p in DEFAULT_PRODUCTS)
+    return None
+
+
+def chat_pipeline(message, max_new_tokens=128, temperature=0.8, top_p=0.95):
+    """Intercept storefront topics first; else generate with the model."""
+    sf = storefront_qna(message)
+    if sf:
+        return sf
+    return model_generate(message, max_new_tokens, temperature, top_p)
+
+
+# ---------------- Gradio UI (Tabs + Accordion) ----------------
+CSS = """
+:root { --bg:#0b0d12; --panel:#0f172a; --border:#1f2940; --text:#e5e7eb; --muted:#9ca3af; }
+.gradio-container { background: var(--bg) !important; color: var(--text) !important; }
+.panel { border:1px solid var(--border); border-radius:16px; background:var(--panel); }
+.small { font-size:12px; color: var(--muted); }
+"""
+
+with gr.Blocks(title="Storefront Chat", css=CSS) as demo:
+    gr.Markdown("## Storefront Chat")
+
+    with gr.Tabs():
+        # --- TAB 1: Chat (sliders tucked away) ---
+        with gr.TabItem("Chat"):
+            with gr.Group(elem_classes=["panel"]):
+                chat = gr.Chatbot(height=360, bubble_full_width=False, label="Chat")
+                with gr.Row():
+                    msg = gr.Textbox(placeholder="Ask about parking rules, attire, cap & gown, pickup times…", scale=5)
+                    send = gr.Button("Send", scale=1)
+
+                # Quick chips
+                with gr.Row():
+                    chip1 = gr.Button("Parking rules", variant="secondary")
+                    chip2 = gr.Button("Multiple passes", variant="secondary")
+                    chip3 = gr.Button("Attire", variant="secondary")
+                    chip4 = gr.Button("When do lots open?", variant="secondary")
+
+                # Advanced chat options in an Accordion
+                with gr.Accordion("Advanced chat options", open=False):
+                    max_new = gr.Slider(32, 512, 128, 1, label="Max new tokens")
+                    temp    = gr.Slider(0.1, 1.5, 0.8, 0.05, label="Temperature")
+                    topp    = gr.Slider(0.1, 1.0, 0.95, 0.05, label="Top-p")
+
+                # Simple health & capabilities as small buttons
+                with gr.Row():
+                    health_btn = gr.Button("Health", variant="secondary")
+                    caps_btn   = gr.Button("Capabilities", variant="secondary")
+                    status_md  = gr.Markdown("Status: not checked", elem_classes=["small"])
+
+        # --- TAB 2: Products ---
+        with gr.TabItem("Products"):
+            gr.Markdown("### Available Items")
+            cols = list(DEFAULT_PRODUCTS[0].keys()) if DEFAULT_PRODUCTS else ["SKU","Name","Price","Notes"]
+            data = [[p.get(c,"") for c in cols] for p in DEFAULT_PRODUCTS]
+            products_tbl = gr.Dataframe(headers=cols, value=data, interactive=False, wrap=True, label="Products")
+
+        # --- TAB 3: Rules ---
+        with gr.TabItem("Rules"):
+            gr.Markdown("### Venue rules")
+            gr.Markdown("- " + "\n- ".join(DEFAULT_VENUE))
+            gr.Markdown("### Parking rules")
+            gr.Markdown("- " + "\n- ".join(DEFAULT_PARKING))
+
+        # --- TAB 4: Logistics ---
+        with gr.TabItem("Logistics"):
+            gr.Markdown(
+                "### Event Logistics\n"
+                "- Shipping available until 10 days before event (typ. 3–5 business days)\n"
+                "- Pickup: Student Center Bookstore during the week prior to event\n"
+                "- Graduates arrive 90 minutes early; guests 60 minutes early\n"
+                "- Lots A & B open 2 hours before; overflow as needed\n"
+                "\n*Try asking the bot:* “What time should I arrive?” • “Where do I pick up the gown?”"
+            )
+
+    # ---- Wiring ----
+    def on_send(history, message, max_new_tokens, temperature, top_p):
+        t = (message or "").strip()
+        if not t:
+            return history, ""
+        history = history + [[t, None]]
+        reply = chat_pipeline(t, max_new_tokens, temperature, top_p)
+        history[-1][1] = reply
+        return history, ""
+
+    send.click(on_send, [chat, msg, max_new, temp, topp], [chat, msg])
+    msg.submit(on_send, [chat, msg, max_new, temp, topp], [chat, msg])
+
+    # Quick chips → prefill textbox
+    chip1.click(lambda: "What are the parking rules?", outputs=msg)
+    chip2.click(lambda: "Can I buy multiple parking passes?", outputs=msg)
+    chip3.click(lambda: "Is formal attire required?", outputs=msg)
+    chip4.click(lambda: "What time do the parking lots open?", outputs=msg)
+
+    # Health / capabilities
+    def ui_health_check():
+        return (f"### Status: ✅ Healthy\n"
+                f"- Model: `{MODEL_NAME}`\n"
+                f"- Storefront module: {'yes' if USE_HELPERS else 'no'}\n"
+                f"- Storefront JSON: {'loaded' if bool(STORE_DATA) else 'not found'}")
+
+    def ui_capabilities():
+        caps = [
+            "Chat (LLM text-generation)",
+            "Storefront Q&A (parking, attire, products, logistics)",
+            "Adjustable: max_new_tokens, temperature, top-p",
+        ]
+        return "### Capabilities\n- " + "\n- ".join(caps)
+
+    def _health():
+        return ui_health_check(), "Status: ✅ Healthy"
+
+    health_btn.click(_health, outputs=[chat, status_md])
+    caps_btn.click(ui_capabilities, outputs=chat)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")))