added ability to create change and delete experiments

app.py

@@ -24,6 +24,7 @@ all_graph = False
 temp_mqtt_auto = None
 rpm_mqtt = None
 led_mqtt = None
+experiments = []
 
 # if st.session_state.get("channels", None) is not None:
 #     st.session_state["channel_a"] = int(st.session_state["channels"].get("A", 0))
@@ -79,7 +80,6 @@ else:
         submit_button = st.button("Reconnect")
 
 
-
 # Create the MQTT client
 # @st.cache_resource
 def create_mqtt_client(hivemq_host, hivemq_port, hivemq_username, hivemq_password):
@@ -108,11 +108,13 @@ def on_message_worker(client, userdata, message):
     global temp_mqtt_auto
     global rpm_mqtt
     global led_mqtt
+    global experiments
     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)
+    experiments = data.get("experiments", [])
 
 def on_message(client, userdata, message):
     payload = message.payload.decode("utf-8")
@@ -165,7 +167,7 @@ def get_running_jobs(client, reactor):
     }
     payload_str = json.dumps(payload)
     client.publish("pioreactor/control", payload_str)
-
+    global experiment
     experiment = None
     running = []
 
@@ -177,6 +179,15 @@ def get_running_jobs(client, reactor):
 
     while experiment is None and (time.time() - start_time) < timeout:
         time.sleep(1)
+
+    if experiment is None: 
+        st.error("Failed to get info")
+        st.session_state["experiment"] = None
+        time.sleep(3)
+        st.rerun()
+
+    if experiment != st.session_state["experiment"]:
+        st.session_state["experiment"] = experiment
     
     if st.session_state["jobs"]["temperature_automation"]:
         if temp_mqtt_auto == "thermostat":
@@ -192,10 +203,16 @@ def get_running_jobs(client, reactor):
     
     # st.session_state["channels"] = led_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))
+    if led_mqtt is not None:
+        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))
+
+    st.session_state["experiments"] = experiments 
+
+    for run in running:
+        st.session_state["jobs"][run] = True
 
 async def get_graph(placeholder):
     try:
@@ -245,7 +262,6 @@ if submit_button:
 
         if st.session_state["experiment"] is None:
             st.error("No experiment assigned to the reactor.")
-            st.stop()
         
         for run in running:
             st.session_state["jobs"][run] = True
@@ -266,6 +282,8 @@ if submit_button:
         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))
+
+        st.session_state["experiments"] = experiments
     
     # Save the client to the session state
     st.session_state["client"] = client
@@ -275,6 +293,102 @@ if submit_button:
 if st.session_state["experiment"] is not None:
     st.title(f"Experiment: {st.session_state['experiment']}")
 
+    with st.form("experiment_form"):
+        new_name = st.text_input("New Experiment Name")
+        submit_new_experiment = st.form_submit_button("Create New Experiment")
+
+    change_experiment_submit = False
+    delete_experiment = False
+    
+    with st.form("experiment_edit"):
+        if len(st.session_state["experiments"]) > 0:
+            experiment_change = st.selectbox("Select Experiment", st.session_state["experiments"])
+            cols = st.columns(2)
+            with cols[0]:
+                change_experiment_submit = st.form_submit_button("Change Experiment")
+            with cols[1]:
+                delete_experiment = st.form_submit_button("Delete Experiment")
+        else:
+            st.write("No experiments available.")
+            st.form_submit_button("Nothing Button")
+    
+    if submit_new_experiment:
+        payload = {
+            "command": "new_experiment",
+            "experiment": new_name,
+            "reactor": reactor
+        }
+        payload_str = json.dumps(payload)
+
+        st.session_state["client"].loop_start()
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        get_running_jobs(st.session_state["client"], reactor)
+
+        if new_name in st.session_state["experiments"]:
+            st.success("Experiment created successfully!")
+        else:
+            st.error("Failed to create experiment.")
+
+        
+        st.session_state["client"].loop_stop()
+
+        time.sleep(3)
+        st.rerun()
+    
+    if change_experiment_submit:
+        payload = {
+            "command": "change_experiment",
+            "experiment": st.session_state["experiment"],
+            "experiment_new": experiment_change,
+            "reactor": reactor
+        }
+        payload_str = json.dumps(payload)
+
+        st.session_state["client"].loop_start()
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        time.sleep(3)
+
+        get_running_jobs(st.session_state["client"], reactor)
+
+        if experiment_change == st.session_state["experiment"]:
+            st.success("Experiment changed successfully!")
+
+        else:
+            st.error("Failed to change experiment.")
+
+        st.info(f"Experiment name changed to {experiment_change} from {st.session_state['experiment']}")
+
+        st.session_state["client"].loop_stop()
+
+        time.sleep(3)
+        st.rerun()
+
+    if delete_experiment:
+        payload = {
+            "command": "delete_experiment",
+            "experiment": experiment_change,
+            "reactor": reactor
+        }
+        payload_str = json.dumps(payload)
+
+        time.sleep(1)
+
+        st.session_state["client"].loop_start()
+        st.session_state["client"].publish("pioreactor/control", payload_str, qos=1)
+
+        get_running_jobs(st.session_state["client"], reactor)
+
+        if experiment_change not in st.session_state["experiments"]:
+            st.success("Experiment deleted successfully!")
+        else:
+            st.error("Failed to delete experiment.")
+
+        st.session_state["client"].loop_stop()
+        time.sleep(3)
+        st.rerun()
+
     st.header("Running Jobs")
         
     cols = st.columns(5)
