Update app.py

app.py

@@ -68,6 +68,40 @@ KU_GENERAL_RE = re.compile(r"\b(khalifa university|ku)\b.*\b(admission|admission
 LIBRARY_HOURS_URL = "https://library.ku.ac.ae/hours"
 KU_MAIN_URL = "https://www.ku.ac.ae/"
 
+GREETING_FOLLOWUP_RE = re.compile(r"^(yes|yes please|yes check|sure|okay|ok|yeah|yep|please do|go ahead)$", re.IGNORECASE)
+
+def _last_assistant_message(history):
+    if not history:
+        return ""
+    for m in reversed(history):
+        if m.get("role") == "assistant":
+            return m.get("content", "") or ""
+    return ""
+
+def _is_greeting_menu_followup(question: str, history) -> bool:
+    q = (question or "").strip()
+    if not GREETING_FOLLOWUP_RE.match(q):
+        return False
+    last = _last_assistant_message(history).lower()
+    return (
+        "i’m <strong>libbee</strong>" in last
+        or "i'm <strong>libbee</strong>" in last
+        or "are you looking for one of these" in last
+        or "khalifa university library ai assistant" in last
+    )
+
+def _greeting_menu_clarify_answer() -> str:
+    return (
+        "Sure — what would you like help with?<br><br>"
+        "You can choose one of these or type your question directly:<br>"
+        "• Find articles and books<br>"
+        "• Get full text for an article<br>"
+        "• Submit an Interlibrary Loan (ILL) request<br>"
+        "• Contact a librarian<br>"
+        "• Check library hours"
+    )
+
+
 GROUNDED_LIBRARY_MAP = {
     "ill": "interlibrary loan ILL document delivery full text unavailable article not available borrow from another library",
     "fulltext": "full text libkey nomad article access pdf unavailable interlibrary loan",
@@ -967,8 +1001,7 @@ async def agent_query(req: AgentRequest):
 
     # ---- Early greeting handling ----
     q_clean = question.strip()
-    q_norm = re.sub(r"\s+", " ", q_clean.lower()).strip(" !?.")
-    if PURE_GREETING_RE.match(q_clean) or q_norm in {"hi", "hello", "hey", "good morning", "good afternoon", "good evening"}:
+    if PURE_GREETING_RE.match(q_clean):
         answer = (
             "Hi! I’m <strong>LibBee</strong>, the Khalifa University Library AI Assistant.<br><br>"
             "I can help you find articles and books, search databases, access full text, "
@@ -992,6 +1025,26 @@ async def agent_query(req: AgentRequest):
             "source_mode": "social",
         }
 
+    # ---- Follow-up to the greeting menu ----
+    if _is_greeting_menu_followup(question, history):
+        answer = _greeting_menu_clarify_answer()
+        elapsed = time.time() - start
+        return {
+            "answer": answer,
+            "intent": "social_greeting",
+            "tools_used": [],
+            "search_results": [],
+            "sources": [],
+            "model_used": req.model,
+            "response_time": round(elapsed, 2),
+            "corrected_query": question,
+            "natural_query": question,
+            "database_query": question,
+            "original_question": question,
+            "is_follow_up": True,
+            "source_mode": "social",
+        }
+
 
     if _looks_library_hours_question(question):
         elapsed = time.time() - start