Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,312 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import re
+import random
+import string
+from datetime import datetime, timedelta
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# ---------------------------------------------------------
+# Simple Pizza Support Chatbot (Streamlit)
+# - Rule-based NLU (keyword & pattern matching)
+# - Menu lookup, offers, order placement, order tracking, FAQs
+# - No external APIs, perfect for a live workshop demo
+# ---------------------------------------------------------
+
+st.set_page_config(page_title="Pizza Order Chatbot Demo", page_icon="🍕", layout="centered")
+
+# ----------------------------
+# Demo Data
+# ----------------------------
+MENU = {
+    "Classic": [
+        {"name": "Margherita", "desc": "Cheese, tomato, basil", "prices": {"small": 199, "medium": 299, "large": 399}},
+        {"name": "Farmhouse", "desc": "Onion, capsicum, corn, olives", "prices": {"small": 249, "medium": 349, "large": 449}},
+    ],
+    "Meat": [
+        {"name": "Pepperoni", "desc": "Pepperoni & extra cheese", "prices": {"small": 279, "medium": 379, "large": 479}},
+        {"name": "BBQ Chicken", "desc": "Smoky chicken, onion, BBQ sauce", "prices": {"small": 299, "medium": 399, "large": 499}},
+    ],
+    "Special": [
+        {"name": "Paneer Tikka", "desc": "Spiced paneer, onion, capsicum", "prices": {"small": 269, "medium": 369, "large": 469}},
+        {"name": "Veggie Overload", "desc": "Mushroom, corn, olives, jalapeño", "prices": {"small": 289, "medium": 389, "large": 489}},
+    ],
+}
+
+OFFERS = [
+    "Use code RIDHI to get 5% off on all orders!",
+    "Buy 2 Large pizzas and get 1 Garlic Bread free.",
+    "Free delivery for orders above ₹599.",
+]
+
+FAQ = {
+    "timings": "We deliver daily from 10 AM to 11 PM.",
+    "payment": "We accept UPI, cards, netbanking, and cash on delivery.",
+    "delivery time": "Average delivery time is 30-40 minutes depending on your location.",
+    "refund": "If there is an issue, please contact support and refunds will be processed within 3-5 working days.",
+}
+
+SIZES = ["small", "medium", "large"]
+
+# ----------------------------
+# Utilities
+# ----------------------------
+
+def _rand_order_id(prefix="OD"):
+    return prefix + "-" + ''.join(random.choices(string.ascii_uppercase + string.digits, k=6))
+
+
+def format_menu() -> str:
+    lines = ["Here is our pizza menu:\n"]
+    for category, items in MENU.items():
+        lines.append(f"**{category}**")
+        for it in items:
+            price_line = ", ".join([f"{sz.title()}: ₹{amt}" for sz, amt in it["prices"].items()])
+            lines.append(f"- **{it['name']}** – {it['desc']} ({price_line})")
+        lines.append("")
+    return "\n".join(lines)
+
+
+def list_offers() -> str:
+    return "\n".join([f"- {o}" for o in OFFERS])
+
+
+def menu_lookup(pizza_name: str):
+    pizza_name_l = pizza_name.lower()
+    for _, items in MENU.items():
+        for it in items:
+            if it["name"].lower() == pizza_name_l:
+                return it
+    # fuzzy contains
+    for _, items in MENU.items():
+        for it in items:
+            if pizza_name_l in it["name"].lower():
+                return it
+    return None
+
+
+def extract_size(text: str):
+    for s in SIZES:
+        if re.search(rf"\b{s}\b", text, flags=re.I):
+            return s
+    return None
+
+
+def extract_pizza(text: str):
+    # search exact names first
+    for _, items in MENU.items():
+        for it in items:
+            if re.search(rf"\b{re.escape(it['name'])}\b", text, flags=re.I):
+                return it['name']
+    # fallback: partial tokens
+    tokens = re.findall(r"[A-Za-z]+", text)
+    for t in tokens:
+        p = menu_lookup(t)
+        if p:
+            return p['name']
+    return None
+
+
+def estimate_eta(minutes=35):
+    return (datetime.now() + timedelta(minutes=minutes)).strftime("%I:%M %p")
+
+# ----------------------------
+# Intent Detection (very simple & explainable for workshop)
+# ----------------------------
+
+INTENT_PATTERNS = {
+    "greet": [r"\bhi\b", r"\bhello\b", r"\bhey\b"],
+    "menu": [r"menu", r"what.*pizzas", r"list.*pizza", r"show.*pizza"],
+    "offers": [r"offer", r"discount", r"coupon", r"promo"],
+    "order": [r"order", r"i want", r"i'd like", r"buy", r"add to cart"],
+    "track": [r"track", r"where.*order", r"status", r"deliver"],
+    "help": [r"help", r"support", r"issue"],
+    "thanks": [r"thanks", r"thank you"],
+}
+
+
+def detect_intent(text: str) -> str:
+    t = text.lower().strip()
+    for intent, patterns in INTENT_PATTERNS.items():
+        for pat in patterns:
+            if re.search(pat, t):
+                return intent
+    # FAQs
+    for key in FAQ.keys():
+        if key in t:
+            return "faq"
+    return "unknown"
+
+# ----------------------------
+# Order & State Management
+# ----------------------------
+
+if "chat" not in st.session_state:
+    st.session_state.chat = []  # list of dicts: {role, content}
+if "orders" not in st.session_state:
+    st.session_state.orders = {}  # order_id -> {pizza, size, price, status, eta}
+if "last_order_id" not in st.session_state:
+    st.session_state.last_order_id = None
+
+
+def add_bot(msg: str):
+    st.session_state.chat.append({"role": "assistant", "content": msg})
+
+
+def add_user(msg: str):
+    st.session_state.chat.append({"role": "user", "content": msg})
+
+
+def place_order(pizza_name: str, size: str):
+    item = menu_lookup(pizza_name)
+    if not item:
+        return None, "Sorry, I couldn't find that pizza. Use 'menu' to see options."
+    if size not in item["prices"]:
+        return None, f"Please choose a valid size for {item['name']}: small / medium / large."
+
+    order_id = _rand_order_id()
+    price = item["prices"][size]
+    eta = estimate_eta(random.choice([25, 30, 35, 40]))
+    st.session_state.orders[order_id] = {
+        "pizza": item["name"],
+        "size": size,
+        "price": price,
+        "status": "Confirmed",
+        "eta": eta,
+    }
+    st.session_state.last_order_id = order_id
+    return order_id, (
+        f"Order placed!\n\n"
+        f"**Order ID:** {order_id}\n"
+        f"**Item:** {item['name']} ({size.title()})\n"
+        f"**Amount:** ₹{price}\n"
+        f"**Estimated Delivery:** {eta}\n\n"
+        f"Use `track {order_id}` anytime to check status."
+    )
+
+
+def track_order(order_id: str):
+    data = st.session_state.orders.get(order_id)
+    if not data:
+        return "I couldn't find that order ID. Please check and try again."
+
+    # Simulate simple status transitions for demo
+    transitions = ["Confirmed", "Being Prepared", "Out for Delivery", "Delivered"]
+    cur_status = data["status"]
+    try:
+        idx = transitions.index(cur_status)
+        if idx < len(transitions) - 1 and random.random() < 0.5:
+            data["status"] = transitions[idx + 1]
+            st.session_state.orders[order_id] = data
+    except ValueError:
+        pass
+
+    return (
+        f"**Order ID:** {order_id}\n"
+        f"**Item:** {data['pizza']} ({data['size'].title()})\n"
+        f"**Status:** {data['status']}\n"
+        f"**ETA:** {data['eta']}"
+    )
+
+# ----------------------------
+# Bot Brain
+# ----------------------------
+
+def bot_reply(user_text: str) -> str:
+    intent = detect_intent(user_text)
+
+    if intent == "greet":
+        return (
+            "Hello! I am your Pizza Order Assistant. You can say things like:\n"
+            "- 'menu' to see pizzas\n- 'any offers?'\n- 'order a medium Margherita'\n- 'track OD-ABC123'"
+        )
+
+    if intent == "menu":
+        return format_menu()
+
+    if intent == "offers":
+        return "Here are the current offers:\n\n" + list_offers()
+
+    if intent == "faq":
+        # find which FAQ
+        t = user_text.lower()
+        for k, v in FAQ.items():
+            if k in t:
+                return f"**{k.title()}**: {v}"
+        return "You can ask about timings, payment, delivery time, or refund."
+
+    if intent == "order":
+        pizza = extract_pizza(user_text) or "Margherita"
+        size = extract_size(user_text) or "medium"
+        order_id, msg = place_order(pizza, size)
+        return msg
+
+    if intent == "track":
+        # allow 'track <order_id>' or fallback to last order
+        m = re.search(r"(od-[a-z0-9]{6})", user_text.lower())
+        if m:
+            oid = m.group(1).upper()
+        else:
+            oid = st.session_state.last_order_id
+        if not oid:
+            return "Please provide an order ID, e.g., 'track OD-ABC123'. If you just ordered, try again in a moment."
+        return track_order(oid)
+
+    if intent == "help":
+        return (
+            "I can help you with:\n"
+            "- Viewing the menu (type 'menu')\n"
+            "- Placing an order (e.g., 'order a large Pepperoni')\n"
+            "- Tracking orders (e.g., 'track OD-XXXXXX')\n"
+            "- Checking offers (type 'offers')\n"
+            "- FAQs: timings, payment, delivery time, refund"
+        )
+
+    if intent == "thanks":
+        return "You're welcome! Anything else I can do for you?"
+
+    # Unknown → assistively guide
+    return (
+        "I didn't quite get that. Try:\n"
+        "- 'menu' to see pizzas\n- 'offers' to see discounts\n- 'order a medium Margherita'\n- 'track OD-XXXXXX' to track your order"
+    )
+
+# ----------------------------
+# UI
+# ----------------------------
+with st.sidebar:
+    st.title("🍕 Pizza Bot – Demo")
+    st.caption("A simple, explainable chatbot for workshops")
+
+    if st.button("Reset Conversation"):
+        st.session_state.chat = []
+        st.session_state.last_order_id = None
+        st.experimental_rerun()
+
+    st.subheader("Quick Actions")
+    if st.button("Show Menu"):
+        add_user("menu")
+        add_bot(format_menu())
+    if st.button("View Offers"):
+        add_user("offers")
+        add_bot("Here are the current offers:\n\n" + list_offers())
+
+st.markdown("""
+# 🍕 Pizza Order Chatbot (Workshop Demo)
+Welcome! Type a message below. Try: **menu**, **offers**, **order a medium Margherita**, **track OD-XXXXXX**
+""")
+
+for msg in st.session_state.chat:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"]) 
+
+user_text = st.chat_input("Type here…")
+if user_text:
+    add_user(user_text)
+    reply = bot_reply(user_text)
+    add_bot(reply)
+    with st.chat_message("assistant"):
+        st.markdown(reply)
+
+# Footer note for presenter
+st.caption(
+    "Presenter tip: Explain how intents are detected via regex and how state (orders, last order) is kept in session_state."
+)