dosing actions

app.py

@@ -1228,6 +1228,141 @@ if st.session_state["experiment"] is not None:
 
     st.divider()
 
+    st.header("Dosing")
+
+    cols = st.columns(3)
+
+    with cols[0]:
+        st.subheader("Add Media")
+        with st.form("media_form"):
+            media_ml = st.number_input("Media (mL)", min_value=0, step=1, value=0)
+            media_submit = st.form_submit_button("Add Media")
+    
+    with cols[1]:
+        st.subheader("Remove Waste")
+        with st.form("waste_form"):
+            waste_ml = st.number_input("Waste (mL)", min_value=0, step=1, value=0)
+            waste_submit = st.form_submit_button("Remove Waste")
+    
+    with cols[2]:
+        st.subheader("Cycle Media")
+        with st.form("cycle_form"):
+            cycle_s = st.number_input("Cycle (s)", min_value=0, step=1, value=0)
+            cycle_submit = st.form_submit_button("Cycle Media")
+    
+    if media_submit:
+        payload = {
+            "command": "pump_add_media",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "volume": media_ml
+        }
+
+        payload_str = json.dumps(payload)
+
+        st.session_state["client"].loop_start()
+
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        time.sleep(1)
+
+        # Check if the dosing job is running
+        # get_running_jobs(st.session_state["client"], reactor)
+
+        # st.session_state["client"].loop_stop()
+
+        # if "dosing" in running:
+        #     st.success("Media added successfully!")
+        # else:
+        #     st.error("Failed to add media.")
+        #     st.session_state["experiment"] = None
+        
+        # for run in running:
+        #     st.session_state["jobs"][run] = True
+
+        # st.session_state["experiment"] = experiment
+
+        st.success("Media added successfully!")
+
+        time.sleep(3)
+        st.rerun()
+    
+    if waste_submit:
+        payload = {
+            "command": "pump_remove_waste",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "volume": waste_ml
+        }
+
+        payload_str = json.dumps(payload)
+
+        st.session_state["client"].loop_start()
+
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        time.sleep(1)
+
+        # Check if the dosing job is running
+        # get_running_jobs(st.session_state["client"], reactor)
+
+        # st.session_state["client"].loop_stop()
+
+        # if "dosing" in running:
+        #     st.success("Waste removed successfully!")
+        # else:
+        #     st.error("Failed to remove waste.")
+        #     st.session_state["experiment"] = None
+        
+        # for run in running:
+        #     st.session_state["jobs"][run] = True
+
+        # st.session_state["experiment"] = experiment
+
+        st.success("Waste removed successfully!")
+
+        time.sleep(3)
+        st.rerun()
+
+    if cycle_submit:
+        payload = {
+            "command": "pump_cycle_media",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "duration": cycle_s
+        }
+
+        payload_str = json.dumps(payload)
+
+        st.session_state["client"].loop_start()
+
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        time.sleep(1)
+
+        # Check if the dosing job is running
+        # get_running_jobs(st.session_state["client"], reactor)
+
+        # st.session_state["client"].loop_stop()
+
+        # if "dosing" in running:
+        #     st.success("Media cycled successfully!")
+        # else:
+        #     st.error("Failed to cycle media.")
+        #     st.session_state["experiment"] = None
+        
+        # for run in running:
+        #     st.session_state["jobs"][run] = True
+
+        # st.session_state["experiment"] = experiment
+
+        st.success("Media cycled successfully!")
+
+        time.sleep(3)
+        st.rerun()
+
+    st.divider()
+
     st.header("Graphs")