added to custom

app.py

@@ -429,6 +429,50 @@ def pinch_default(experiment, state):
     else:
         print("Invalid state")
         
+def relay(experiment, host, port, username, password, pioreactor, state):
+    global client_custom
+    if client_custom is None:
+        custom_client(host, port, username, password)
+    
+    if state == "start":
+        payload = {
+            "command": "start_relay",
+            "experiment": experiment,
+            "reactor": pioreactor
+        }
+        client_custom.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "stop":
+        payload = {
+            "command": "stop_relay",
+            "experiment": experiment,
+            "reactor": pioreactor
+        }
+        client_custom.publish(f"pioreactor/control", json.dumps(payload))
+    else:
+        print("Invalid state")
+        
+def pinch(experiment, host, port, username, password, pioreactor, state):
+    global client_custom
+    if client_custom is None:
+        custom_client(host, port, username, password)
+    
+    if state == "open":
+        payload = {
+            "command": "relay_on",
+            "experiment": experiment,
+            "reactor": pioreactor
+        }
+        client_custom.publish(f"pioreactor/control", json.dumps(payload))
+    elif state == "close":
+        payload = {
+            "command": "relay_off",
+            "experiment": experiment,
+            "reactor": pioreactor
+        }
+        client_custom.publish(f"pioreactor/control", json.dumps(payload))
+    else:
+        print("Invalid state")
+        
         
 def od_default(experiment, state):
     global client
@@ -1313,6 +1357,34 @@ with gr.Blocks() as demo:
                     )
                     
 
+        with gr.Blocks():
+            gr.Markdown("# Pinch Valve")
+            
+            gr.Markdown("Ensure that relay is running in Status. If not, start the automation before using the pinch valve.")
+            
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### Relay Automation")
+                    
+                    relay_state = gr.Radio(choices=["start", "stop"], label="State", value="start")
+                    relay_button = gr.Button("Send Command")
+                    
+                    relay_button.click(
+                        fn=relay,
+                        inputs=[host_input, port_input, username_input, password_input, pioreactor_input, experiment_input, relay_state]
+                    )
+                
+                with gr.Column():
+                    gr.Markdown("### Pinch Valve")
+                    
+                    pinch_state = gr.Radio(choices=["open", "close"], label="State", value="open")
+                    pinch_button = gr.Button("Send Command")
+                    
+                    pinch_button.click(
+                        fn=pinch,
+                        inputs=[host_input, port_input, username_input, password_input, pioreactor_input, experiment_input, pinch_state]
+                    )
+
 
 # Launch the interface
 demo.launch()