Update engine.py

engine.py

@@ -279,8 +279,10 @@ 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",
@@ -290,16 +292,18 @@ def process_question(question):
             "note": None
         }
 
-    # Handle WHY questions
+    # -------------------------------
+    # WHY follow-up handling
+    # -------------------------------
     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 **{LAST_SUGGESTED_DATE[:4]}** because that’s the most recent year "
+                f"I suggested **{year}** 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"
+                "Your database doesn’t contain newer records yet.\n\n"
+                "You can explore:\n"
                 "• Data from 2021\n"
                 "• Trends over time\n"
                 "• Patient summaries"
@@ -307,7 +311,9 @@ def process_question(question):
             "data": []
         }
 
-    # Handle confirmations
+    # -------------------------------
+    # YES / confirmation handling
+    # -------------------------------
     if is_confirmation(q) and LAST_PROMPT_TYPE == "NO_DATA":
         return {
             "status": "ok",
@@ -322,9 +328,19 @@ 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",
@@ -332,6 +348,9 @@ def process_question(question):
             "data": []
         }
 
+    # -------------------------------
+    # Unsupported questions
+    # -------------------------------
     unsupported = get_unsupported_reason(question)
     if unsupported:
         return {
@@ -347,6 +366,9 @@ def process_question(question):
             "data": []
         }
 
+    # -------------------------------
+    # LLM → SQL
+    # -------------------------------
     sql = call_llm(build_prompt(question))
 
     if sql == "NOT_ANSWERABLE":
@@ -359,9 +381,13 @@ 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."),
@@ -370,9 +396,13 @@ 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."),
@@ -380,12 +410,13 @@ def process_question(question):
             "data": []
         }
 
-    LAST_PROMPT_TYPE = None
-    LAST_SUGGESTED_DATE = None
-
+    # -------------------------------
+    # Successful response
+    # -------------------------------
     return {
         "status": "ok",
         "sql": sql,
         "columns": cols,
         "data": rows
     }
+