added led control

app.py

@@ -22,6 +22,14 @@ norm_od_graph = None
 growth_rate_graph = None
 all_graph = False
 temp_mqtt_auto = None
+rpm_mqtt = None
+led_mqtt = None
+
+if st.session_state.get("channels", None) is not None:
+    st.session_state["channel_a"] = int(st.session_state["channels"].get("A", 0))
+    st.session_state["channel_b"] = int(st.session_state["channels"].get("B", 0))
+    st.session_state["channel_c"] = int(st.session_state["channels"].get("C", 0))
+    st.session_state["channel_d"] = int(st.session_state["channels"].get("D", 0))
 
 REQUEST_INTERVAL = 60
 
@@ -98,9 +106,13 @@ def on_message_worker(client, userdata, message):
     global experiment
     global running
     global temp_mqtt_auto
+    global rpm_mqtt
+    global led_mqtt
     experiment = data.get("experiment", None)
     running = data.get("running", [])
     temp_mqtt_auto = data.get("temperature_automation", None)
+    rpm_mqtt = data.get("stirring", None)
+    led_mqtt = data.get("leds", None)
 
 def on_message(client, userdata, message):
     payload = message.payload.decode("utf-8")
@@ -166,6 +178,20 @@ def get_running_jobs(client, reactor):
     while experiment is None and (time.time() - start_time) < timeout:
         time.sleep(1)
     
+    if st.session_state["jobs"]["temperature_automation"]:
+        if temp_mqtt_auto == "thermostat":
+            st.session_state["temp_auto"] = "Heat To Temp"
+        elif temp_mqtt_auto == "only_record_temperature":
+            st.session_state["temp_auto"] = "Record Temp Only"
+
+        st.session_state["temp"] = None
+        
+    if st.session_state["jobs"]["stirring"]:
+        print(rpm_mqtt)
+        st.session_state["stirring_speed"] = rpm_mqtt
+    
+    st.session_state["channels"] = led_mqtt
+
 async def get_graph(placeholder):
     try:
         st.session_state["client"].loop_start()
@@ -226,6 +252,15 @@ if submit_button:
                 st.session_state["temp_auto"] = "Record Temp Only"
 
             st.session_state["temp"] = None
+        
+        if st.session_state["jobs"]["stirring"]:
+            print(rpm_mqtt)
+            st.session_state["stirring_speed"] = rpm_mqtt
+        
+        st.session_state["channel_a"] = int(led_mqtt.get("A", 0))
+        st.session_state["channel_b"] = int(led_mqtt.get("B", 0))
+        st.session_state["channel_c"] = int(led_mqtt.get("C", 0))   
+        st.session_state["channel_d"] = int(led_mqtt.get("D", 0))
     
     # Save the client to the session state
     st.session_state["client"] = client
@@ -901,6 +936,148 @@ if st.session_state["experiment"] is not None:
     
     # st.divider()
 
+    st.header("LEDS")
+
+    cols = st.columns(4)
+
+    with cols[0]:
+        st.write("Channel A")
+        with st.form("led_form_a"):
+            channel_a = st.number_input("Brightness", min_value=0, max_value=100, step=1, value=st.session_state.get("channel_a", 0), key="channel_a")
+            channel_a_submit = st.form_submit_button("Update Channel A")
+    
+    with cols[1]:
+        st.write("Channel B")
+        with st.form("led_form_b"):
+            channel_b = st.number_input("Brightness", min_value=0, max_value=100, step=1, value=st.session_state.get("channel_b", 0), key="channel_b")
+            channel_b_submit = st.form_submit_button("Update Channel B")
+
+    with cols[2]:
+        st.write("Channel C")
+        with st.form("led_form_c"):
+            channel_c = st.number_input("Brightness", min_value=0, max_value=100, step=1, value=st.session_state.get("channel_c", 0), key="channel_c") 
+            channel_c_submit = st.form_submit_button("Update Channel C")
+
+    with cols[3]:
+        st.write("Channel D")
+        with st.form("led_form_d"):
+            channel_d = st.number_input("Brightness", min_value=0, max_value=100, step=1, value=st.session_state.get("channel_d", 0), key="channel_d")
+            channel_d_submit = st.form_submit_button("Update Channel D")
+
+    
+    if channel_a_submit:
+        pass
+
+    if channel_b_submit:
+        payload = {
+            "command": "set_led_intensity",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "brightness": channel_b,
+            "led": "B"
+        }
+
+        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 LED automation job is running
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if channel_b == int(led_mqtt.get("B", 0)):
+            st.success("LED Intensity updated!")
+        else:
+            st.error("Failed to update LED Intensity.")
+            st.session_state["experiment"] = None
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+
+        st.session_state["experiment"] = experiment
+
+        time.sleep(3)
+        st.rerun()
+
+    if channel_c_submit:
+        payload = {
+            "command": "set_led_intensity",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "brightness": channel_c,
+            "led": "C"
+        }
+
+        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 LED automation job is running
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if channel_c == int(led_mqtt.get("C", 0)):
+            st.success("LED Intensity updated!")
+        else:
+            st.error("Failed to update LED Intensity.")
+            st.session_state["experiment"] = None
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+
+        st.session_state["experiment"] = experiment
+
+        time.sleep(3)
+        st.rerun()
+
+    if channel_d_submit:
+        payload = {
+            "command": "set_led_intensity",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "brightness": channel_d,
+            "led": "D"
+        }
+
+        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 LED automation job is running
+        get_running_jobs(st.session_state["client"], reactor)
+
+        st.session_state["client"].loop_stop()
+
+        if channel_d == int(led_mqtt.get("D", 0)):
+            st.success("LED Intensity updated!")
+        else:
+            st.error("Failed to update LED Intensity.")
+            st.session_state["experiment"] = None
+        
+        for run in running:
+            st.session_state["jobs"][run] = True
+
+        st.session_state["experiment"] = experiment
+
+        time.sleep(3)
+        st.rerun()
+
+    st.divider()
+
     st.header("Graphs")