Added custom cred and stirring automation

app.py

@@ -2,28 +2,58 @@ import streamlit as st
 import paho.mqtt.client as mqtt
 import json
 import time
+import os
 
 
+experiment = None
+running = []
+
+if "experiment" not in st.session_state:
+    st.session_state["experiment"] = None
+    st.session_state["jobs"] = {
+        "temperature_automation": False, 
+        "growth_rate_calculating": False, 
+        "stirring": False, 
+        "od_reading": False, 
+        "led_automation": False
+    }
+    st.session_state["client"] = None
+
 # Create a form to get the MQTT broker details
 st.title("PioReactor MQTT Client")
-with st.form("mqtt_form"):
-    hivemq_host = st.text_input("HiveMQ Host", "9bb69fc9bf6b4092abc4c92d76e602cd.s1.eu.hivemq.cloud", type="password")
-    hivemq_username = st.text_input("HiveMQ Username", "Website", type="password")
-    hivemq_password = st.text_input("HiveMQ Password", "Abcabcabc1", type="password")
-    hivemq_port = st.number_input("HiveMQ Port", min_value=1, max_value=65535, step=1, value=8883)
-    reactor = st.text_input("PioReactor", "pio1")
 
-    submit_button = st.form_submit_button("Connect")
+# Create a button to choose whether or not to use custom credentials
+use_custom_credentials = st.checkbox("Use Custom Credentials")
+
+if use_custom_credentials:
+    with st.form("mqtt_form"):
+        # hivemq_host = st.text_input("HiveMQ Host", type="password")
+        # hivemq_username = st.text_input("HiveMQ Username", type="password")
+        # hivemq_password = st.text_input("HiveMQ Password", type="password")
+        # hivemq_port = st.number_input("HiveMQ Port", min_value=1, max_value=65535, step=1, value=8883)
+        # reactor = st.text_input("PioReactor")
+
+        hivemq_host = st.text_input("HiveMQ Host", "9bb69fc9bf6b4092abc4c92d76e602cd.s1.eu.hivemq.cloud",type="password")
+        hivemq_username = st.text_input("HiveMQ Username", "Website",type="password")
+        hivemq_password = st.text_input("HiveMQ Password", "Abcabcabc1",type="password")
+        hivemq_port = st.number_input("HiveMQ Port", min_value=1, max_value=65535, step=1, value=8883)
+        reactor = st.text_input("PioReactor", "pio1")
+
+        submit_button = st.form_submit_button("Connect")
+else:
+    hivemq_host = os.getenv("HIVEMQ_HOST")
+    hivemq_username = os.getenv("HIVEMQ_USERNAME")
+    hivemq_password = os.getenv("HIVEMQ_PASSWORD")
+    hivemq_port = os.getenv("HIVEMQ_PORT")
+    reactor = os.getenv("REACTOR")
+
+    # button to submit the form
+    if st.session_state["experiment"] is None:
+        submit_button = st.button("Connect")
+    else:
+        submit_button = st.button("Reconnect")
+
 
-experiment = None
-running = []
-jobs = {
-    "temperature_automation": False, 
-    "growth_rate_calculating": False, 
-    "stirring": False, 
-    "od_reading": False, 
-    "led_automation": False
-}
 
 # Create the MQTT client
 @st.cache_resource
@@ -78,37 +108,175 @@ if submit_button:
         while experiment is None and (time.time() - start_time) < timeout:
             time.sleep(1)
         
+        st.session_state["experiment"] = experiment
+
+        # Unsubscribe from the worker topic
+        # client.unsubscribe(f"pioreactor/{reactor}/worker")
+        # client.message_callback_remove(f"pioreactor/{reactor}/worker")
         client.loop_stop()
-        client.unsubscribe(f"pioreactor/{reactor}/worker")
 
-        if experiment is None:
+        if st.session_state["experiment"] is None:
             st.error("No experiment assigned to the reactor.")
             st.stop()
         
         for run in running:
