Update app.py

app.py

@@ -14,6 +14,8 @@ from datetime import datetime
 from pathlib import Path
 import streamlit as st
 
+if "firebase_record_id" not in st.session_state:
+    st.session_state.firebase_record_id = None
 
 # -----------------------
 # Sidebar: Short User Manual
@@ -126,22 +128,19 @@ else:
 # -----------------------
 def get_latest_wound_data():
     if not db:
-        return None
+        return None, None
     try:
         data = db.child("data").get().val()
-        if not data:
-            return None
-        # ✅ Nếu data là dict và chứa key "Wound" trực tiếp
-        if "Wound" in data:
-            return data
-        # ✅ Nếu có nhiều bản ghi (record_001, record_002, ...)
-        elif isinstance(data, dict):
-            latest_key = list(data.keys())[-1]
-            return data[latest_key]
-        return data
+        if not data or not isinstance(data, dict):
+            return None, None
+
+        latest_key = list(data.keys())[-1]
+        return latest_key, data[latest_key]
+
     except Exception as e:
         st.error(f"Error reading Firebase data: {e}")
-        return None
+        return None, None
+
 
 
 # -----------------------
@@ -149,7 +148,11 @@ def get_latest_wound_data():
 # -----------------------
 st.title("🩺 First Aid Assistant — Firebase Connected")
 
-data = get_latest_wound_data()
+record_id, data = get_latest_wound_data()
+
+if record_id:
+    st.session_state.firebase_record_id = record_id
+
 
 if isinstance(data, str):
     try:
@@ -189,7 +192,11 @@ else:
 st.markdown("---")
 if st.button("🔄 Refresh Firebase data"):
     st.toast("Refreshing data from Firebase...", icon="🔁")
-    data = get_latest_wound_data()
+record_id, data = get_latest_wound_data()
+
+if record_id:
+    st.session_state.firebase_record_id = record_id
+
     if data:
         st.success("✅ Data refreshed successfully!")
 
@@ -646,29 +653,28 @@ def show_and_optionally_reset_config():
 # Function: Save Victim Info to Firebase
 def save_victim_info(victim: dict):
     """
-    Save a victim record to Firebase Realtime Database using Pyrebase4.
-    Automatically generates a unique record ID.
-    Returns the saved record (dict).
+    Save victim info using EXISTING Firebase record ID
+    (same ID as data/<ID>, e.g. 001).
     """
     if not db:
         raise ValueError("Firebase not connected.")
 
-    import time
+    # Get existing record ID from session
+    record_id = st.session_state.get("firebase_record_id")
+    if not record_id:
+        raise ValueError("No Firebase record ID found.")
 
-    # Unique ID for each victim record
-    record_id = f"victim_{int(time.time())}"
-
-    # Ensure wound_type & timestamp fields exist
     victim_data = victim.copy()
-    victim_data["timestamp"] = time.strftime("%d/%m/%Y %H:%M:%S")
+    victim_data["timestamp"] = datetime.utcnow().strftime("%d/%m/%Y %H:%M:%S")
 
     try:
-        # Save data under "victims/<record_id>"
-        db.child("victims").child(record_id).set(victim_data)
+        # Update instead of set (do NOT overwrite AI result)
+        db.child("victims").child(record_id).update(victim_data)
         return victim_data
     except Exception as e:
         raise RuntimeError(f"Failed to save victim info: {e}")
 
+
 # Advanced section
 if st.session_state.get("basic_done") and st.session_state.get("wound_type"):
     st.header("Advanced assistance")