added temp automation control

app.py

@@ -280,3 +280,199 @@ if st.session_state["experiment"] is not None:
 
         time.sleep(3)
         st.rerun()
+
+    st.divider()
+
+    st.header("Temperature Automation")
+
+    if st.session_state["jobs"]["temperature_automation"]:
+        temp = None
+        if st.session_state.get("temp_auto", None) is None:
+            st.write("Current Temperature Automation Unknown, changes will result in restart of the automation to new settings.")
+            temp_auto = st.selectbox("Temperature Automation", ["Record Temp Only", "Heat To Temp"], placeholder="UNKNOWN")
+            with st.form("temperature_form"):
+                if temp_auto == "Heat To Temp":
+                    temp = st.slider("Temperature", min_value=0.0, max_value=60.0, step=0.1, value=30.0)
+                    cols = st.columns(2)
+                    with cols[0]:
+                        temp_submit = st.form_submit_button("Restart Temperature Automation")
+                    with cols[1]:
+                        temp_stop = st.form_submit_button("Stop Temperature Automation")
+                elif temp_auto == "Record Temp Only":
+                    cols = st.columns(2)
+                    with cols[0]:
+                        temp_submit = st.form_submit_button("Restart Temperature Automation")
+                    with cols[1]:
+                        temp_stop = st.form_submit_button("Stop Temperature Automation")
+        else:
+            if st.session_state["temp_auto"] == "Heat To Temp":
+                index = 1
+            elif st.session_state["temp_auto"] == "Record Temp Only":
+                index = 0
+            temp_auto = st.selectbox("Temperature Automation", ["Record Temp Only", "Heat To Temp"], index=index)
+            with st.form("temperature_form"):
+                st.write(f"Current Temperature Automation: {st.session_state['temp_auto']}")
+                if temp_auto == "Heat To Temp":
+                    if st.session_state.get("temp", None) is not None:
+                        st.write(f"Current Temperature: {st.session_state['temp']} °C")
+                    temp = st.slider("Temperature", min_value=0.0, max_value=60.0, step=0.1, value=st.session_state["temp"] if st.session_state["temp"] is not None else 30.0)
+                    cols = st.columns(2)
+                    with cols[0]:
+                        temp_submit = st.form_submit_button("Update Temperature Automation")
+                    with cols[1]:
+                        temp_stop = st.form_submit_button("Stop Temperature Automation")
+                elif temp_auto == "Record Temp Only":
+                    cols = st.columns(2)
+                    with cols[0]:
+                        temp_submit = st.form_submit_button("Update Temperature Automation", disabled=(st.session_state["temp_auto"] == "Record Temp Only"))
+                    with cols[1]:
+                        temp_stop = st.form_submit_button("Stop Temperature Automation")
+
+    else:
+        temp_auto = st.selectbox("Temperature Automation", ["Record Temp Only", "Heat To Temp"], placeholder="Select an option")
+        with st.form("temperature_form"):
+            if temp_auto == "Heat To Temp":
+                temp = st.slider("Temperature", min_value=0.0, max_value=60.0, step=0.1, value=30.0)
+                temp_submit = st.form_submit_button("Start Temperature Automation")
+                temp_stop = False
+            elif temp_auto == "Record Temp Only":
+                temp_submit = st.form_submit_button("Start Temperature Automation")
+                temp_stop = False
+    
+    if temp_submit:
+        # Convert the temperature automation to the payload
+        if temp_auto == "Heat To Temp":
+                payload_auto = "thermostat"
+        elif temp_auto == "Record Temp Only":
+            payload_auto = "only_record_temperature"
+
+        # Update the temperature automation
+        if st.session_state["jobs"]["temperature_automation"]:
+            #Check if the temperature automation is the same as the current one
+            if temp_auto == st.session_state.get("temp_auto", None):
+                # Same automation, update the temperature
+                # Update the temperature automation
+                payload = {
+                    "command": "temp_update",
+                    "reactor": reactor,
+                    "experiment": st.session_state["experiment"],
+                    "settings": {
+                        "target_temperature": temp
+                    }
+                }
+            else:
+                # Different automation, restart the automation
+                payload = {
+                    "command": "temp_restart",
+                    "reactor": reactor,
+                    "experiment": st.session_state["experiment"],
+                    "automation": payload_auto,
+                    "temp": temp
+                }
+        else:
+            # Start the temperature automation
+            payload = {
+                "command": "set_temperature_automation",
+                "reactor": reactor,
+                "experiment": st.session_state["experiment"],
+                "automation": payload_auto,
+                "temp": temp
+            }
+        
+        payload_str = json.dumps(payload)
+
+        # Start looping
+        st.session_state["client"].loop_start()
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        time.sleep(1)
+
+        # Check if the stirring job is running
+        payload = {
+            "command": "get_worker",
+            "reactor": reactor
+        }
+
+        experiment = None
+        running = []
+
+        st.session_state["client"].subscribe(f"pioreactor/{reactor}/worker")
+        st.session_state["client"].message_callback_add(f"pioreactor/{reactor}/worker", on_message_worker)
+
+        payload_str = json.dumps(payload)
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+        
+        timeout = 10
+        start_time = time.time()
+
+        while experiment is None and (time.time() - start_time) < timeout:
+            time.sleep(1)
+
+        st.session_state["client"].loop_stop()
+        
+        if "temperature_automation" in running and st.session_state["jobs"]["temperature_automation"]:
+            st.success("Temperature automation updated!")
+            st.session_state["temp_auto"] = temp_auto
+            st.session_state["temp"] = temp
+        elif "temperature_automation" in running:
+            st.success("Temperature automation started successfully!")
+            st.session_state["jobs"]["temperature_automation"] = True
+            st.session_state["temp_auto"] = temp_auto
+            st.session_state["temp"] = temp
+        else:
+            st.error("Failed to update temperature automation speed.")
+        
+        time.sleep(3)
+        st.rerun()
+
+    if temp_stop:
+        payload = {
+            "command": "temp_update",
+            "reactor": reactor,
+            "experiment": st.session_state["experiment"],
+            "settings": {
+                "$state": "disconnect"
+            }
+        }
+
+        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 temperature automation job has stopped
+        payload = {
+            "command": "get_worker",
+            "reactor": reactor
+        }
+
+        experiment = None
+        running = []
+
+        st.session_state["client"].subscribe(f"pioreactor/{reactor}/worker")
+        st.session_state["client"].message_callback_add(f"pioreactor/{reactor}/worker", on_message_worker)
+
+        payload_str = json.dumps(payload)
+
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        timeout = 10
+        start_time = time.time()
+
+        while experiment is None and (time.time() - start_time) < timeout:
+            time.sleep(1)
+
+        if "temperature_automation" not in running:
+            st.success("Temperature automation stopped successfully!")
+            st.session_state["temp_auto"] = None
+            st.session_state["temp"] = None
+            st.session_state["jobs"]["temperature_automation"] = False
+        else:
+            st.error("Failed to stop temperature automation.")
+        
+        time.sleep(3)
+        st.rerun()
+
+    st.divider()