-            jobs[run] = True
-
-if experiment is not None:
-    st.title(f"Experiment: {experiment}")
+            st.session_state["jobs"][run] = True
     
+    # Save the client to the session state
+    st.session_state["client"] = client
+
+
+# Display the experiment and running jobs if available
+if st.session_state["experiment"] is not None:
+    st.title(f"Experiment: {st.session_state['experiment']}")
+
+    st.header("Running Jobs")
+        
     cols = st.columns(5)
 
     with cols[0]:
         st.write("Temperature")
-        st.write(jobs["temperature_automation"])
+        st.write(st.session_state["jobs"]["temperature_automation"])
 
     with cols[1]:
         st.write("Growth Rate")
-        st.write(jobs["growth_rate_calculating"])
+        st.write(st.session_state["jobs"]["growth_rate_calculating"])
 
     with cols[2]:
         st.write("Stirring")
-        st.write(jobs["stirring"])
+        st.write(st.session_state["jobs"]["stirring"])
 
     with cols[3]:
         st.write("OD Reading")
-        st.write(jobs["od_reading"])
+        st.write(st.session_state["jobs"]["od_reading"])
 
     with cols[4]:
         st.write("LED")
-        st.write(jobs["led_automation"])    
+        st.write(st.session_state["jobs"]["led_automation"])
+
+    st.divider()
+
+    # Display the job details
+    st.header("Stirring")
+    if st.session_state["jobs"]["stirring"]:
+        with st.form("stirring_form"):
+            if "stirring_speed" not in st.session_state:
+                st.write("Current Stirring Speed Unknown")
+            else:
+                st.write(f"Current Stirring Speed: {st.session_state['stirring_speed']} RPM")
+            stirring_speed = st.slider("Stirring Speed", min_value=100, max_value=3000, step=50, value=st.session_state.get("stirring_speed", 500))
+            
+            cols = st.columns(2)
+            with cols[0]:
+                stirring = st.form_submit_button("Update Stirring Speed")
+            with cols[1]:
+                stop_stirring = st.form_submit_button("Stop Stirring")
+    else:
+        with st.form("stirring_form"):
+            stirring_speed = st.slider("Stirring Speed", min_value=100, max_value=3000, step=50, value=500)
+            stirring = st.form_submit_button("Start Stirring")
+        
+        stop_stirring = False
+
+    if stirring:
+        if st.session_state["jobs"]["stirring"]:
+            payload = {
+                "command": "update_stirring_rpm",
+                "reactor": reactor,
+                "experiment": st.session_state["experiment"],
+                "rpm": stirring_speed
+            }
+        else:
+            payload = {
+                "command": "start_stirring",
+                "experiment": st.session_state["experiment"],
+                "reactor": reactor,
+                "rpm": stirring_speed
+            }
+        
+        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 "stirring" in running and st.session_state["jobs"]["stirring"]:
+            st.success("Stirring speed updated successfully!")
+            st.session_state["stirring_speed"] = stirring_speed
+        elif "stirring" in running:
+            st.success("Stirring started successfully!")
+            st.session_state["jobs"]["stirring"] = True
+            st.session_state["stirring_speed"] = stirring_speed
+        else:
+            st.error("Failed to update stirring speed.")
+        
+        time.sleep(3)
+        st.rerun()
+
+    if stop_stirring:
+        payload = {
+            "command": "stop_stirring",
+            "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)
+        st.session_state["client"].loop_stop()
+
+        # Check if the stirring job is running
+        payload = {
+            "command": "get_worker",
+            "reactor": reactor
+        }
+
+        st.session_state["client"].subscribe(f"pioreactor/{reactor}/worker")
+        st.session_state["client"].message_callback_add(f"pioreactor/{reactor}/worker", on_message_worker)
+
+        experiment = None
+        running = []
+
+        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 "stirring" not in running:
+            st.success("Stirring stopped successfully!")
+            st.session_state["stirring_speed"] = None
+            st.session_state["jobs"]["stirring"] = False
+        else:
+            st.error("Failed to stop stirring.")
+
+        time.sleep(3)
+        st.rerun()