a

app.py

@@ -3,19 +3,28 @@ import os
 import gradio as gr
 import json
 import time
+import threading
 
+# Environment Variables
 HOST = os.environ.get("host")
 PORT = int(os.environ.get("port"))
 USERNAME = os.environ.get("username")
 PASSWORD = os.environ.get("password")
 
+# Global variables to store received data
+latest_data = {
+    "bed_temperature": "N/A",
+    "nozzle_temperature": "N/A",
+    "status": "N/A",
+    "update_time": "Waiting for data...",
+}
+
+# MQTT Client setup
 client = None
-bed_temp = gr.Number()
-nozzle_temp = gr.Number()
-status = gr.Textbox()
-update_time = gr.Textbox()
+response_topic = None  # Will be set dynamically
 
 def create_client(host, port, username, password):
+    global client
     client = mqtt.Client()
     client.username_pw_set(username, password)
     client.tls_set(tls_version=mqtt.ssl.PROTOCOL_TLS)
@@ -23,48 +32,58 @@ def create_client(host, port, username, password):
     client.on_message = on_message
     client.connect(host, port)
     client.loop_start()
-    return client
 
 def on_connect(client, userdata, flags, rc):
-    print("Connected with result code "+str(rc))
-    
+    print(f"Connected with result code {rc}")
+
 def on_message(client, userdata, message):
+    global latest_data
     print("Received message")
-    global bed_temp, nozzle_temp, status, update_time
-    data = json.loads(message.payload)
-    bed_temp = data["bed_temperature"]
-    nozzle_temp = data["nozzle_temperature"]
-    status = data["status"]
-    update_time = str(time.time())
-    
-    
+    try:
+        data = json.loads(message.payload)
+        latest_data["bed_temperature"] = data.get("bed_temperature", "N/A")
+        latest_data["nozzle_temperature"] = data.get("nozzle_temperature", "N/A")
+        latest_data["status"] = data.get("status", "N/A")
+        latest_data["update_time"] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
+
+    except Exception as e:
+        print(f"Error parsing MQTT message: {e}")
+
 def get_data(serial):
-    print("Getting data")
-    global client
+    """Request data from the MQTT broker."""
+    global client, response_topic
+
     if client is None:
-        client = create_client(HOST, PORT, USERNAME, PASSWORD)
-    request = f"bambu_a1_mini/request/{serial}"
-    response = f"bambu_a1_mini/response/{serial}"
-    client.subscribe(response)
-    client.publish(request, json.dumps("HI"))
+        create_client(HOST, PORT, USERNAME, PASSWORD)
 
+    request_topic = f"bambu_a1_mini/request/{serial}"
+    response_topic = f"bambu_a1_mini/response/{serial}"
+    
+    print(f"Subscribing to {response_topic}")
+    client.subscribe(response_topic)
+    
+    # Send a request to get data
+    client.publish(request_topic, json.dumps("HI"))
+    
+    return "Request sent. Waiting for response..."
+
+def display_data():
+    """Function to return the latest received data."""
+    return latest_data["status"], latest_data["bed_temperature"], latest_data["nozzle_temperature"], latest_data["update_time"]
+
+# Gradio UI
 with gr.Blocks() as blocks:
     serial = gr.Textbox("0309CA471800852", label="Serial Number")
-    send_button = gr.Button(value="Send")
-    
+    send_button = gr.Button(value="Request Data")
+
     status_text = gr.Textbox(label="Status", interactive=False)
     bed_temp_text = gr.Textbox(label="Bed Temperature", interactive=False)
     nozzle_temp_text = gr.Textbox(label="Nozzle Temperature", interactive=False)
-    
-    send_button.click(
-        fn=get_data,
-        inputs=serial,
-        # outputs=[status_text, bed_temp_text, nozzle_temp_text],
-    )
-
-    status.change(lambda: status, status_text, status_text)
-    bed_temp.change(lambda: bed_temp, bed_temp_text, bed_temp_text)
-    nozzle_temp.change(lambda: nozzle_temp, nozzle_temp_text, nozzle_temp_text)
-    
-    
-blocks.launch()
+    update_time_text = gr.Textbox(label="Last Update", interactive=False)
+
+    send_button.click(fn=get_data, inputs=[serial], outputs=[])
+
+    # Poll the latest received data
+    blocks.load(fn=display_data, inputs=[], outputs=[status_text, bed_temp_text, nozzle_temp_text, update_time_text], every=2)
+
+blocks.launch()