@@ -1176,59 +1290,60 @@ if st.session_state["experiment"] is not None:
             st.error("Failed to retrieve data.")
             st.stop()
 
-
-        for temp in temp_graph:
-            utc_time = datetime.strptime(temp["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
-            local_tz = pytz.timezone("America/New_York")
-            utc_time = utc_time.replace(tzinfo=pytz.utc)
-            local_time = utc_time.astimezone(local_tz)
-            temp["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
-
-        for od in od_graph:
-            utc_time = datetime.strptime(od["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
-            local_tz = pytz.timezone("America/New_York")
-            utc_time = utc_time.replace(tzinfo=pytz.utc)
-            local_time = utc_time.astimezone(local_tz)
-            od["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+        if len(temp_graph) != 0:
+
+            for temp in temp_graph:
+                utc_time = datetime.strptime(temp["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
+                local_tz = pytz.timezone("America/New_York")
+                utc_time = utc_time.replace(tzinfo=pytz.utc)
+                local_time = utc_time.astimezone(local_tz)
+                temp["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+                df = pd.DataFrame(temp_graph)
+                df = df.set_index("x")
+            placeholder.line_chart(df, x_label="Time", y_label="Temperature (°C)")
+            st.session_state["df"] = df
+
+        if len(od_graph) != 0:
+
+            for od in od_graph:
+                utc_time = datetime.strptime(od["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
+                local_tz = pytz.timezone("America/New_York")
+                utc_time = utc_time.replace(tzinfo=pytz.utc)
+                local_time = utc_time.astimezone(local_tz)
+                od["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+                df2 = pd.DataFrame(od_graph)
+                df2 = df2.set_index("x")
+            placeholder2.line_chart(df2, x_label="Time", y_label="OD Reading")
+            st.session_state["df2"] = df2
+
+        if len(norm_od_graph) != 0:
         
-        for norm_od in norm_od_graph:
-            utc_time = datetime.strptime(norm_od["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
-            local_tz = pytz.timezone("America/New_York")
-            utc_time = utc_time.replace(tzinfo=pytz.utc)
-            local_time = utc_time.astimezone(local_tz)
-            norm_od["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
-
-        for growth_rate in growth_rate_graph:
-            utc_time = datetime.strptime(growth_rate["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
-            local_tz = pytz.timezone("America/New_York")
-            utc_time = utc_time.replace(tzinfo=pytz.utc)
-            local_time = utc_time.astimezone(local_tz)
-            growth_rate["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+            for norm_od in norm_od_graph:
+                utc_time = datetime.strptime(norm_od["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
+                local_tz = pytz.timezone("America/New_York")
+                utc_time = utc_time.replace(tzinfo=pytz.utc)
+                local_time = utc_time.astimezone(local_tz)
+                norm_od["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+                df3 = pd.DataFrame(norm_od_graph)
+                df3 = df3.set_index("x")
+            placeholder3.line_chart(df3, x_label="Time", y_label="Normalized OD Reading")
+            st.session_state["df3"] = df3
+
+        if len(growth_rate_graph) != 0:
+
+            for growth_rate in growth_rate_graph:
+                utc_time = datetime.strptime(growth_rate["x"], "%Y-%m-%dT%H:%M:%S.%fZ")
+                local_tz = pytz.timezone("America/New_York")
+                utc_time = utc_time.replace(tzinfo=pytz.utc)
+                local_time = utc_time.astimezone(local_tz)
+                growth_rate["x"] = local_time.strftime("%Y-%m-%d %H:%M:%S")
+                df4 = pd.DataFrame(growth_rate_graph)
+                df4 = df4.set_index("x")
+            placeholder4.line_chart(df4, x_label="Time", y_label="Growth Rate")
+            st.session_state["df4"] = df4
 
         # Filter the data based on the amount of data requested
         # Data is in 4 minute intervals
-       
-        df = pd.DataFrame(temp_graph)
-        df = df.set_index("x")
-
-        df2 = pd.DataFrame(od_graph)
-        df2 = df2.set_index("x")
-
-        df3 = pd.DataFrame(norm_od_graph)
-        df3 = df3.set_index("x")
-
-        df4 = pd.DataFrame(growth_rate_graph)
-        df4 = df4.set_index("x")
-
-        placeholder.line_chart(df, x_label="Time", y_label="Temperature (°C)")
-        placeholder2.line_chart(df2, x_label="Time", y_label="OD Reading")
-        placeholder3.line_chart(df3, x_label="Time", y_label="Normalized OD Reading")
-        placeholder4.line_chart(df4, x_label="Time", y_label="Growth Rate")
-    
-        st.session_state["df"] = df
-        st.session_state["df2"] = df2
-        st.session_state["df3"] = df3
-        st.session_state["df4"] = df4
         
         time.sleep(REQUEST_INTERVAL)