Update engine.py

engine.py

@@ -16,6 +16,7 @@ conn = sqlite3.connect("hospital.db", check_same_thread=False)
 # =========================
 
 LAST_PROMPT_TYPE = None
+LAST_SUGGESTED_DATE = None
 
 # =========================
 # HUMAN RESPONSE HELPERS
@@ -30,6 +31,9 @@ def friendly(text):
 def is_confirmation(text):
     return text.strip().lower() in ["yes", "yep", "yeah", "ok", "okay", "sure"]
 
+def is_why_question(text):
+    return text.strip().lower().startswith("why")
+
 # =========================
 # SPELL CORRECTION
 # =========================
@@ -227,12 +231,29 @@ def has_underlying_data(sql):
 # =========================
 
 def process_question(question):
-    global LAST_PROMPT_TYPE
+    global LAST_PROMPT_TYPE, LAST_SUGGESTED_DATE
+
+    q = question.strip().lower()
 
-    question = question.strip().lower()
+    # Handle WHY questions
+    if is_why_question(q) and LAST_PROMPT_TYPE == "NO_DATA":
+        return {
+            "status": "ok",
+            "message": (
+                f"I suggested **{LAST_SUGGESTED_DATE[:4]}** because that’s the most recent year "
+                f"for which data exists in the system.\n\n"
+                "Your database doesn’t contain newer records, so questions like "
+                "“today” or “this year” can’t be answered yet.\n\n"
+                "You can still explore:\n"
+                "• Data from 2021\n"
+                "• Trends over time\n"
+                "• Patient summaries"
+            ),
+            "data": []
+        }
 
-    # Handle confirmation replies like "yes"
-    if is_confirmation(question) and LAST_PROMPT_TYPE == "NO_DATA":
+    # Handle confirmations
+    if is_confirmation(q) and LAST_PROMPT_TYPE == "NO_DATA":
         return {
             "status": "ok",
             "message": (
@@ -241,8 +262,7 @@ def process_question(question):
                 "• How many patients were admitted in 2021?\n"
                 "• Patient count by gender\n"
                 "• Total visits by month\n"
-                "• Most common conditions\n\n"
-                "Just type one of these or ask your own question."
+                "• Most common conditions"
             ),
             "data": []
         }
@@ -286,25 +306,26 @@ def process_question(question):
 
     if is_aggregate_only_query(sql) and not has_underlying_data(sql):
         LAST_PROMPT_TYPE = "NO_DATA"
-        latest = get_latest_data_date()
+        LAST_SUGGESTED_DATE = get_latest_data_date()
         return {
             "status": "ok",
             "message": friendly("No data is available for that time period."),
-            "note": f"Available data is only up to {latest}.",
+            "note": f"Available data is only up to {LAST_SUGGESTED_DATE}.",
             "data": []
         }
 
     if not rows:
         LAST_PROMPT_TYPE = "NO_DATA"
-        latest = get_latest_data_date()
+        LAST_SUGGESTED_DATE = get_latest_data_date()
         return {
             "status": "ok",
             "message": friendly("No records found."),
-            "note": f"Available data is only up to {latest}.",
+            "note": f"Available data is only up to {LAST_SUGGESTED_DATE}.",
             "data": []
         }
 
     LAST_PROMPT_TYPE = None
+    LAST_SUGGESTED_DATE = None
 
     return {
         "status": "ok",