on error force reconnect

app.py

@@ -328,6 +328,8 @@ if st.session_state["experiment"] is not None:
             st.session_state["stirring_speed"] = stirring_speed
         else:
             st.error("Failed to update stirring speed.")
+            st.session_state["experiment"] = None
+            st.session_state["jobs"]["stirring_speed"] = None
         
         time.sleep(3)
         st.rerun()
@@ -516,6 +518,8 @@ if st.session_state["experiment"] is not None:
             st.session_state["temp"] = temp
         else:
             st.error("Failed to update temperature automation speed.")
+            st.session_state["experiment"] = None
+            st.session_state["jobs"]["temperature_automation"] = None
         
         time.sleep(3)
         st.rerun()
@@ -563,9 +567,9 @@ if st.session_state["experiment"] is not None:
             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.")
+            st.session_state["experiment"] = None
         
         time.sleep(3)
         st.rerun()
@@ -616,6 +620,7 @@ if st.session_state["experiment"] is not None:
             st.session_state["jobs"]["od_reading"] = True
         else:
             st.error("Failed to start OD Reading.")
+            st.session_state["experiment"] = None
         
         for run in running:
             st.session_state["jobs"][run] = True
@@ -648,6 +653,7 @@ if st.session_state["experiment"] is not None:
             st.session_state["jobs"]["od_reading"] = False
         else:
             st.error("Failed to stop OD Reading.")
+            st.session_state["experiment"] = None
         
         for run in running:
             st.session_state["jobs"][run] = True
@@ -680,6 +686,7 @@ if st.session_state["experiment"] is not None:
             st.session_state["jobs"]["growth_rate_calculating"] = True
         else:
             st.error("Failed to start Growth Rate Calculation.")
+            st.session_state["experiment"] = None
         
         for run in running:
             st.session_state["jobs"][run] = True
@@ -712,6 +719,7 @@ if st.session_state["experiment"] is not None:
             st.session_state["jobs"]["growth_rate_calculating"] = False
         else:
             st.error("Failed to stop Growth Rate Calculation.")
+            st.session_state["experiment"] = None
         
         for run in running:
             st.session_state["jobs"][run] = True