a

app.py

@@ -3,6 +3,7 @@ import os
 import gradio as gr
 import json
 import time
+import threading
 
 # Environment Variables
 HOST = os.environ.get("host")
@@ -10,13 +11,19 @@ PORT = int(os.environ.get("port"))
 USERNAME = os.environ.get("username")
 PASSWORD = os.environ.get("password")
 
-# MQTT Variables
+# 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
-response_topic = None
-latest_data = {"status": "N/A", "bed_temperature": "N/A", "nozzle_temperature": "N/A", "update_time": "Waiting for data"}
+response_topic = None  # Will be set dynamically
 
 def create_client(host, port, username, password):
-    """Initialize the MQTT client."""
     global client
     client = mqtt.Client()
     client.username_pw_set(username, password)
@@ -27,11 +34,9 @@ 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:
@@ -40,7 +45,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}")
 
@@ -53,15 +58,25 @@ 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"))
+    
+    global latest_data
+    latest_data = None
+    timeout = 10
+    while latest_data is None and timeout > 0:
+        time.sleep(1)
+        timeout -= 1
+        
+    return latest_data["status"], latest_data["bed_temperature"], latest_data["nozzle_temperature"], latest_data["update_time"]
 
-    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:
@@ -73,14 +88,6 @@ with gr.Blocks() as blocks:
     nozzle_temp_text = gr.Textbox(label="Nozzle Temperature", interactive=False)
     update_time_text = gr.Textbox(label="Last Update", interactive=False)
 
-    send_button.click(fn=get_data, inputs=[serial], outputs=[])
-
-    @gr.render(inputs=[])
-    def display_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"]
-
-    # Dynamic rendering using @gr.render
-    display_data() >> [status_text, bed_temp_text, nozzle_temp_text, update_time_text]
+    send_button.click(fn=get_data, inputs=[serial], outputs=[status_text, bed_temp_text, nozzle_temp_text, update_time_text])
 
 blocks.launch()