Upload engine.py

engine.py

@@ -737,6 +737,40 @@ def process_question(question):
             "message": humanize(describe_schema()),
             "data": []
         }
+    # ----------------------------------
+    # # LAST DATA / RECENT DATA HANDLING
+    # # ----------------------------------
+    if any(x in q for x in ["last data", "latest data"]):
+        return {
+        "status": "ok",
+        "message": f"Latest data available is from {get_latest_data_date()}",
+        "data": []
+    }
+
+    if "last" in q and "day" in q and ("visit" in q or "admission" in q):
+        sql = """
+    SELECT subject_id, admittime
+    FROM admissions
+    WHERE admittime >= date(
+        (SELECT MAX(admittime) FROM admissions),
+        '-30 days'
+    )
+    ORDER BY admittime DESC
+    """
+    cols, rows = run_query(sql)
+
+    log_interaction(
+        user_q=question,
+        sql=sql,
+        result=rows
+    )
+
+    return {
+        "status": "ok",
+        "sql": sql,
+        "columns": cols,
+        "data": rows
+    }
 
     # ----------------------------------
     # Unsupported question check
@@ -746,7 +780,7 @@ def process_question(question):
         user_q=question,
         error="Unsupported question"
     )
-    return {
+        return {
         "status": "ok",
         "message": (
             "That information isn’t available in the system.\n\n"
@@ -760,8 +794,6 @@ def process_question(question):
         "data": []
     }
 
-
-
     # ----------------------------------
     # Generate SQL
     # ----------------------------------
@@ -855,25 +887,4 @@ def process_question(question):
         "sql": sql,
         "columns": cols,
         "data": rows
-    }
-
-def download_transcript_txt():
-    lines = []
-    for i, entry in enumerate(st.session_state.transcript, 1):
-        lines.append(f"\n--- Query {i} ---")
-        lines.append(f"Time: {entry['timestamp']}")
-        lines.append(f"Question: {entry['question']}")
-
-        if entry.get("sql"):
-            lines.append(f"SQL: {entry['sql']}")
-
-        if entry.get("result_preview"):
-            lines.append(f"Result Preview: {entry['result_preview']}")
-
-        if entry.get("error"):
-            lines.append(f"Error: {entry['error']}")
-
-    return "\n".join(lines)
-
-
-
+    }