global client

app.py

@@ -285,7 +285,9 @@ def void_client():
         client_custom = None
         print("Client disconnected")
 
-def stirring_default(rpm, experiment, state, client):
+def stirring_default(rpm, experiment, state):
+    global client
+    
     if state == "start":
         payload = {
             "command": "start_stirring",
@@ -312,9 +314,11 @@ def stirring_default(rpm, experiment, state, client):
     else:
         print("Invalid state")
         
-def temp_automation_default(temperature, experiment, state, option, client):
+def temp_automation_default(temperature, experiment, state, option):
     payload_options = "thermostat" if option == "Heat To Temp" else "only_record_temperature"
     
+    global client
+    
     if state == "start":
         payload = {
             "command": "set_temperature_automation",
@@ -375,7 +379,7 @@ with gr.Blocks() as demo:
             
             st.click(
                 fn=stirring_default,
-                inputs=[rpm, experiment_input, stir_state, client]
+                inputs=[rpm, experiment_input, stir_state]
             )
         
         with gr.Blocks():
@@ -405,7 +409,7 @@ with gr.Blocks() as demo:
             
             temp.click(
                 fn=temp_automation_default,
-                inputs=[temperature_slider, experiment_input, temp_state, temp_option, client]
+                inputs=[temperature_slider, experiment_input, temp_state, temp_option]
             )