Update engine.py

engine.py

@@ -279,10 +279,8 @@ def process_question(question):
     global LAST_PROMPT_TYPE, LAST_SUGGESTED_DATE
 
     q = question.strip().lower()
-
-    # -------------------------------
+    
     # Patient summary intent
-    # -------------------------------
     if is_patient_summary_question(q):
         return {
             "status": "ok",
@@ -292,18 +290,16 @@ def process_question(question):
             "note": None
         }
 
-    # -------------------------------
-    # WHY follow-up handling
-    # -------------------------------
+    # Handle WHY questions
     if is_why_question(q) and LAST_PROMPT_TYPE == "NO_DATA":
-        year = LAST_SUGGESTED_DATE[:4] if LAST_SUGGESTED_DATE else "the latest available year"
         return {
             "status": "ok",
             "message": (
-                f"I suggested **{year}** because that’s the most recent year "
+                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 yet.\n\n"
-                "You can explore:\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"
@@ -311,9 +307,7 @@ def process_question(question):
             "data": []
         }
 
-    # -------------------------------
-    # YES / confirmation handling
-    # -------------------------------
+    # Handle confirmations
     if is_confirmation(q) and LAST_PROMPT_TYPE == "NO_DATA":
         return {
             "status": "ok",
@@ -328,19 +322,9 @@ def process_question(question):
             "data": []
         }
 
-    # -------------------------------
-    # Normalize question
-    # -------------------------------
     question = correct_spelling(question)
     question = normalize_time_question(question)
 
-    # Reset state once user asks a fresh question
-    LAST_PROMPT_TYPE = None
-    LAST_SUGGESTED_DATE = None
-
-    # -------------------------------
-    # Metadata queries
-    # -------------------------------
     if any(x in question for x in ["what data", "what tables", "which data"]):
         return {
             "status": "ok",
@@ -348,9 +332,6 @@ def process_question(question):
             "data": []
         }
 
-    # -------------------------------
-    # Unsupported questions
-    # -------------------------------
     unsupported = get_unsupported_reason(question)
     if unsupported:
         return {
@@ -366,9 +347,6 @@ def process_question(question):
             "data": []
         }
 
-    # -------------------------------
-    # LLM → SQL
-    # -------------------------------
     sql = call_llm(build_prompt(question))
 
     if sql == "NOT_ANSWERABLE":
@@ -381,13 +359,9 @@ def process_question(question):
     sql = validate_sql(sanitize_sql(sql))
     cols, rows = run_query(sql)
 
-    # -------------------------------
-    # No data (aggregate case)
-    # -------------------------------
     if is_aggregate_only_query(sql) and not has_underlying_data(sql):
         LAST_PROMPT_TYPE = "NO_DATA"
         LAST_SUGGESTED_DATE = get_latest_data_date()
-
         return {
             "status": "ok",
             "message": friendly("No data is available for that time period."),
@@ -396,13 +370,9 @@ def process_question(question):
             "sql": None
         }
 
-    # -------------------------------
-    # Empty result set
-    # -------------------------------
     if not rows:
         LAST_PROMPT_TYPE = "NO_DATA"
         LAST_SUGGESTED_DATE = get_latest_data_date()
-
         return {
             "status": "ok",
             "message": friendly("No records found."),
@@ -410,13 +380,12 @@ def process_question(question):
             "data": []
         }
 
-    # -------------------------------
-    # Successful response
-    # -------------------------------
+    LAST_PROMPT_TYPE = None
+    LAST_SUGGESTED_DATE = None
+
     return {
         "status": "ok",
         "sql": sql,
         "columns": cols,
         "data": rows
     }
-