Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import re
 import gradio as gr
 
 # --------- CPU hygiene (nice-to-have) ----------
@@ -53,10 +54,10 @@ def get_vader():
         print("[startup] Loaded VADER sentiment analyzer")
     return _vader
 
-# -------- Helpers --------
+# -------- Sentiment helpers --------
 def detect_sentiment_bucket(text: str):
     """
-    Return ('neg'|'neu'|'pos', compound_score).
+    Return ('neg'|'neu'|'pos', compound_score) using VADER.
     Thresholds chosen for clear buckets in chat settings.
     """
     scores = get_vader().polarity_scores(text or "")
@@ -67,17 +68,59 @@ def detect_sentiment_bucket(text: str):
         return "pos", c
     return "neu", c
 
-def apply_tone_prefix(reply_text: str, bucket: str) -> str:
-    """Prepend a tiny tone wrapper without changing factual content."""
+def is_question(text: str) -> bool:
+    t = (text or "").strip()
+    if "?" in t:
+        return True
+    # Heuristic for question-like openings
+    return bool(re.match(r"^(who|what|when|where|why|how|do|does|did|can|could|is|are|was|were|should|would|will)\b", t.lower()))
+
+def is_thanks_or_praise(text: str) -> bool:
+    t = (text or "").lower()
+    return any(k in t for k in [
+        "thanks", "thank you", "appreciate", "appreciated",
+        "great answer", "nice", "awesome", "love", "helpful",
+        "that helped", "that was good", "i like your response"
+    ])
+
+POS_QUESTION_PREFIXES = [
+    "Good question! ",
+    "Nice one—here’s the gist: ",
+    "Let’s dig in. ",
+]
+
+POS_PRAISE_PREFIXES = [
+    "You’re welcome—glad that helped. ",
+    "Appreciate the kind words! ",
+    "Happy it was useful. ",
+]
+
+POS_STATEMENT_PREFIXES = [
+    "Sounds good. ",
+    "Got it. ",
+    "All right—here’s the short version. ",
+]
+
+# Your custom negative message:
+NEG_PREFIX = "Calm down. You're being a little too negative! "
+
+def choose_positive_prefix(message: str) -> str:
+    if is_thanks_or_praise(message):
+        return POS_PRAISE_PREFIXES[0]
+    if is_question(message):
+        return POS_QUESTION_PREFIXES[0]
+    return POS_STATEMENT_PREFIXES[0]
+
+def apply_tone_prefix(reply_text: str, bucket: str, message: str = "") -> str:
     if bucket == "pos":
-        prefix = "Great question! "
+        prefix = choose_positive_prefix(message)
     elif bucket == "neg":
-        prefix = "Calm down. You're being a little too negative! "
+        prefix = NEG_PREFIX
     else:
         prefix = ""
     return (prefix + (reply_text or "")).strip()
 
-# ---- Use the LLM if the generator doesn't understand user prompt ----
+# ---- LLM fallback (used when rules don't match) ----
 def ai_fallback(prompt: str) -> str:
     try:
         gen = get_t2t()
@@ -117,12 +160,12 @@ def reply(message, history):
         base = ai_fallback(message)
 
     # 3) tone wrapper (content unchanged)
-    return apply_tone_prefix(base, bucket)
+    return apply_tone_prefix(base, bucket, message)
 
 # -------- UI --------
 demo = gr.ChatInterface(
     fn=reply,
-    title="Cats of Ancient Egypt Chatbot"
+    title="😺 Cats of Ancient Egypt Chatbot 😺"
 )
 
 if __name__ == "__main__":