Update app.py

app.py

@@ -1,314 +1,292 @@
-# app.py — WaterWise Home (uses /tmp cache; loads dataset from HF; returns top-3)
+# app.py — WaterWise Home (stable cache + robust handlers + light/dark-safe UI)
 
 import os
-import json
-import random
-from datetime import datetime
-
-import pandas as pd
+from pathlib import Path
+import csv
+import re
+import numpy as np
 import gradio as gr
-from sentence_transformers import SentenceTransformer, util
-
-# ======= FIX: put HF caches in a writable dir (/tmp) =======
-CACHE_DIR = "/tmp/hf_cache"
-os.environ["HOME"] = "/tmp"
-os.environ["HF_HOME"] = CACHE_DIR
-os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
-os.environ["HUGGINGFACE_HUB_CACHE"] = CACHE_DIR
-os.environ["SENTENCE_TRANSFORMERS_HOME"] = CACHE_DIR
-os.makedirs(CACHE_DIR, exist_ok=True)
-# ===========================================================
-
-# =======================
-# Config & Theme
-# =======================
-DATA_URL = "https://huggingface.co/datasets/Liat2025/waterwise_tips_2025/resolve/main/tips_dataset.csv"
-HISTORY_FILE = "history.json"
-
-NAVY        = "#0f2c56"
-LIGHT_BLUE  = "#e6f2ff"
-BUTTON_BLUE = "#287bff"
-
-QUOTES = [
-    "Every drop counts. 🌍💧",
-    "Small changes make a big wave. 🌊",
-    "Water saved today is life saved tomorrow. 🌱",
-    "The future flows from the choices you make. 💦",
-    "Your efforts ripple into something bigger. 🌊",
-    "Conserve water, preserve life. 🌍",
-    "Drops become streams, streams become rivers. 🌊",
-    "Saving water saves the planet. 🌱",
-]
-
-SEASONS = ["❄️ Winter", "🌸 Spring", "☀️ Summer", "🍂 Fall"]
-
-SEASON_CONTEXT = {
-    "❄️ Winter": "indoor efficiency, pipe protection, heater settings, hot water, low outdoor watering",
-    "🌸 Spring": "garden prep, soil moisture, planting, rain barrels, irrigation tune-up",
-    "☀️ Summer": "hot weather, evaporation, early morning watering, mulch, lawn care, outdoor hoses",
-    "🍂 Fall":   "leaf cleanup, aeration, winterizing irrigation, reduce watering frequency, shutoff valves",
-}
+
+# -----------------------
+# HF cache: FIX PERMISSIONS (must be before importing HF libs)
+# -----------------------
+os.environ["HF_HOME"] = "/home/user/.cache/huggingface"
+os.environ["HF_HUB_CACHE"] = "/home/user/.cache/huggingface/hub"
+os.environ["TRANSFORMERS_CACHE"] = "/home/user/.cache/huggingface/transformers"
+Path("/home/user/.cache/huggingface").mkdir(parents=True, exist_ok=True)
+
+# -----------------------
+# Load tips CSV (strict)
+# -----------------------
+CSV_PATH = "tips_dataset.csv"
+if not Path(CSV_PATH).exists():
+    raise FileNotFoundError(
+        f"Missing {CSV_PATH}. Upload it next to app.py with the first line exactly 'tip'."
+    )
+
+tips: list[str] = []
+with open(CSV_PATH, "r", encoding="utf-8") as f:
+    reader = csv.reader(f)
+    rows = list(reader)
+
+if not rows:
+    raise ValueError("tips_dataset.csv is empty.")
+header = rows[0]
+if len(header) != 1 or header[0].strip().lower() != "tip":
+    raise ValueError("First line of tips_dataset.csv must be exactly one column named 'tip'.")
+
+for r in rows[1:]:
+    if not r:
+        continue
+    t = (r[0] or "").strip()
+    if t:
+        tips.append(t)
+
+if len(tips) != 1000:
+    raise ValueError(f"Expected exactly 1000 tips, found {len(tips)}. Check tips_dataset.csv formatting.")
+
+# -----------------------
+# Embedding model (robust)
+# -----------------------
+from sentence_transformers import SentenceTransformer  # noqa: E402
+
+def load_embedder():
+    names = [
+        "sentence-transformers/all-MiniLM-L6-v2",  # canonical repo
+        "all-MiniLM-L6-v2",                        # alias fallback
+    ]
+    last_err = None
+    for name in names:
+        try:
+            return SentenceTransformer(name, cache_folder="models")
+        except Exception as e:
+            last_err = e
+    raise RuntimeError(f"Failed to load embedder. Last error: {last_err}")
+
+model = load_embedder()
+# Warmup so the first user click doesn't trigger a delayed download
+_ = model.encode(["warmup"], convert_to_tensor=False)
+
+# -----------------------
+# Precompute tip embeddings
+# -----------------------
+def _normalize(m: np.ndarray) -> np.ndarray:
+    norms = np.linalg.norm(m, axis=1, keepdims=True) + 1e-12
+    return m / norms
+
+tip_embs = model.encode(tips, convert_to_numpy=True, show_progress_bar=False)
+tip_embs = _normalize(tip_embs.astype(np.float32))
+
+# -----------------------
+# Recommender
+# -----------------------
 SEASON_KEYWORDS = {
-    "❄️ Winter": ["heater", "insulate", "pipe", "hot water", "freeze", "indoors"],
-    "🌸 Spring": ["plant", "soil", "rain barrel", "sprinkler", "garden prep", "seedling"],
-    "☀️ Summer": ["mulch", "evaporation", "morning", "hose", "sprinkler", "lawn", "shade"],
-    "🍂 Fall":   ["winterize", "shutoff", "aerate", "leaf", "drain", "blowout"],
+    "Spring": ["spring", "rain", "garden", "flowers", "bloom"],
+    "Summer": ["summer", "heat", "hot", "sprinkler", "pool", "evapor"],
+    "Fall":   ["fall", "autumn", "leaves", "harvest"],
+    "Winter": ["winter", "cold", "freeze", "heater", "boiler", "pipe"],
 }
