duplicate method

app.py

@@ -10,13 +10,6 @@ USERNAME = os.environ.get("username")
 PASSWORD = os.environ.get("password")
 PIOREACTOR = os.environ.get("pioreactor")
 
-
-HOST_gr = gr.Textbox(label="Host", value=HOST, visible=False)
-PORT_gr = gr.Number(label="Port", value=PORT, visible=False)
-USERNAME_gr = gr.Textbox(label="Username", value=USERNAME, visible=False)
-PASSWORD_gr = gr.Textbox(label="Password", value=PASSWORD, visible=False)
-PIOREACTOR_gr = gr.Textbox(label="Pioreactor", value=PIOREACTOR, visible=False)
-
 client = None
 client_custom = None
 
@@ -292,12 +285,85 @@ def void_client():
         client_custom = None
         print("Client disconnected")
 
+def stirring_default(rpm, experiment, state, client):
+    if state == "start":
+        payload = {
+            "command": "start_stirring",
+            "experiment": experiment,
+            "reactor": PIOREACTOR,
+            "rpm": rpm
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "stop":
+        payload = {
+            "command": "stop_stirring",
+            "experiment": experiment,
+            "reactor": PIOREACTOR
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "update":
+        payload = {
+            "command": "update_stirring_rpm",
+            "experiment": experiment,
+            "reactor": PIOREACTOR,
+            "rpm": rpm
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    else:
+        print("Invalid state")
+        
+def temp_automation_default(temperature, experiment, state, option, client):
+    payload_options = "thermostat" if option == "Heat To Temp" else "only_record_temperature"
+    
+    if state == "start":
+        payload = {
+            "command": "set_temperature_automation",
+            "experiment": experiment,
+            "reactor": PIOREACTOR,
+            "temp": temperature,
+            "automation": payload_options
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "stop":
+        payload = {
+            "command": "temp_update",
+            "experiment": experiment,
+            "reactor": PIOREACTOR,
+            "settings": {
+                "$state": "disconnected"
+            }
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "update":
+        payload = {
+            "command": "temp_update",
+            "experiment": experiment,
+            "reactor": PIOREACTOR,
+            "settings": {
+                "target_temperature": temperature
+            }
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "restart":
+        payload = {
+            "command": "temp_restart",
+            "experiment": experiment,
+            "reactor": PIOREACTOR,
+            "automation": payload_options,
+            "temp": temperature
+        }
+        client.publish(f"pioreactor/control", json.dumps(payload))
+    else:
+        print("Invalid state")
+
 # Define the interface components
 with gr.Blocks() as demo:
     with gr.Tab("Default"):
         experiment_input = gr.Textbox(label="Experiment", value="Ed")
         experiment_input.input(fn=void_client)
         
+        client = create_client(HOST, PORT, USERNAME, PASSWORD)
+        
         with gr.Blocks():
             gr.Markdown("# Stirring")
             rpm = gr.Slider(minimum=0, maximum=2000, step=50, label="RPM")
@@ -308,8 +374,8 @@ with gr.Blocks() as demo:
                 st = gr.Button("Send Command")
             
             st.click(
-                fn=stirring,
-                inputs=[rpm, HOST_gr, PORT_gr, USERNAME_gr, PASSWORD_gr, PIOREACTOR_gr, experiment_input, stir_state]
+                fn=stirring_default,
+                inputs=[rpm, experiment_input, stir_state, client]
             )
         
         with gr.Blocks():
@@ -338,8 +404,8 @@ with gr.Blocks() as demo:
             )
             
             temp.click(
-                fn=temp_automation,
-                inputs=[temperature_slider, HOST_gr, PORT_gr, USERNAME_gr, PASSWORD_gr, PIOREACTOR_gr, experiment_input, temp_state, temp_option]
+                fn=temp_automation_default,
+                inputs=[temperature_slider, experiment_input, temp_state, temp_option, client]
             )