Upload engine.py

engine.py

@@ -219,6 +219,34 @@ def interpret_empty_result(question: str):
     latest = get_latest_data_date()
     return f"No results found. Available data is up to {latest}."
 
+# =========================
+# Data Range Check
+# =========================
+from datetime import datetime
+
+def is_request_out_of_data_range(question: str) -> bool:
+    latest = get_latest_data_date()
+
+    if not latest:
+        return True
+
+    latest_date = datetime.fromisoformat(latest.replace("Z", "").split("T")[0])
+    now = datetime.now()
+
+    q = question.lower()
+
+    if "this year" in q:
+        return latest_date.year < now.year
+
+    if "last month" in q:
+        return (now.year, now.month - 1) > (latest_date.year, latest_date.month)
+
+    if "recent" in q or "last 30" in q:
+        return (now - latest_date).days > 30
+
+    return False
+
+
 
 # =========================
 # ORCHESTRATOR (Single Entry Point)
@@ -236,7 +264,17 @@ def process_question(question: str):
         }
 
     # 2. Time relevance
-    time_note = check_time_relevance(question)
+    # 2. Time relevance check
+    if is_request_out_of_data_range(question):
+        latest = get_latest_data_date()
+        return {
+            "status": "ok",
+            "message": f"No data available for the requested time period. Latest available data is from {latest}.",
+            "data": [],
+            "sql": None,
+            "note": None
+            }
+
 
     # 3. Generate SQL
     sql = generate_sql(question)