Upload UI.py

UI.py

@@ -44,16 +44,6 @@ if user_input:
     with st.spinner("Thinking..."):
         try:
             result = process_question(user_input)
-
-            # ✅ Store transcript safely
-            st.session_state.transcript.append({
-                "timestamp": datetime.utcnow().isoformat(),
-                "question": user_input,
-                "sql": result.get("sql"),
-                "result_preview": result.get("data", [])[:10],
-                "error": result.get("message") if result.get("status") != "ok" else None
-            })
-
         except Exception as e:
             result = {
                 "status": "error",
@@ -93,7 +83,21 @@ if user_input:
     else:
         reply = f"❌ {result.get('message', 'Something went wrong')}"
 
-    # Show assistant message
+    # =========================
+    # SAVE TRANSCRIPT ✅ FIXED
+    # =========================
+    st.session_state.transcript.append({
+        "timestamp": datetime.utcnow().isoformat(),
+        "question": user_input,
+        "reply": reply,            # ✅ THIS WAS MISSING
+        "sql": result.get("sql"),
+        "result_preview": result.get("data", [])[:10],
+        "error": result.get("message") if result.get("status") != "ok" else None
+    })
+
+    # =========================
+    # SHOW ASSISTANT MESSAGE
+    # =========================
     st.session_state.messages.append(
         {"role": "assistant", "content": reply}
     )
@@ -102,7 +106,7 @@ if user_input:
         st.markdown(reply, unsafe_allow_html=True)
 
 # =========================
-# DOWNLOAD TRANSCRIPT (BOTTOM ONLY)
+# DOWNLOAD TRANSCRIPT
 # =========================
 def download_transcript_txt():
     lines = []
@@ -111,18 +115,19 @@ def download_transcript_txt():
         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("reply"):
+            lines.append("Reply:")
+            lines.append(entry["reply"])
 
-        if entry.get("result_preview"):
-            lines.append(f"Result Preview: {entry['result_preview']}")
+        if entry.get("sql"):
+            lines.append(f"SQL:\n{entry['sql']}")
 
         if entry.get("error"):
             lines.append(f"Error: {entry['error']}")
 
     return "\n".join(lines)
 
-
+# Show download button only if data exists
 if st.session_state.transcript:
     st.divider()
     st.download_button(