-SEASON_BONUS = 0.08
-
-# =======================
-# Data Loading (from HF Dataset)
-# =======================
-def load_tips():
-    # Robust CSV read: take FIRST column only, skip malformed rows
-    df = pd.read_csv(
-        DATA_URL,
-        engine="python",
-        usecols=[0],        # only the first column
-        names=["tip"],      # call that column "tip"
-        header=0,           # first row is header
-        on_bad_lines="skip" # skip problematic lines
+
+def recommend(user_text: str, season: str):
+    user_text = (user_text or "").strip()
+    if not user_text:
+        return (
+            "Tell me one thing you did with water today (e.g., 'long shower', 'ran dishwasher twice'). "
+            "Choose a season, then hit the button 🙂",
+            "",
+        )
+
+    # Embed query
+    q = model.encode([user_text], convert_to_numpy=True).astype(np.float32)
+    q = q / (np.linalg.norm(q, axis=1, keepdims=True) + 1e-12)
+
+    # Cosine similarity
+    sims = (tip_embs @ q.T).squeeze(1)
+
+    # Gentle seasonal keyword boost
+    kws = SEASON_KEYWORDS.get(season, [])
+    if kws:
+        boost = np.zeros_like(sims, dtype=np.float32)
+        for i, tip in enumerate(tips):
+            tl = tip.lower()
+            if any(k in tl for k in kws):
+                boost[i] = 0.03
+        sims = sims + boost
+
+    top_idx = np.argsort(-sims)[:3].tolist()
+    top_tips = [f"• {tips[i]}" for i in top_idx]
+
+    summary = (
+        f"Based on: **“{user_text}”** · Season: **{season}**\n\n"
+        "Here are 3 smart, doable tips to try next:"
     )
-    tips = df["tip"].astype(str).str.strip()
-    tips = tips[tips.str.len() > 0].drop_duplicates().tolist()
-    if len(tips) < 10:
-        raise ValueError(f"Need at least ~10 tips to run; found {len(tips)}.")
-    return tips
-
-TIPS = load_tips()
-
-# =======================
-# Embeddings
-# =======================
-MODEL = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
-TIP_EMB = MODEL.encode(
-    TIPS, convert_to_tensor=True, normalize_embeddings=True, show_progress_bar=False
-)
-
-# =======================
-# Helpers
-# =======================
-def estimate_water_saved(tip_text: str) -> int:
-    s = tip_text.lower()
-    if "shower" in s: return 9
-    if "dishwasher" in s: return 6
-    if "laundry" in s: return 7
-    if "garden" in s or "watering" in s: return 4
-    if "tap" in s or "faucet" in s: return 3
-    return 2
-
-def encourage():
-    return random.choice([
-        "Nice! Small changes add up fast 💧",
-        "Great step — keep going! 🌱",
-        "You’re building better habits already 🙌",
-        "Love the effort — consistency wins! ⭐",
-    ])
-
-def load_history():
-    if not os.path.exists(HISTORY_FILE):
-        return [], 0
-    try:
-        with open(HISTORY_FILE, "r", encoding="utf-8") as f:
-            raw = f.read().strip()
-        if not raw:
-            return [], 0
-        data = json.loads(raw)
-        if not isinstance(data, list):
-            return [], 0
-    except Exception:
-        return [], 0
-    total = sum(int(item.get("liters_saved", 0)) for item in data)
-    return data, total
-
-def save_history(data):
-    with open(HISTORY_FILE, "w", encoding="utf-8") as f:
-        json.dump(data, f, indent=2, ensure_ascii=False)
-
-def _plain_season(s):
-    return (s or "").split()[-1] if s else ""
-
-def total_saved_card_html(total_liters: int):
-    showers = total_liters / 9
-    dishwashers = total_liters / 6
-    watering_cans = total_liters / 4
-    quote = random.choice(QUOTES)
-    return f"""
-    <div id='stat-card'>
-      <div class='stat-number'>💧 {total_liters} liters saved</div>
-      <div class='stat-quote'>{quote}</div>
-      <div class='stat-impact'>≈ {showers:.1f} showers • {dishwashers:.1f} dishwasher loads • {watering_cans:.1f} watering cans</div>
-    </div>
+    return summary, "\n".join(top_tips)
+
+# -----------------------
+# Savings heuristics (safe)
+# -----------------------
+def parse_minutes(text: str, default=10):
+    if not text:
+        return default
+    m = re.search(r'(\d+)\s*(min|minute|minutes)\b', text.lower())
+    return int(m.group(1)) if m else default
+
+def estimate_saved_liters(text: str, season: str) -> float:
+    """
+    Tiny heuristic. NEVER raises. Adjust numbers as you like.
     """
+    t = (text or "").lower()
+    mins = parse_minutes(t, default=10)
+    saved = 0.0
+    if "shower" in t:
+        # suggest -2 minutes ~ 6 L/min => ~12 L saved
+        saved += 12.0
+    if "dishwasher" in t:
+        # suggest full loads
+        saved += 6.0
+    if "garden" in t or "watering" in t or "lawn" in t:
+        saved += 8.0 if season == "Summer" else 5.0
+    return float(max(0.0, saved))
+
+# -----------------------
+# Combined submit handler (robust)
+# -----------------------
+def on_submit(user_text: str, season: str, state: dict | None):
+    try:
+        # State default
+        if not isinstance(state, dict):
+            state = {"saved_liters": 0.0, "history": []}
+
+        # Recs
+        summary_md, tips_md = recommend(user_text, season)
+
+        # Savings
+        delta = estimate_saved_liters(user_text, season)
+        state["saved_liters"] = float(state.get("saved_liters", 0.0)) + float(delta)
+        state["history"].append({"text": user_text, "season": season, "saved": float(delta)})
+
+        saved_headline = f"💧 {state['saved_liters']:.1f} liters saved"
+        saved_detail = (
+            "Drops become streams, streams become rivers. 🌊\n\n"
+            f"≈ {state['saved_liters']/12.0:.1f} showers · "
+            f"{state['saved_liters']/6.0:.1f} dishwasher loads · "
+            f"{state['saved_liters']/10.0:.1f} watering cans"
+        )
 
-# =======================
-# Core: recommend & persist (season-aware)
-# =======================
-def recommend(activity, season):
-    if not activity or not activity.strip():
-        _, total = load_history()
-        return "_Please describe a water-using activity first._", "", total_saved_card_html(total)
-
-    ctx = SEASON_CONTEXT.get(season or "", "")
-    query = activity.strip()
-    if season:
-        query += f" — season: {season}. Focus on: {ctx}"
-
-    q_emb = MODEL.encode([query], convert_to_tensor=True, normalize_embeddings=True)
-    sims = util.cos_sim(q_emb, TIP_EMB)[0].detach().cpu().tolist()
-
-    if season:
-        keys = SEASON_KEYWORDS.get(season, [])
-        boosted = []
-        for i, base in enumerate(sims):
-            tip_lower = TIPS[i].lower()
-            bonus = SEASON_BONUS if any(k.lower() in tip_lower for k in keys) else 0.0
-            boosted.append((i, base + bonus))
-        boosted.sort(key=lambda x: x[1], reverse=True)
-        top_idx = [i for i, _ in boosted[:3]]
-    else:
-        top_idx = sorted(range(len(sims)), key=lambda i: sims[i], reverse=True)[:3]
-
-    picked = [TIPS[i] for i in top_idx]
-
-    if season:
-        keys_l = [k.lower() for k in SEASON_KEYWORDS.get(season, [])]
-        badged = []
-        for i, t in enumerate(picked):
-            mark = " _(great for " + season.split()[-1] + ")_" if any(k in t.lower() for k in keys_l) else ""
-            badged.append(f"**{i+1}.** {t}{mark}")
-        tips_md = "\n\n".join(badged)
-    else:
-        tips_md = "\n\n".join([f"**{i+1}.** {t}" for i, t in enumerate(picked)])
-
-    saving = sum(estimate_water_saved(t) for t in picked)
-
-    history, _ = load_history()
-    history.append({
-        "datetime": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        "activity": activity.strip(),
-        "season": _plain_season(season),
-        "tips": picked,
-        "liters_saved": saving,
-    })
-    save_history(history)
-    _, total = load_history()
-
-    return tips_md, encourage(), total_saved_card_html(total)
-
-def clear_all():
-    save_history([])
-    return "", "", total_saved_card_html(0)
-
-# =======================
-# UI (with CSS)
-# =======================
-CUSTOM_CSS = f"""
-:root {{ color-scheme: light !important; }}
-body, .gradio-container{{
-  background-color: {NAVY} !important;
-  font-family: Inter, ui-sans-serif, system-ui, -apple-system, Segoe UI, Roboto, Arial, Noto Sans;
-  color: #e6f2ff;
-}}
-#header-box{{
-  background: linear-gradient(135deg, {LIGHT_BLUE}, #ffffff);
-  padding: 16px; border-radius: 16px; text-align:center;
-  box-shadow: 0 6px 14px rgba(0,0,0,0.12); margin: 12px auto 10px;
-}}
-#header-box h1{{ color: {NAVY}; font-size: 2rem; font-weight: 800; margin:0 0 6px; }}
-#header-box p{{  color: {NAVY}; margin:0; }}
-.label, .gr-markdown, .markdown-body, .gr-html{{ color:#e6f2ff !important; }}
-textarea, input{{
-  background:#ffffff !important; color:#0b2540 !important;
-  border:2px solid {BUTTON_BLUE} !important; border-radius:10px !important;
-}}
-button{{
-  background:{BUTTON_BLUE} !important; color:#ffffff !important;
-  font-weight:700 !important; border-radius:10px !important; border:none !important;
-  box-shadow:0 4px 10px rgba(0,0,0,.18);
-}}
-#season_pills .gr-radio .wrap {{ display:flex; flex-wrap:wrap; gap:12px; }}
-#season_pills input[type="radio"] {{ position:absolute; opacity:0; width:0; height:0; }}
-#season_pills .gr-radio .wrap > input[type="radio"] + label {{
-  display:inline-block; padding:14px 22px; border-radius:14px;
-  background:#1c3e76; color:#ffffff; border:2px solid transparent;
-  cursor:pointer; font-weight:800; letter-spacing:.2px;
-  box-shadow:0 3px 10px rgba(0,0,0,.15);
-  transition:transform .04s ease, background .15s ease, box-shadow .15s ease;
-}}
-#season_pills .gr-radio .wrap > input[type="radio"] + label:hover {{
-  transform: translateY(-1px); box-shadow:0 6px 16px rgba(0,0,0,.22);
-}}
-#season_pills .gr-radio .wrap > input[type="radio"]:checked + label {{
-  background:{BUTTON_BLUE}; border-color:#ffffff; color:#ffffff;
-  box-shadow:0 10px 22px rgba(40,123,255,.40); transform: translateY(-1px);
-}}
-#season_pills .gr-radio .wrap > input[type="radio"]:focus + label {{
-  outline:3px solid #a8c7ff; outline-offset:2px;
-}}
-#stat-card{{ background:linear-gradient(135deg, {LIGHT_BLUE}, #ffffff); border-radius:12px; padding:14px; text-align:center; box-shadow:0 6px 14px rgba(0,0,0,.12); }}
-#stat-card .stat-number{{ font-size:1.7rem; font-weight:900; color:{BUTTON_BLUE}; }}
-#stat-card .stat-quote{{  margin-top:6px; font-size:1.05rem; color:{NAVY}; opacity:.9; }}
-#stat-card .stat-impact{{ margin-top:8px; font-size:.95rem; color:{NAVY}; opacity:.85; font-weight:600; }}
-#examples_label {{ color:#cfe3ff; opacity:.9; margin-top:6px; }}
-"""
+        return summary_md, tips_md, saved_headline, saved_detail, state
 
-with gr.Blocks(css=CUSTOM_CSS, title="WaterWise Home") as demo:
-    with gr.Column(elem_id="header-box"):
-        gr.Markdown("<h1>💧 WaterWise Home</h1>")
-        gr.Markdown("<p>I'm your smart chip! Tell me how you used water today, and I'll track your usage, give tips, and challenge you to save more.</p>")
+    except Exception as e:
+        import traceback; traceback.print_exc()
+        err = f"Sorry — something went wrong: {e}"
+        return err, "", "💧 0.0 liters saved", "", state or {"saved_liters": 0.0, "history": []}
 
-    with gr.Row():
-        activity = gr.Textbox(
-            label="Log your water use for today",
-            placeholder="e.g., Took a 10-minute shower • Ran a full dishwasher • Watered the garden",
-            lines=2,
+def on_clear(state: dict | None):
+    state = {"saved_liters": 0.0, "history": []}
+    return "💧 0.0 liters saved", "", state
+
+def set_example(text: str, season: str):
+    return text, season
+
+# -----------------------
+# UI (light/dark safe)
+# -----------------------
+CUSTOM_CSS = """
+:root {
+  --water-blue: #1570ef;
+  --water-blue-2: #2e90fa;
+  --card-bg: #ffffff;
+}
+body { background: linear-gradient(180deg, #dfefff 0%, #0c2d4a 800px, #0c2d4a 100%); }
+.gradio-container { max-width: 980px !important; margin: 0 auto !important; }
+#title-card { background: #ffffff; border-radius: 24px; padding: 22px; box-shadow: 0 10px 30px rgba(16,56,112,0.12); }
+#subtitle { color: #1f3b67; }
+label, .gr-text, .gr-markdown { color: #0b2c56 !important; }
+button { border-radius: 999px !important; }
+.gr-button { background: var(--water-blue) !important; border: none !important; color: white !important; }
+.gr-button:hover { background: var(--water-blue-2) !important; }
+.input-card, .output-card { background: #1f2a36; color: #e6f1ff; border-radius: 16px; padding: 18px; box-shadow: 0 8px 24px rgba(0,0,0,0.25); }
+.centered { display: flex; justify-content: center; align-items: center; text-align: center; }
+
+/* ---- Force readable look in dark theme ---- */
+[data-theme="dark"] body { background: linear-gradient(180deg, #dfefff 0%, #0c2d4a 800px, #0c2d4a 100%); }
+[data-theme="dark"] #title-card { background: #ffffff !important; }
+[data-theme="dark"] label,
+[data-theme="dark"] .gr-text,
+[data-theme="dark"] .gr-markdown { color: #e6f1ff !important; }
+[data-theme="dark"] .gr-button { background: var(--water-blue) !important; color: #ffffff !important; }
+"""
+
+theme = gr.themes.Soft(primary_hue="blue", neutral_hue="slate")
+
+with gr.Blocks(css=CUSTOM_CSS, theme=theme, title="WaterWise Home — Your smart water assistant") as demo:
+    # Header
+    with gr.Column(elem_id="title-card"):
+        gr.Markdown("# 💧 WaterWise Home", elem_classes=["centered"])
+        gr.Markdown(
+            "I'm your smart chip! Tell me how you used water today, and I'll track your usage, give tips, "
+            "and challenge you to save more.",
+            elem_id="subtitle",
+            elem_classes=["centered"],
         )
-        season = gr.Radio(SEASONS, label="🌤️ What season is it?", value=None, interactive=True, elem_id="season_pills")
 
-    gr.Markdown("#### 🔎 Try an example", elem_id="examples_label")
+    # Input / Output
     with gr.Row():
-        ex1 = gr.Button("🚿 10-min shower — example (Summer)", variant="secondary")
-        ex2 = gr.Button("🍽 Full dishwasher — example (Winter)", variant="secondary")
-        ex3 = gr.Button("🌱 Watered garden at noon — example (Summer)", variant="secondary")
-
-    def _ex1(): return "Took a 10-minute shower", "☀️ Summer"
-    def _ex2(): return "Ran a full dishwasher", "❄️ Winter"
-    def _ex3(): return "Watered the garden at noon", "☀️ Summer"
-    ex1.click(fn=_ex1, outputs=[activity, season])
-    ex2.click(fn=_ex2, outputs=[activity, season])
-    ex3.click(fn=_ex3, outputs=[activity, season])
+        with gr.Column(scale=1, elem_classes=["input-card"]):
+            user_text = gr.Textbox(
+                label="Log your water use for today",
+                placeholder="e.g., Took a 12-minute shower, ran the dishwasher twice, watered the lawn",
+                lines=2,
+                value=""
+            )
+            season = gr.Radio(
+                label="🌤️ What season is it?",
+                choices=["Winter", "Spring", "Summer", "Fall"],
+                value="Winter",
+                interactive=True,
+            )
+
+        with gr.Column(scale=1, elem_classes=["output-card"]):
+            summary_md = gr.Markdown()
+            tips_md = gr.Markdown()
+
+    # Examples row
+    gr.Markdown("🔎 **Try an example**")
+    with gr.Row():
+        ex1 = gr.Button("🛁 10-min shower — example (Summer)")
+        ex2 = gr.Button("🍽️ Full dishwasher — example (Winter)")
+        ex3 = gr.Button("🌱 Watered garden at noon — example (Summer)")
 
-    selected_season_md = gr.Markdown("_Selected: **none**_")
+    gr.Markdown("**Selected:** ❄️ Winter")
 
     with gr.Row():
-        get_btn = gr.Button("💡 Get My Water-Saving Tips")
-        clear_btn = gr.Button("🧹 Clear History")
-
-    tips_md = gr.Markdown("_(Your top 3 tips will appear here.)_")
-    msg_md  = gr.Markdown("")
-    gr.Markdown("### 💧 Lifetime Water Saved")
-    total_md  = gr.HTML(total_saved_card_html(0))
-
-    def on_season_change(s):
-        if not s:
-            return "_Selected: **none**_"
-        return f"_Selected: **{s}**_"
-    season.change(fn=on_season_change, inputs=season, outputs=selected_season_md)
-
-    def on_get(activity_text, szn):
-        tips_text, msg, stat_html = recommend(activity_text, szn)
-        return tips_text, msg, stat_html
-    get_btn.click(fn=on_get, inputs=[activity, season], outputs=[tips_md, msg_md, total_md])
-
-    def on_clear_click():
-        return clear_all()
-    clear_btn.click(fn=on_clear_click, outputs=[tips_md, msg_md, total_md])
-
-    def init_load():
-        _, total = load_history()
-        return "", "", total_saved_card_html(total), "_Selected: **none**_"
-    demo.load(fn=init_load, inputs=None, outputs=[tips_md, msg_md, total_md, selected_season_md])
+        submit = gr.Button("💡 Get My Water-Saving Tips", scale=1)
+        clear_btn = gr.Button("🧹 Clear History", scale=1)
+
+    # Savings section
+    gr.Markdown("💧 **Lifetime Water Saved**")
+    saved_headline = gr.Markdown("💧 0.0 liters saved")
+    saved_detail = gr.Markdown("Drops become streams, streams become rivers. 🌊\n\n≈ 0.0 showers · 0.0 dishwasher loads · 0.0 watering cans")
+
+    # App state
+    state = gr.State({"saved_liters": 0.0, "history": []})
+
+    # Wiring
+    submit.click(
+        fn=on_submit,
+        inputs=[user_text, season, state],
+        outputs=[summary_md, tips_md, saved_headline, saved_detail, state],
+    )
+
+    clear_btn.click(
+        fn=on_clear,
+        inputs=[state],
+        outputs=[saved_headline, saved_detail, state],
+    )
+
+    ex1.click(lambda: set_example("10 minute shower", "Summer"), None, [user_text, season])
+    ex2.click(lambda: set_example("Full dishwasher", "Winter"), None, [user_text, season])
+    ex3.click(lambda: set_example("Watered garden at noon", "Summer"), None, [user_text, season])
 
+# Local dev
 if __name__ == "__main__":
     demo.launch()