Update app.py

app.py

@@ -1,6 +1,11 @@
 import json, re
 import gradio as gr
 
+try:
+    from transformers import pipeline
+except ImportError:  # pragma: no cover - handled gracefully if transformers missing
+    pipeline = None
+
 # Load knowledge
 with open("kb.json", "r", encoding="utf-8") as f:
     KB = json.load(f)
@@ -28,6 +33,59 @@ GUIDE_QS = [
     ("extras", "Extras available? (pick any, or 'None')", ["Fan","Washing machine","Garden","Rooftop","None"])
 ]
 
+INTENT_LABELS = [
+    "water",
+    "electricity",
+    "waste",
+    "greeting",
+    "smalltalk",
+    "affirmation",
+    "gratitude",
+    "checklist",
+    "goodbye"
+]
+INTENT_THRESHOLD = 0.45
+_INTENT_PIPELINE = None
+_INTENT_PIPELINE_ERR = None
+
+
+def load_intent_classifier():
+    """Lazily load the zero-shot intent classifier pipeline."""
+    global _INTENT_PIPELINE, _INTENT_PIPELINE_ERR
+    if pipeline is None:
+        _INTENT_PIPELINE_ERR = "transformers not installed"
+        return None
+    if _INTENT_PIPELINE is not None or _INTENT_PIPELINE_ERR:
+        return _INTENT_PIPELINE
+    try:
+        _INTENT_PIPELINE = pipeline(
+            "zero-shot-classification",
+            model="typeform/distilbert-base-uncased-mnli"
+        )
+    except Exception as err:  # pragma: no cover - model download/setup issues
+        _INTENT_PIPELINE_ERR = err
+        return None
+    return _INTENT_PIPELINE
+
+
+def classify_intent(text):
+    """Return (label, score) for the strongest intent, or (None, 0.0)."""
+    if not text:
+        return None, 0.0
+    clf = load_intent_classifier()
+    if clf is None:
+        return None, 0.0
+    result = clf(text, candidate_labels=INTENT_LABELS, multi_label=True)
+    labels = result.get("labels") or []
+    scores = result.get("scores") or []
+    if not labels or not scores:
+        return None, 0.0
+    best_label, best_score = max(zip(labels, scores), key=lambda pair: pair[1])
+    if best_score < INTENT_THRESHOLD:
+        return None, best_score
+    return best_label, best_score
+
+
 def map_habit_to_cards(focus, habit):
     focus = focus.lower()
     if focus=="water":
@@ -116,6 +174,7 @@ def bot_reply(user, state):
     # TUTOR mode quick topic guess
     t = (user or "").lower()
     normalized = re.sub(r"[^\w\s]", "", t).strip()
+    intent_label, intent_score = classify_intent(user or "")
     if any(w in t for w in ["water","shower","tap","leak","plant","tree","bottle"]):
         topic="water"
     elif any(w in t for w in ["ac","power","electric","fan","bulb","led"]):
@@ -125,6 +184,9 @@ def bot_reply(user, state):
     else:
         topic=None
 
+    if topic is None and intent_label in {"water","electricity","waste"}:
+        topic = intent_label
+
     if topic:
         home_hint = "apartment" if "apartment" in t else ("house" if "house" in t else None)
         cards = find_cards(topic, home_hint) or find_cards(topic)
@@ -132,19 +194,29 @@ def bot_reply(user, state):
         return short_answer(card), state
 
     # small talk and acknowledgements
+    if intent_label == "greeting" and intent_score >= INTENT_THRESHOLD:
+        return "Salaam! I'm your Green Guide bot - ask me about water, electricity, or waste habits anytime.", state
     if re.search(r"\b(hi|hello|hey|salaam|assalam|assalamu alaikum|assalamualaikum)\b", t):
         return "Salaam! I'm your Green Guide bot - ask me about water, electricity, or waste habits anytime.", state
+    if intent_label == "smalltalk" and intent_score >= INTENT_THRESHOLD:
+        return "Happy to chat! Got any water, electricity, or waste question for me?", state
     if "how are you" in t:
         return "I'm running on code and clean energy facts! How can I help with your sustainability goal?", state
+    if intent_label in {"affirmation","gratitude"} and intent_score >= INTENT_THRESHOLD:
+        return "Noted! Tell me what you want to improve: water, electricity, or waste, and I'll share tips.", state
     if normalized in {"ok","okay","yes","yup","sure","cool","great","thanks","thank you","thankyou","nice"}:
         return "Noted! Tell me what you want to improve: water, electricity, or waste, and I'll share tips.", state
+    if intent_label == "goodbye" and intent_score >= INTENT_THRESHOLD:
+        return "Khuda hafiz! Come back anytime you need another green tip.", state
 
     # checklists
+    if intent_label == "checklist" and intent_score >= INTENT_THRESHOLD:
+        return "Need a checklist? Type **water**, **power**, or **waste** and I'll drop a 7-day plan.", state
     if re.search(r"\b(checklist|pledge)\b", t):
-        if "power" in t: 
-            return "Power Ninja 7-day: LED swap · 26°C nights · fan-first · unplug at 10pm · kill standby · use daylight · share result", state
-        if "waste" in t: 
-            return "Waste Warrior 7-day: 2-bin · rinse items · paper stack · e-waste box · no plastic bag · find drop-off · share photo", state
+        if "power" in t:
+            return "Power Ninja 7-day: LED swap ? 26?C nights ? fan-first ? unplug at 10pm ? kill standby ? use daylight ? share result", state
+        if "waste" in t:
+            return "Waste Warrior 7-day: 2-bin ? rinse items ? paper stack ? e-waste box ? no plastic bag ? find drop-off ? share photo", state
         return "Water Saver 7-day: fix drip · 5-min showers · reuse rinse for plants · full loads · cold bottle in fridge · log habits · share one tip", state
 
     return "Try **Tutor**: ask me anything about water/electricity/waste. Or type **Guide** for a mini plan.", state