Update app.py

app.py

@@ -423,7 +423,9 @@ st.markdown("---")
 if st.button("🔄 " + ("Refresh Firebase data" if st.session_state.lang == "en" else "Làm mới dữ liệu Firebase")):
     render_latest_firebase_data(show_toast=True)
 
-# Advanced assistance block
+# -----------------------
+# Advanced assistance block (fixed: adds weight/height, VN/EN, saves to victim profile)
+# -----------------------
 if st.session_state.get("basic_done") and st.session_state.get("wound_type"):
     st.header("Advanced assistance")
     st.markdown(f"**Wound type:** `{st.session_state.wound_type}`")
@@ -431,14 +433,26 @@ if st.session_state.get("basic_done") and st.session_state.get("wound_type"):
     with st.form("victim_form", clear_on_submit=True):
         st.subheader(t("victim_name"))
         name = st.text_input(t("victim_name"))
-        phone = st.text_input(t("phone"), help=("Emergency contact number" if st.session_state.lang == "en" else "Số liên hệ khẩn cấp"))
-        age = st.text_input(t("age"), help=("Enter an integer or leave blank" if st.session_state.lang == "en" else "Nhập số nguyên hoặc để trống"))
-        notes = st.text_area(
-                "Notes (allergies, meds, conscious status)" 
-                if st.session_state.lang == "en" 
-                else "Ghi chú thêm (dị ứng, thuốc đang dùng, tình trạng tỉnh táo)",
-                max_chars=500
-            )
+        phone = st.text_input(t("phone"),
+                              help=("Emergency contact number" if st.session_state.lang == "en" else "Số liên hệ khẩn cấp"))
+        age = st.text_input(t("age"),
+                            help=("Enter an integer or leave blank" if st.session_state.lang == "en" else "Nhập số nguyên hoặc để trống"))
+        # new localized weight / height fields
+        weight = st.text_input(
+            (t("weight") if "weight" in TRANSLATIONS else "Weight (kg)"),
+            help=(t("weight_help") if "weight_help" in TRANSLATIONS else ("Enter a number or leave blank" if st.session_state.lang == "en" else "Nhập số hoặc để trống"))
+        )
+        height = st.text_input(
+            (t("height") if "height" in TRANSLATIONS else "Height (cm)"),
+            help=(t("height_help") if "height_help" in TRANSLATIONS else ("Enter a number or leave blank" if st.session_state.lang == "en" else "Nhập số hoặc để trống"))
+        )
+
+        # localized notes label (falls back if t("notes") missing)
+        notes_label = t("notes") if "notes" in TRANSLATIONS else (
+            "Notes (allergies, meds, conscious status)" if st.session_state.lang == "en"
+            else "Ghi chú thêm (dị ứng, thuốc đang dùng, tình trạng tỉnh táo)"
+        )
+        notes = st.text_area(notes_label, max_chars=500)
 
         col1, col2 = st.columns(2)
         with col1:
@@ -446,38 +460,71 @@ if st.session_state.get("basic_done") and st.session_state.get("wound_type"):
         with col2:
             emergency_submit = st.form_submit_button(t("emergency_btn"))
 
-        # common validation
-        def validate_fields(age_str, phone_str):
+        # common validation (now parses weight & height as floats)
+        def validate_fields(age_str, phone_str, weight_str, height_str):
             age_v = None
+            weight_v = None
+            height_v = None
+
+            # age
             if age_str and age_str.strip():
                 try:
                     age_v = int(age_str.strip())
                 except Exception:
-                    st.error(("Age must be an integer (or leave blank)." if st.session_state.lang == "en" else "Tuổi phải là số nguyên (hoặc để trống)."))
+                    st.error(("Age must be an integer (or leave blank)." if st.session_state.lang == "en"
+                              else "Tuổi phải là số nguyên (hoặc để trống)."))
+                    st.stop()
+
+            # weight
+            if weight_str and weight_str.strip():
+                try:
+                    weight_v = float(weight_str.strip())
+                except Exception:
+                    st.error(("Weight must be a number (kg) or leave blank." if st.session_state.lang == "en"
+                              else "Cân nặng phải là số (kg) hoặc để trống."))
                     st.stop()
