Upload 2 files

engine.py

@@ -136,17 +136,43 @@ def extract_relevant_tables(question, max_tables=4):
     
     # Build hints only for tables that actually exist
     hint_mappings = {
-        "patient": ["patients"],
-        "admission": ["admissions"],
-        "visit": ["admissions", "icustays"],
-        "icu": ["icustays", "chartevents"],
-        "diagnosis": ["diagnoses_icd"],
-        "procedure": ["procedures_icd"],
-        "medication": ["prescriptions", "emar", "pharmacy"],
-        "lab": ["labevents"],
-        "vital": ["chartevents"],
-        "stay": ["icustays"]
-    }
+    # Patients & visits
+    "patient": ["patients"],
+    "patients": ["patients"],
+
+    "admission": ["admissions"],
+    "admissions": ["admissions"],
+    "visit": ["admissions", "icustays"],
+    "visits": ["admissions", "icustays"],
+
+    # ICU
+    "icu": ["icustays", "chartevents"],
+    "stay": ["icustays"],
+    "stays": ["icustays"],
+
+    # Diagnoses / conditions
+    "diagnosis": ["diagnoses_icd"],
+    "diagnoses": ["diagnoses_icd"],
+    "condition": ["diagnoses_icd"],
+    "conditions": ["diagnoses_icd"],
+
+    # Procedures
+    "procedure": ["procedures_icd"],
+    "procedures": ["procedures_icd"],
+
+    # Medications
+    "medication": ["prescriptions", "emar", "pharmacy"],
+    "medications": ["prescriptions", "emar", "pharmacy"],
+    "drug": ["prescriptions"],
+    "drugs": ["prescriptions"],
+
+    # Labs & vitals
+    "lab": ["labevents"],
+    "labs": ["labevents"],
+    "vital": ["chartevents"],
+    "vitals": ["chartevents"],
+}
+
     
     # Only include hints for tables that exist in the schema
     for intent, possible_tables in hint_mappings.items():
@@ -716,18 +742,24 @@ def process_question(question):
     # Unsupported question check
     # ----------------------------------
     if not is_question_supported(question):
-        return {
-            "status": "ok",
-            "message": (
-                "That information isn’t available in the system.\n\n"
-                "You can ask about:\n"
-                "• Patients\n"
-                "• Visits\n"
-                "• Conditions\n"
-                "• Medications"
-            ),
-            "data": []
-        }
+        log_interaction(
+        user_q=question,
+        error="Unsupported question"
+    )
+
+    return {
+        "status": "ok",
+        "message": (
+            "That information isn’t available in the system.\n\n"
+            "You can ask about:\n"
+            "• Patients\n"
+            "• Visits\n"
+            "• Conditions\n"
+            "• Medications"
+        ),
+        "data": []
+    }
+
 
     # ----------------------------------
     # Generate SQL
@@ -746,6 +778,7 @@ def process_question(question):
     }
 
 
+
     if sql == "NOT_ANSWERABLE":
         return {
             "status": "ok",
@@ -822,13 +855,10 @@ def process_question(question):
         "columns": cols,
         "data": rows
     }
-def download_transcript_json():
-    import json
-    return json.dumps(TRANSCRIPT, indent=2)
 
 def download_transcript_txt():
     lines = []
-    for i, entry in enumerate(TRANSCRIPT, 1):
+    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']}")
@@ -845,3 +875,4 @@ def download_transcript_txt():
     return "\n".join(lines)
 
 
+