update

app.py

@@ -33,14 +33,13 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-# Global variables to store received data
+# Global variables
+client = None
 latest_data = {
     "bed_temperature": "N/A",
     "nozzle_temperature": "N/A",
-    "status": "N/A",
-    "update_time": "Waiting for data...",
-    "current_image_url": None,
-    "print_progress": 0
+    "status": "Disconnected",
+    "update_time": "Not connected"
 }
 
 # Initialize analysis components
@@ -112,18 +111,63 @@ def create_3mf_package(gcode_content: str, gcode_filename: str = "Metadata/plate
     zip_buffer.seek(0)
     return zip_buffer
 
-def create_client(host, port, username, password):
+def create_client(host, port, username, password, max_retries=3):
+    """Create MQTT client with retry mechanism"""
     global client
-    client = mqtt.Client()
-    client.username_pw_set(username, password)
-    client.tls_set(tls_version=mqtt.ssl.PROTOCOL_TLS)
-    client.on_connect = on_connect
-    client.on_message = on_message
-    client.connect(host, port)
-    client.loop_start()
+    
+    for attempt in range(max_retries):
+        try:
+            # Create new client
+            client = mqtt.Client(protocol=mqtt.MQTTv311)
+            client.username_pw_set(username, password)
+            client.tls_set(tls_version=mqtt.ssl.PROTOCOL_TLS)
+            client.on_connect = on_connect
+            client.on_message = on_message
+            client.on_disconnect = on_disconnect
+            
+            # Try to connect
+            print(f"Attempting to connect to MQTT broker (attempt {attempt + 1}/{max_retries})")
+            client.connect(host, port, keepalive=60)
+            client.loop_start()
+            return True
+            
+        except Exception as e:
+            print(f"Connection attempt {attempt + 1} failed: {e}")
+            if client:
+                client.loop_stop()
+                client = None
+            if attempt < max_retries - 1:
+                time.sleep(5)  # Wait before retry
+                
+    print("Failed to connect to MQTT broker after all attempts")
+    return False
 
 def on_connect(client, userdata, flags, rc):
-    print(f"Connected with result code {rc}")
+    """Callback when connected to MQTT broker"""
+    if rc == 0:
+        print("Successfully connected to MQTT broker")
+        serial = os.getenv("PRINTER_SERIAL", "0309CA471800852")
+        response_topic = f"bambu_a1_mini/response/{serial}"
+        client.subscribe(response_topic)
+        print(f"Subscribed to {response_topic}")
+        
+        # Update status
+        global latest_data
+        latest_data["status"] = "Connected"
+        latest_data["update_time"] = time.strftime("%Y-%m-%d %H:%M:%S")
+    else:
+        print(f"Failed to connect to MQTT broker with code: {rc}")
+
+def on_disconnect(client, userdata, rc):
+    """Callback when disconnected from MQTT broker"""
+    print("Disconnected from MQTT broker")
+    global latest_data
+    latest_data["status"] = "Disconnected"
+    latest_data["update_time"] = time.strftime("%Y-%m-%d %H:%M:%S")
+    
+    if rc != 0:
+        print("Unexpected disconnection. Attempting to reconnect...")
+        create_client(HOST, PORT, USERNAME, PASSWORD)
 
 def on_message(client, userdata, message):
     global latest_data
@@ -172,8 +216,13 @@ def send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed):
     except Exception as e:
         return f"Error sending parameters: {e}"
 
-# Initialize MQTT client
-mqtt_client = create_client(HOST, PORT, USERNAME, PASSWORD)
+# Initialize MQTT client (but don't fail if it doesn't connect)
+try:
+    mqtt_connected = create_client(HOST, PORT, USERNAME, PASSWORD)
+    if not mqtt_connected:
+        print("Warning: Starting without MQTT connection")
+except Exception as e:
+    print(f"Warning: Failed to initialize MQTT: {e}")
 
 def update_printer_status():
     """Get current printer status"""