add for custom

app.py

@@ -488,6 +488,34 @@ def cycle_media_default(cycle, experiment):
     }
     client.publish(f"pioreactor/control", json.dumps(payload))
 
+def get_status(host, port, username, password, pioreactor, exp):
+    global client_custom
+    if client_custom is None:
+        custom_client(host, port, username, password)
+    
+    client_custom.subscribe(f"pioreactor/{pioreactor}/worker")
+    client_custom.on_message = on_message_worker
+    global experiment
+    experiment = None
+    
+    payload = {
+        "command": "get_worker",
+        "experiment": exp,
+        "reactor": pioreactor
+    }
+    
+    client_custom.publish(f"pioreactor/control", json.dumps(payload))
+    
+    timeout = 10
+    start = time.time()
+    
+    while experiment is None and (time.time() - start) < timeout:
+        time.sleep(0.1)
+        
+    client_custom.unsubscribe(f"pioreactor/{pioreactor}/worker")
+        
+    return experiment, running, temp_mqtt_auto, rpm_mqtt, led_mqtt, experiments
+
 def get_status_default(exp):
     global client
     client.subscribe(f"pioreactor/{PIOREACTOR}/worker")
@@ -509,6 +537,8 @@ def get_status_default(exp):
     while experiment is None and (time.time() - start) < timeout:
         time.sleep(0.1)
         
+    client.unsubscribe(f"pioreactor/{PIOREACTOR}/worker")
+        
     return experiment, running, temp_mqtt_auto, rpm_mqtt, led_mqtt, experiments
 
 # Define the interface components
@@ -705,6 +735,25 @@ with gr.Blocks() as demo:
             pioreactor_input.input(fn=void_client)
             experiment_input.input(fn=void_client)
 
+        with gr.Blocks():
+            gr.Markdown("# Status")
+            get_status = gr.Button("Get Status")
+            
+            with gr.Row():
+                experiment_output = gr.Textbox(label="Experiment", interactive=False)
+                running_output = gr.Textbox(label="Running", interactive=False)
+            with gr.Row():
+                temp_output = gr.Textbox(label="Temperature Automation", interactive=False)
+                rpm_output = gr.Textbox(label="RPM", interactive=False) 
+                led_output = gr.Textbox(label="LEDs", interactive=False)
+                experiments_output = gr.Textbox(label="Experiments", interactive=False)
+            
+            get_status.click(
+                fn=get_status_default,
+                inputs=[experiment_input],
+                outputs=[experiment_output, running_output, temp_output, rpm_output, led_output, experiments_output]
+            )
+
         with gr.Blocks():
             gr.Markdown("# Stirring")
             rpm_input = gr.Slider(minimum=0, maximum=2000, step=50, label="RPM")