Merge pull request #5 from ginidok/huggingface-deploy-action

cdss.py

@@ -277,7 +277,8 @@ def process_and_update(
     return (
         asdict(ps),
         *state_to_panels(ps),
-        str(ps.labs),
+        str(ps.labs),  # For labs_text
+        str(ps.labs),  # For labs_show
         interpretation,
         history_df,
         df_for_table,
@@ -293,11 +294,26 @@ def process_and_update(
 
 def state_to_panels(state: PatientState) -> Tuple:
     v = state.vitals
-    return (state.scenario, state.notes, v.sbp, v.dbp, v.hr, v.rr, v.temp_c, v.spo2)
+    return (
+        state.scenario,
+        state.patient_type,
+        state.notes,
+        v.sbp,
+        v.dbp,
+        v.hr,
+        v.rr,
+        v.temp_c,
+        v.spo2,
+    )
 
 
-def inject_scenario(tag: str, cdss_on: bool):
-    return process_and_update(SCENARIOS[tag](), pd.DataFrame(), "", cdss_on)
+def inject_scenario(tag: str, cdss_on: bool, history_df: pd.DataFrame, historic_text: str):
+    ps = SCENARIOS[tag]()
+    if historic_text: # Add a newline if text already exists
+        historic_text += f"\n[{datetime.now().strftime('%H:%M:%S')}] Scenario Injected: {ps.scenario}"
+    else:
+        historic_text = f"[{datetime.now().strftime('%H:%M:%S')}] Scenario Injected: {ps.scenario}"
+    return process_and_update(ps, history_df, historic_text, cdss_on)
 
 
 def manual_edit(
@@ -310,6 +326,7 @@ def manual_edit(
     notes,
     labs_text,
     cdss_on,
+    patient_type,
     current_state,
     history_df,
     historic_text,
@@ -320,7 +337,7 @@ def manual_edit(
         labs = {"raw": labs_text}
     ps = PatientState(
         current_state.get("scenario", "Manual"),
-        current_state.get("patient_type", "Mother"),
+        patient_type,
         notes,
         labs,
         Vitals(int(sbp), int(dbp), int(hr), int(rr), float(temp_c), int(spo2)),
@@ -411,6 +428,9 @@ with gr.Blocks(
             labs_show = gr.Textbox(label="Labs (Parsed)", interactive=False)
         with gr.Column(scale=1):
             with gr.Group():
+                patient_type_radio = gr.Radio(
+                    ["Mother", "Neonate", "Gyn"], label="Patient Type", value="Mother"
+                )
                 sbp = gr.Number(label="SBP")
                 dbp = gr.Number(label="DBP")
                 hr = gr.Number(label="HR")
@@ -431,6 +451,7 @@ with gr.Blocks(
     ui_outputs = [
         state,
         scenario_lbl,
+        patient_type_radio,
         notes,
         sbp,
         dbp,
@@ -438,6 +459,7 @@ with gr.Blocks(
         rr,
         temp_c,
         spo2,
+        labs_text,
         labs_show,
         interpretation,
         history_df,
@@ -461,6 +483,7 @@ with gr.Blocks(
         notes,
         labs_text,
         cdss_toggle,
+        patient_type_radio,
         state,
         history_df,
         historic_text,
@@ -482,7 +505,7 @@ with gr.Blocks(
             btn_C3,
         ],
     ):
-        btn.click(inject_scenario, [gr.State(tag), cdss_toggle], ui_outputs)
+        btn.click(inject_scenario, [gr.State(tag), cdss_toggle, history_df, historic_text], ui_outputs)
 
     csv_outputs = [history_df, df_view, bp_plot, hr_plot, rr_plot, temp_plot, spo2_plot]
     csv_file.change(load_csv, [csv_file, history_df], csv_outputs)