SissiFeng commited on
Commit
d3419a2
·
1 Parent(s): a9a864d
Files changed (1) hide show
  1. app.py +46 -49
app.py CHANGED
@@ -235,7 +235,7 @@ def health_check():
235
  # 创建 Gradio 应用
236
  demo = gr.Blocks(title="Bambu A1 Mini Print Control")
237
 
238
- # 在 Blocks 上下文中注册 API 端点
239
  with demo:
240
  gr.Markdown("# Bambu A1 Mini Print Control")
241
 
@@ -439,54 +439,51 @@ with demo:
439
  logger.info(f"API call: send_print_parameters with nozzle={nozzle_temp}, bed={bed_temp}, speed={print_speed}, fan={fan_speed}")
440
  return send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed)
441
 
442
- # 注册 capture_frame API
443
- capture_frame_api = demo.load(
444
- fn=api_capture_frame,
445
- inputs=[
446
- gr.Textbox(label="Image URL"), # 接收图像 URL
447
- gr.Checkbox(label="Use Test Image", value=False),
448
- gr.Textbox(label="Test Image Name", value="")
449
- ],
450
- outputs="json",
451
- api_name="capture_frame"
452
- )
453
-
454
- # 注册 lambda API
455
- lambda_api = demo.load(
456
- fn=api_lambda,
457
- inputs=[
458
- gr.Textbox(label="Image Base64 or URL"), # 接收图像数据或 URL
459
- gr.Number(label="Nozzle Temperature", value=200),
460
- gr.Number(label="Bed Temperature", value=60),
461
- gr.Number(label="Print Speed", value=60),
462
- gr.Number(label="Fan Speed", value=100),
463
- gr.Checkbox(label="Use Test Image", value=False),
464
- gr.Textbox(label="Test Image Name", value="")
465
- ],
466
- outputs="json",
467
- api_name="lambda"
468
- )
469
-
470
- # 注册 get_data API
471
- get_data_api = demo.load(
472
- fn=api_get_data,
473
- inputs=None,
474
- outputs="json",
475
- api_name="get_data"
476
- )
477
-
478
- # 注册 send_print_parameters API
479
- send_params_api = demo.load(
480
- fn=api_send_print_parameters,
481
- inputs=[
482
- gr.Number(label="Nozzle Temperature", value=200),
483
- gr.Number(label="Bed Temperature", value=60),
484
- gr.Number(label="Print Speed", value=60),
485
- gr.Number(label="Fan Speed", value=100)
486
- ],
487
- outputs="text",
488
- api_name="send_print_parameters"
489
- )
490
 
491
  # 启动应用
492
  if __name__ == "__main__":
 
235
  # 创建 Gradio 应用
236
  demo = gr.Blocks(title="Bambu A1 Mini Print Control")
237
 
238
+ # 在 Blocks 上下文中注册所有内容
239
  with demo:
240
  gr.Markdown("# Bambu A1 Mini Print Control")
241
 
 
439
  logger.info(f"API call: send_print_parameters with nozzle={nozzle_temp}, bed={bed_temp}, speed={print_speed}, fan={fan_speed}")
440
  return send_print_parameters(nozzle_temp, bed_temp, print_speed, fan_speed)
441
 
442
+ # 注册 API 端点 - 这里是关键修改,将这些调用移到 with demo: 上下文内
443
+ capture_frame_api = demo.load(
444
+ fn=api_capture_frame,
445
+ inputs=[
446
+ gr.Textbox(label="Image URL"), # 接收图像 URL
447
+ gr.Checkbox(label="Use Test Image", value=False),
448
+ gr.Textbox(label="Test Image Name", value="")
449
+ ],
450
+ outputs="json",
451
+ api_name="capture_frame"
452
+ )
453
+
454
+ lambda_api = demo.load(
455
+ fn=api_lambda,
456
+ inputs=[
457
+ gr.Textbox(label="Image Base64 or URL"), # 接收图像数据或 URL
458
+ gr.Number(label="Nozzle Temperature", value=200),
459
+ gr.Number(label="Bed Temperature", value=60),
460
+ gr.Number(label="Print Speed", value=60),
461
+ gr.Number(label="Fan Speed", value=100),
462
+ gr.Checkbox(label="Use Test Image", value=False),
463
+ gr.Textbox(label="Test Image Name", value="")
464
+ ],
465
+ outputs="json",
466
+ api_name="lambda"
467
+ )
468
+
469
+ get_data_api = demo.load(
470
+ fn=api_get_data,
471
+ inputs=None,
472
+ outputs="json",
473
+ api_name="get_data"
474
+ )
475
+
476
+ send_params_api = demo.load(
477
+ fn=api_send_print_parameters,
478
+ inputs=[
479
+ gr.Number(label="Nozzle Temperature", value=200),
480
+ gr.Number(label="Bed Temperature", value=60),
481
+ gr.Number(label="Print Speed", value=60),
482
+ gr.Number(label="Fan Speed", value=100)
483
+ ],
484
+ outputs="text",
485
+ api_name="send_print_parameters"
486
+ )
 
 
 
487
 
488
  # 启动应用
489
  if __name__ == "__main__":