test of status

app.py

@@ -13,6 +13,13 @@ PIOREACTOR = os.environ.get("pioreactor")
 client = None
 client_custom = None
 
+experiment = None
+running = []
+temp_mqtt_auto = None
+rpm_mqtt = None
+led_mqtt = None
+experiments = []
+
 
 def on_connect(client, userdata, flags, rc):
     print("Connected with result code "+str(rc))
@@ -481,6 +488,27 @@ def cycle_media_default(cycle, experiment):
     }
     client.publish(f"pioreactor/control", json.dumps(payload))
 
+def get_status_default(exp):
+    global client
+    client.subscribe(f"pioreactor/{PIOREACTOR}/worker")
+    client.on_message = on_message_worker
+    global experiment
+    experiment = None
+    
+    payload = {
+        "command": "get_worker",
+        "experiment": exp,
+        "reactor": PIOREACTOR
+    }
+    
+    client.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)
+
 # Define the interface components
 with gr.Blocks() as demo:
     with gr.Tab("Default"):
@@ -489,6 +517,22 @@ with gr.Blocks() as demo:
         
         client = create_client(HOST, PORT, USERNAME, PASSWORD)
         
+        with gr.Blocks():
+            gr.Markdown("# Status")
+            get_status = gr.Button("Get Status")
+            
+            gr.Textbox(label="Experiment", value=experiment, readonly=True)
+            gr.Textbox(label="Running", value=running, readonly=True)
+            gr.Textbox(label="Temperature Automation", value=temp_mqtt_auto, readonly=True)
+            gr.Textbox(label="RPM", value=rpm_mqtt, readonly=True)
+            gr.Textbox(label="LEDs", value=led_mqtt, readonly=True)
+            gr.Textbox(label="Experiments", value=experiments, readonly=True)
+            
+            get_status.click(
+                fn=get_status_default,
+                inputs=[experiment_input]
+            )
+        
         with gr.Blocks():
             gr.Markdown("# Stirring")
             rpm = gr.Slider(minimum=0, maximum=2000, step=50, label="RPM")