+
+            # height
+            if height_str and height_str.strip():
+                try:
+                    height_v = float(height_str.strip())
+                except Exception:
+                    st.error(("Height must be a number (cm) or leave blank." if st.session_state.lang == "en"
+                              else "Chiều cao phải là số (cm) hoặc để trống."))
+                    st.stop()
+
+            # phone
             phone_clean = phone_str.strip()
             if phone_clean and not phone_clean.replace("+", "").isdigit():
-                st.error(("Phone number should contain digits only." if st.session_state.lang == "en" else "Số điện thoại chỉ chứa chữ số."))
+                st.error(("Phone number should contain digits only." if st.session_state.lang == "en"
+                          else "Số điện thoại chỉ chứa chữ số."))
                 st.stop()
-            return age_v, phone_clean
+
+            return age_v, weight_v, height_v, phone_clean
 
         if (submit or emergency_submit) and not st.session_state.victim_saved:
-            age_v, phone_clean = validate_fields(age, phone)
+            # validate including weight/height
+            age_v, weight_v, height_v, phone_clean = validate_fields(age, phone, weight, height)
+
             victim = {
                 "name": name or "unknown",
                 "phone": phone_clean or None,
                 "age": age_v,
+                "weight_kg": weight_v,
+                "height_cm": height_v,
                 "notes": notes or "",
                 "wound_type": st.session_state.wound_type,
                 "saved_at": datetime.utcnow().isoformat() + "Z",
                 "emergency": bool(emergency_submit),
             }
+
             try:
                 record = save_victim_info(victim)
                 st.session_state.last_saved_record = record
                 st.session_state.victim_saved = True
                 if emergency_submit:
-                    st.error(("EMERGENCY recorded. Call local emergency services now." if st.session_state.lang == "en" else "Đã ghi nhận KHẨN CẤP. Gọi dịch vụ khẩn cấp ngay."))
+                    st.error(("EMERGENCY recorded. Call local emergency services now." if st.session_state.lang == "en"
+                              else "Đã ghi nhận KHẨN CẤP. Gọi dịch vụ khẩn cấp ngay."))
                 else:
                     st.success(("Victim info saved." if st.session_state.lang == "en" else "Đã lưu thông tin nạn nhân."))
             except Exception as e:
@@ -487,7 +534,11 @@ if st.session_state.get("basic_done") and st.session_state.get("wound_type"):
             # For non-emergency: ask OpenAI for guidance
             if submit and st.session_state.victim_saved:
                 system_prompt = make_system_prompt(st.session_state.wound_type, victim)
-                initial_user_msg = ("The patient has completed the basic steps. Provide the next, in-depth first aid guidance and any monitoring steps. Keep guidance concise and list immediate priorities first." if st.session_state.lang == "en" else "Bệnh nhân đã hoàn thành các bước cơ bản. Cung cấp hướng dẫn sơ cứu chi tiết tiếp theo và các bước giám sát. Giữ gọn và nêu ưu tiên trước.")
+                initial_user_msg = (
+                    "The patient has completed the basic steps. Provide the next, in-depth first aid guidance and any monitoring steps. Keep guidance concise and list immediate priorities first."
+                    if st.session_state.lang == "en"
+                    else "Bệnh nhân đã hoàn thành các bước cơ bản. Cung cấp hướng dẫn sơ cứu chi tiết tiếp theo và các bước giám sát. Giữ gọn và nêu ưu tiên trước."
+                )
                 with st.spinner(("Contacting assistant..." if st.session_state.lang == "en" else "Đang liên hệ trợ lý...")):
                     resp = query_openai(system_prompt, initial_user_msg)
                 if "error" in resp:
@@ -499,6 +550,7 @@ if st.session_state.get("basic_done") and st.session_state.get("wound_type"):
                     st.markdown("**" + t("assistant_guidance") + "**")
                     st.text_area(t("assistant_guidance"), value=assistant_text, height=250, key="assistant_output")
 
+
     # Chat follow-ups
     st.subheader(("Ask follow-up questions" if st.session_state.lang == "en" else "Hỏi thêm"))
     st.markdown(("Ask the assistant further questions related to the victim or care. Keep questions concise." if st.session_state.lang == "en" else "Hỏi trợ lý các câu liên quan tới nạn nhân hoặc chăm sóc. Giữ câu hỏi ngắn."))