Upload 2 files

UI.py

@@ -1,5 +1,7 @@
 import streamlit as st
-from engine import process_question
+from engine import process_question,download_transcript_txt,download_transcript_json
+
+
 
 st.set_page_config(page_title="Hospital AI Assistant", layout="wide")
 
@@ -88,3 +90,30 @@ if user_input:
 
     with st.chat_message("assistant"):
         st.markdown(reply, unsafe_allow_html=True)
+# =========================
+# Download Conversation
+# =========================
+st.divider()
+st.subheader("📥 Download Conversation")
+
+col1, col2 = st.columns(2)
+
+with col1:
+    if st.button("Download Transcript (TXT)"):
+        txt = download_transcript_txt()
+        st.download_button(
+            label="Download TXT",
+            data=txt,
+            file_name="chat_transcript.txt",
+            mime="text/plain"
+        )
+
+with col2:
+    if st.button("Download Transcript (JSON)"):
+        js = download_transcript_json()
+        st.download_button(
+            label="Download JSON",
+            data=js,
+            file_name="chat_transcript.json",
+            mime="application/json"
+        )

engine.py

@@ -3,6 +3,10 @@ import re
 import sqlite3
 from openai import OpenAI
 from difflib import get_close_matches
+from datetime import datetime
+
+TRANSCRIPT = []
+
 
 # =========================
 # SETUP
@@ -623,6 +627,16 @@ def build_table_summary(table_name):
 # =========================
 
 def process_question(question):
+    
+    def log_interaction(user_q, sql=None, result=None, error=None):
+        TRANSCRIPT.append({
+        "timestamp": datetime.utcnow().isoformat(),
+        "question": user_q,
+        "sql": sql,
+        "result_preview": result[:10] if isinstance(result, list) else result,
+        "error": error
+    })
+
     global LAST_PROMPT_TYPE, LAST_SUGGESTED_DATE
 
     q = question.strip().lower()
@@ -720,12 +734,16 @@ def process_question(question):
     try:
         sql = call_llm(build_prompt(question))
     except ValueError as e:
-        # Handle case where no relevant tables found
-        return {
-            "status": "ok",
-            "message": str(e),
-            "data": []
-        }
+        log_interaction(
+        user_q=question,
+        error=str(e)
+    )
+    return {
+        "status": "ok",
+        "message": str(e),
+        "data": []
+    }
+
 
     if sql == "NOT_ANSWERABLE":
         return {
@@ -739,6 +757,13 @@ def process_question(question):
     sql = correct_table_names(sql)
     sql = validate_sql(sql)
     cols, rows = run_query(sql)
+    
+    log_interaction(
+    user_q=question,
+    sql=sql,
+    result=rows
+)
+
 
     # ----------------------------------
     # No data handling
@@ -755,6 +780,12 @@ def process_question(question):
         }
 
     if not rows:
+        log_interaction(
+    user_q=question,
+    sql=sql,
+    result=[]
+)
+
         LAST_PROMPT_TYPE = "NO_DATA"
         LAST_SUGGESTED_DATE = get_latest_data_date()
 
@@ -774,4 +805,26 @@ 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):
+        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)
+