Update app.py

app.py

@@ -47,6 +47,8 @@ latest_data = {
     "status": "N/A",
     "update_time": "Waiting for data...",
     "image_url": "N/A",
+    "progress": 0,  
+    "message": "",  
 }
 
 bambu_client = None
@@ -115,6 +117,13 @@ def rpi_on_message(client, userdata, msg):
         
         if "error" in data:
             logger.error(f"Error from RPI: {data['error']}")
+       
+        if status == "Ready":
+            latest_data["progress"] = 0
+            latest_data["message"] = "Printer ready"
+        elif status == "Processing":
+            latest_data["progress"] = 25
+            latest_data["message"] = "Processing G-code..."
         
     except Exception as e:
         logger.error(f"Error processing message from RPI: {e}")
@@ -189,6 +198,11 @@ def send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed):
         logger.error(f"Error sending parameters: {e}")
         return f"Error sending parameters: {e}"
 
+      
+    latest_data["status"] = "Sending"
+    latest_data["progress"] = 10
+    latest_data["message"] = "Sending parameters to printer..."
+
 
 
 def get_image_base64(image):
@@ -293,6 +307,17 @@ def capture_image(url=None, use_test_image=False, test_image_name=None):
     else:
         raise Exception("url is 'N/A'")
 
+def update_print_status():
+   global latest_data
+   return (
+       latest_data["status"],
+       latest_data["nozzle_temperature"],
+       latest_data["bed_temperature"],
+       latest_data["update_time"],
+       latest_data.get("progress", 0),
+       latest_data.get("message", ""),
+   )
+
 
 def health_check():
     status = {
@@ -585,6 +610,20 @@ with demo:
         api_name="send_print_parameters",
     )
 
+    get_status_api = demo.load(
+        fn=health_check,
+        inputs=[],
+        outputs=[
+            gr.Textbox(label="Status"),
+            gr.Textbox(label="Nozzle Temperature"),
+            gr.Textbox(label="Bed Temperature"),
+            gr.Textbox(label="Update Time"),
+            gr.Number(label="Progress"),
+            gr.Textbox(label="Message"),
+        ],
+        api_name="get_status",
+    )
+
 if __name__ == "__main__":
     logger.info("Starting Bambu A1 Mini Print Control application")