a

app.py

@@ -3,7 +3,6 @@ import os
 import gradio as gr
 import json
 import time
-import threading
 
 # Environment Variables
 HOST = os.environ.get("host")
@@ -11,19 +10,13 @@ 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
+# MQTT Variables
 client = None
-response_topic = None  # Will be set dynamically
+response_topic = None
+latest_data = {"status": "N/A", "bed_temperature": "N/A", "nozzle_temperature": "N/A", "update_time": "Waiting for data"}
 
 def create_client(host, port, username, password):
+    """Initialize the MQTT client."""
     global client
     client = mqtt.Client()
     client.username_pw_set(username, password)
@@ -34,9 +27,11 @@ def create_client(host, port, username, password):
     client.loop_start()
 
 def on_connect(client, userdata, flags, rc):
+    """Handle MQTT connection."""
     print(f"Connected with result code {rc}")
 
 def on_message(client, userdata, message):
+    """Handle incoming MQTT messages and update Gradio UI dynamically."""
     global latest_data
     print("Received message")
     try:
@@ -45,7 +40,7 @@ def on_message(client, userdata, message):
         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}")
 
@@ -58,17 +53,18 @@ def get_data(serial):
 
     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..."
 
+@gr.render(inputs=[])
 def display_data():
-    """Function to return the latest received data."""
+    """Function to return the latest received data dynamically."""
     return latest_data["status"], latest_data["bed_temperature"], latest_data["nozzle_temperature"], latest_data["update_time"]
 
 # Gradio UI
@@ -83,7 +79,7 @@ with gr.Blocks() as blocks:
 
     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])
+    # Dynamic rendering using @gr.render
+    display_data() >> [status_text, bed_temp_text, nozzle_temp_text, update_time_text]
 
 blocks.launch()