update

app.py

@@ -132,6 +132,7 @@ def on_connect(client, userdata, flags, rc):
     logger.info(f"Connected with result code {rc}")
     if rc == 0:
         serial = os.getenv("PRINTER_SERIAL", "0309CA471800852")
+        request_topic = f"bambu_a1_mini/request/{serial}"
         response_topic = f"bambu_a1_mini/response/{serial}"
         client.subscribe(response_topic)
         logger.info(f"Subscribed to {response_topic}")
@@ -163,6 +164,27 @@ def send_camera_command(square_id: str):
     logger.info(f"Camera command sent for square {square_id}")
     return "Camera command sent successfully"
 
+def send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed):
+    """Send new print parameters to printer"""
+    try:
+        params = {
+            'nozzle_temp': nozzle_temp,
+            'bed_temp': bed_temp,
+            'print_speed': print_speed,
+            'fan_speed': fan_speed
+        }
+        
+        serial = os.getenv("PRINTER_SERIAL", "0309CA471800852")
+        request_topic = f"bambu_a1_mini/request/{serial}"
+        mqtt_client.publish(request_topic, json.dumps({
+            'command': 'set_parameters',
+            'parameters': params
+        }))
+        
+        return "Parameters sent successfully"
+    except Exception as e:
+        return f"Error sending parameters: {e}"
+
 # Initialize MQTT client
 mqtt_client = create_client(HOST, PORT, USERNAME, PASSWORD)
 
@@ -184,9 +206,7 @@ def update_printer_status():
         f"Last Update: {latest_data['update_time']}"
     )
 
-def analyze_print(image,
-                 nozzle_temp, print_speed, layer_height,
-                 flow_rate, retraction_distance, fan_speed):
+def analyze_print(image, nozzle_temp, bed_temp, print_speed, fan_speed):
     """Analyze print quality and return evaluation results"""
     if image is None:
         return None, 0, 0, 0, 0, 0, 0, 0, 0
@@ -194,10 +214,8 @@ def analyze_print(image,
     # Package current parameters
     current_params = {
         'nozzle_temp': float(nozzle_temp),
+        'bed_temp': float(bed_temp),
         'print_speed': float(print_speed),
-        'layer_height': float(layer_height),
-        'flow_rate': float(flow_rate),
-        'retraction_distance': float(retraction_distance),
         'fan_speed': float(fan_speed)
     }
     
@@ -255,27 +273,32 @@ with gr.Blocks(title="Bambu A1 Mini Print Analysis") as demo:
     with gr.Row():
         with gr.Column():
             # Print parameter inputs
-            nozzle_temp = gr.Slider(minimum=180, maximum=250, step=1, 
-                                  value=200,
-                                  label="Nozzle Temperature (°C)")
-            print_speed = gr.Slider(minimum=20, maximum=150, step=1,
-                                  value=60,
-                                  label="Print Speed (mm/s)")
-            layer_height = gr.Slider(minimum=0.1, maximum=0.4, step=0.01,
-                                  value=0.2,
-                                  label="Layer Height (mm)")
-            flow_rate = gr.Slider(minimum=90, maximum=110, step=1,
-                                value=100,
-                                label="Flow Rate (%)")
-            retraction_distance = gr.Slider(minimum=0, maximum=10, step=0.1,
-                                         value=5,
-                                         label="Retraction Distance (mm)")
-            fan_speed = gr.Slider(minimum=0, maximum=100, step=1,
-                                value=100,
-                                label="Fan Speed (%)")
+            nozzle_temp = gr.Slider(
+                minimum=180, maximum=250, step=1, 
+                value=200,
+                label="Nozzle Temperature (°C)"
+            )
+            bed_temp = gr.Slider(
+                minimum=40, maximum=100, step=1,
+                value=60,
+                label="Bed Temperature (°C)"
+            )
+            print_speed = gr.Slider(
+                minimum=20, maximum=150, step=1,
+                value=60,
+                label="Print Speed (mm/s)"
+            )
+            fan_speed = gr.Slider(
+                minimum=0, maximum=100, step=1,
+                value=100,
+                label="Fan Speed (%)"
+            )
             
-            capture_btn = gr.Button("Capture Image")
-            analyze_btn = gr.Button("Analyze Print")
+            # Buttons
+            with gr.Row():
+                capture_btn = gr.Button("Capture Image")
+                analyze_btn = gr.Button("Analyze Print")
+                send_params_btn = gr.Button("Send Print Parameters")  # 新增按钮
         
         with gr.Column():
             # Results visualization
@@ -320,17 +343,13 @@ with gr.Blocks(title="Bambu A1 Mini Print Analysis") as demo:
             send_print_command(dict(
                 nozzle_temp=params[0],      # nozzle_temp slider
                 print_speed=params[1],      # print_speed slider
-                layer_height=params[2],     # layer_height slider
-                flow_rate=params[3],        # flow_rate slider
-                retraction_distance=params[4], # retraction_distance slider
-                fan_speed=params[5],        # fan_speed slider
-                bed_temp=60  # use default value, because there is no slider for bed_temp
+                bed_temp=params[1],         # bed_temp slider
+                fan_speed=params[2],        # fan_speed slider
             ))
         ),
         inputs=[
             captured_image,
-            nozzle_temp, print_speed, layer_height,
-            flow_rate, retraction_distance, fan_speed
+            nozzle_temp, bed_temp, print_speed, fan_speed
         ],
         outputs=[
             result_image,
@@ -362,6 +381,13 @@ with gr.Blocks(title="Bambu A1 Mini Print Analysis") as demo:
     
     threading.Thread(target=auto_refresh, daemon=True).start()
 
+    # 连接按钮
+    send_params_btn.click(
+        fn=send_print_parameters,
+        inputs=[nozzle_temp, bed_temp, print_speed, fan_speed],
+        outputs=[current_job_status]
+    )
+
 def handle_print_workflow(params):
     """Coordinate print and capture workflow"""
     try: