Upload engine.py

engine.py

@@ -20,7 +20,7 @@ KNOWN_TERMS = [
     "patient", "patients", "condition", "conditions", "diagnosis", "encounter", "encounters",
     "visit", "visits", "observation", "observations", "lab", "labs", "test", "tests",
     "medication", "medications", "drug", "drugs", "prescription", "prescriptions",
-    "diabetes", "hypertension", "asthma", "cancer"
+    "diabetes", "hypertension", "asthma", "cancer", "admitted", "admission"
 ]
 
 
@@ -180,8 +180,10 @@ def run_query(sql: str):
 # =========================
 
 def is_question_answerable(question):
-    schema = load_ai_schema()
-    keywords = ["patient", "encounter", "condition", "observation", "medication", "visit", "diagnosis", "lab", "vital"]
+    keywords = [
+        "patient", "encounter", "condition", "observation",
+        "medication", "visit", "diagnosis", "lab", "vital", "admitted"
+    ]
 
     q = question.lower()
 
@@ -205,7 +207,7 @@ def check_time_relevance(question: str):
     q = question.lower()
     if any(word in q for word in ["last", "recent", "today", "this month", "this year"]):
         latest = get_latest_data_date()
-        return f"Note: Latest available data is from {latest}."
+        return f"Latest available data is from {latest}."
     return None
 
 
@@ -245,8 +247,20 @@ def process_question(question: str):
     # 5. Execute query
     columns, rows = run_query(sql)
 
-    # 6. Handle empty result
+    # 6. Handle empty result with data coverage awareness
     if len(rows) == 0:
+        latest = get_latest_data_date()
+        q = question.lower()
+
+        if any(word in q for word in ["last", "recent", "this month", "this year"]):
+            return {
+                "status": "ok",
+                "sql": sql,
+                "message": f"No data available for the requested time period. Latest available data is from {latest}.",
+                "data": [],
+                "note": None
+            }
+
         return {
             "status": "ok",
             "sql": sql,