Update pages/02_Workflow_UI.py

pages/02_Workflow_UI.py

@@ -231,9 +231,10 @@ _case_summary_header(case)
 tab_pcp, tab_spec = st.tabs(["PCP Referral", "Specialist Review"])
 
 # ===== PCP Referral Tab =====
+# Inside the PCP Referral tab (replace your existing block):
+
 with tab_pcp:
     is_draft = case.get("status") == "draft"
-
     p = case.get("patient", {}) or {}
     c = case.get("consult", {}) or {}
 
@@ -241,14 +242,23 @@ with tab_pcp:
     with c1:
         patient_name = st.text_input("Patient Name", value=p.get("name", ""), disabled=not is_draft, key=f"pt_name_{selected}")
         consult_q = st.text_area("Consult Question", value=c.get("question", ""), height=80, disabled=not is_draft, key=f"q_{selected}")
-        history = st.text_area("Pertinent History", value=c.get("history", ""), height=120, disabled=not is_draft, key=f"h_{selected}")
+        clinical_summary = st.text_area(
+            "Clinical Summary",
+            value=c.get("summary", "") or c.get("history", ""),
+            height=160,
+            placeholder=("Example: 72yo man with NYHA class III HFrEF (EF 30%). "
+                         "BP 110/70 HR 62. Echo: mild MR, no effusion. "
+                         "BNP 230; here for med optimization."),
+            disabled=not is_draft,
+            key=f"summary_{selected}",
+        )
     with c2:
         age = st.number_input("Age", min_value=0, max_value=120, value=int(p.get("age") or 0), disabled=not is_draft, key=f"age_{selected}")
-        sex = st.selectbox("Sex", options=["F", "M", "Other", ""], index=["F","M","Other",""].index(p.get("sex") or ""), disabled=not is_draft, key=f"sex_{selected}")
+        sex = st.selectbox("Sex", ["F", "M", "Other", ""], index=["F","M","Other",""].index(p.get("sex") or ""), disabled=not is_draft, key=f"sex_{selected}")
         specialty = st.text_input("Specialty", value=c.get("specialty", "") or "", disabled=not is_draft, key=f"spec_{selected}")
     with c3:
-        meds = st.text_area("Medications", value=c.get("medications", ""), height=120, disabled=not is_draft, key=f"meds_{selected}")
-        labs = st.text_area("Labs", value=c.get("labs", ""), height=120, disabled=not is_draft, key=f"labs_{selected}")
+        medications = st.text_area("Medications", value=c.get("medications", ""), height=100, disabled=not is_draft, key=f"meds_{selected}")
+        labs = st.text_area("Labs / Key Values", value=c.get("labs", ""), height=100, disabled=not is_draft, key=f"labs_{selected}")
 
     consent = bool(c.get("consent_obtained", False))
     consent = st.checkbox("Patient consent obtained", value=consent, disabled=not is_draft, key=f"consent_{selected}")
@@ -256,8 +266,7 @@ with tab_pcp:
     col = st.columns([1,1,6])[0]
     with col:
         if is_draft:
-            submit_clicked = st.button("Submit Referral", type="primary", use_container_width=True, key=f"submit_{selected}")
-            if submit_clicked:
+            if st.button("Submit Referral", type="primary", use_container_width=True, key=f"submit_{selected}"):
                 if not consent:
                     st.error("Consent is required to submit.")
                 else:
@@ -266,8 +275,8 @@ with tab_pcp:
                         "consult": {
                             "specialty": specialty.strip(),
                             "question": consult_q.strip(),
-                            "history": history.strip(),
-                            "medications": meds.strip(),
+                            "summary": clinical_summary.strip(),
+                            "medications": medications.strip(),
                             "labs": labs.strip(),
                             "consent_obtained": True,
                         },
@@ -279,6 +288,7 @@ with tab_pcp:
         else:
             st.info("Referral is not editable (status is In Review or Completed).")
 
+
 # ===== Specialist Review Tab =====
 with tab_spec:
     status = case.get("status")
@@ -366,6 +376,11 @@ with tab_spec:
                 # Persist generated draft and mark unmodified so chips/guidelines show
                 store.update_case(selected, {"soap_draft": {"subjective": subj, "objective": obj, "assessment": assess, "plan": plan}, "explainability": {"manually_modified": False}})
 
+                # Refresh case so UI shows the new SOAP fields immediately
+                case = _get_case(selected)
+                soap = case.get("soap_draft", {})
+                st.success("SOAP draft generated and saved. You can review it below.")
+
         c1, c2 = st.columns(2)
         with c1:
             subj_new = st.text_area("Subjective", value=soap.get("subjective",""), height=140, disabled=read_only, key=f"subj_{selected}")