added od and growth control

app.py

@@ -129,6 +129,27 @@ async def run_mqtt_client(client):
     client.loop_start()
     await periodic_temp_reading(client, reactor, st.session_state["experiment"])
 
+def get_running_jobs(client, reactor):
+    payload = {
+        "command": "get_worker",
+        "reactor": reactor
+    }
+    payload_str = json.dumps(payload)
+    client.publish("pioreactor/control", payload_str)
+
+    experiment = None
+    running = []
+
+    client.subscribe(f"pioreactor/{reactor}/worker")
+    client.message_callback_add(f"pioreactor/{reactor}/worker", on_message_worker)
+
+    timeout = 10
+    start_time = time.time()
+
+    while experiment is None and (time.time() - start_time) < timeout:
+        time.sleep(1)
+
+
 if submit_button:
     client = create_mqtt_client(hivemq_host, hivemq_port, hivemq_username, hivemq_password)
 
@@ -524,6 +545,157 @@ if st.session_state["experiment"] is not None:
 
     st.divider()
 
+    cols = st.columns(2)
+
+    start_od = False
+    stop_od = False
+    start_growth = False
+    stop_growth = False
+
+    with cols[0]:
+        st.header("OD Reading")
+        if st.session_state["jobs"]["od_reading"]:
+            stop_od = st.button("Stop OD Reading")
+        else:
+            start_od = st.button("Start OD Reading")
+    
+    with cols[1]:
+        st.header("Growth Rate")
+        if st.session_state["jobs"]["growth_rate_calculating"]:
+            stop_growth = st.button("Stop Growth Rate Calculation")
+        else:
+            start_growth = st.button("Start Growth Rate Calculation")
+
+    if start_od:
+        payload = {
+            "command": "start_od_reading",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"]
+        }
+        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 OD reading job is running
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if "od_reading" in running:
+            st.success("OD Reading started successfully!")
+            st.session_state["jobs"]["od_reading"] = True
+        else:
+            st.error("Failed to start OD Reading.")
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+        
+        st.session_state["experiment"] = experiment
+        
+        time.sleep(3)
+        st.rerun()
+    
+    if stop_od:
+        payload = {
+            "command": "stop_od_reading",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"]
+        }
+        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 OD reading job has stopped
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if "od_reading" not in running:
+            st.success("OD Reading stopped successfully!")
+            st.session_state["jobs"]["od_reading"] = False
+        else:
+            st.error("Failed to stop OD Reading.")
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+        
+        st.session_state["experiment"] = experiment
+        
+        time.sleep(3)
+        st.rerun()
+    
+    if start_growth:
+        payload = {
+            "command": "start_growth_rate",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"]
+        }
+        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 growth rate job is running
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if "growth_rate_calculating" in running:
+            st.success("Growth Rate Calculation started successfully!")
+            st.session_state["jobs"]["growth_rate_calculating"] = True
+        else:
+            st.error("Failed to start Growth Rate Calculation.")
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+        
+        st.session_state["experiment"] = experiment
+        
+        time.sleep(3)
+        st.rerun()
+
+    if stop_growth:
+        payload = {
+            "command": "stop_growth_rate",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"]
+        }
+        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 growth rate job has stopped
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if "growth_rate_calculating" not in running:
+            st.success("Growth Rate Calculation stopped successfully!")
+            st.session_state["jobs"]["growth_rate_calculating"] = False
+        else:
+            st.error("Failed to stop Growth Rate Calculation.")
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+        
+        st.session_state["experiment"] = experiment
+        
+        time.sleep(3)
+        st.rerun()
+
+    st.divider()
+
     # Display the temperature graph 
     # Using asyncio to run the loop in the background and periodically update the graph