From 3a23a1b47bf547a0aea5045db41e7f5cd8600ba0 Mon Sep 17 00:00:00 2001 From: jeremygan2021 Date: Thu, 5 Mar 2026 20:09:47 +0800 Subject: [PATCH] printer --- main.py | 54 +++++++++++++++++++++++++++++++++++--- websocket_server/server.py | 17 +++++++----- 2 files changed, 61 insertions(+), 10 deletions(-) diff --git a/main.py b/main.py index 589d773..db569a0 100644 --- a/main.py +++ b/main.py @@ -112,7 +112,7 @@ def connect_wifi(display=None, max_retries=5): -def process_message(msg, display, image_state, image_data_list): +def process_message(msg, display, image_state, image_data_list, printer_uart=None): """处理WebSocket消息""" # Handle binary image data if isinstance(msg, (bytes, bytearray)): @@ -138,6 +138,12 @@ def process_message(msg, display, image_state, image_data_list): print(f"Stream image error: {e}") return image_state, None + + elif image_state == PRINTER_STATE_RECEIVING: + if printer_uart: + printer_uart.write(msg) + return image_state, None + return image_state, None if not isinstance(msg, str): @@ -153,6 +159,14 @@ def process_message(msg, display, image_state, image_data_list): print_asr(msg[4:], display) return image_state, ("asr", msg[4:]) + elif msg.startswith("PRINTER_DATA_START:"): + print(f"Start receiving printer data...") + return PRINTER_STATE_RECEIVING, ("printer_start",) + + elif msg == "PRINTER_DATA_END": + print("Printer data received completely") + return IMAGE_STATE_IDLE, ("printer_done",) + elif msg.startswith("STATUS:"): parts = msg[7:].split(":", 1) status_type = parts[0] @@ -284,6 +298,9 @@ def main(): mic = Microphone() display = Display() + # 初始化打印机 UART + printer_uart = machine.UART(1, baudrate=115200, tx=ttl_tx, rx=ttl_rx) + if display.tft: display.init_ui() display.render_home_screen() @@ -446,9 +463,20 @@ def main(): if display.tft: display.render_result_screen("OPTIMIZING", current_asr_text, False) time.sleep(0.5) + elif ui_screen == UI_SCREEN_RESULT: + # Re-record + print(">>> Re-record (Short Press)") + current_asr_text = "" + confirm_waiting = False + ui_screen = UI_SCREEN_RECORDING + is_recording = False + image_generation_done = False + if display.tft: + display.render_recording_screen("", 0, False) + time.sleep(0.5) elif btn_action == 2: - if ui_screen == UI_SCREEN_CONFIRM or ui_screen == UI_SCREEN_RESULT: + if ui_screen == UI_SCREEN_CONFIRM: print(">>> Re-record") current_asr_text = "" confirm_waiting = False @@ -458,6 +486,17 @@ def main(): if display.tft: display.render_recording_screen("", 0, False) time.sleep(0.5) + elif ui_screen == UI_SCREEN_RESULT: + # Print Image + print(">>> Print Image (Long Press)") + if ws and ws.is_connected(): + try: + ws.send("PRINT_IMAGE") + if display.tft: + display.render_result_screen("PRINTING", "正在请求打印...", True) + except: + ws = None + time.sleep(0.5) elif btn_action == 3: print(">>> Config mode") @@ -484,7 +523,7 @@ def main(): if events: msg = ws.recv() if msg: - image_state, event_data = process_message(msg, display, image_state, image_data_list) + image_state, event_data = process_message(msg, display, image_state, image_data_list, printer_uart) if event_data: if event_data[0] == "asr": @@ -521,6 +560,15 @@ def main(): elif event_data[0] == "error": if display.tft and ui_screen == UI_SCREEN_RESULT: display.render_result_screen("ERROR", current_prompt, False) + + elif event_data[0] == "printer_start": + if display.tft and ui_screen == UI_SCREEN_RESULT: + display.render_result_screen("PRINTING", "正在打印...", True) + + elif event_data[0] == "printer_done": + if display.tft and ui_screen == UI_SCREEN_RESULT: + display.render_result_screen("COMPLETE", "打印完成", True) + time.sleep(1.0) except Exception as e: print(f"WS Recv Error: {e}") diff --git a/websocket_server/server.py b/websocket_server/server.py index 70ccae4..dc34c77 100644 --- a/websocket_server/server.py +++ b/websocket_server/server.py @@ -530,17 +530,20 @@ def optimize_prompt(asr_text, progress_callback=None): if progress_callback: progress_callback(0, "正在准备优化提示词...") - system_prompt = """你是一个AI图像提示词优化专家。将用户简短的语音识别结果转化为详细的、适合AI图像生成的中文提示词。 -要求： -1. 用于热敏打印机的中文提示词图片 -2. 添加适合AI绘画的描述词尺寸宽48mm, 高30mm 的线稿图片，线稿要存粗方便热敏打印 -3. 适合热敏打印机打印图片，还可以是icon方便这个标签打印机打印效果 -4. 简洁但描述详细 -5. 不要添加多余解释，直接输出优化后的提示词""" + system_prompt = """你是一个AI图像提示词优化专家。你的任务是将用户的语音识别结果转化为适合生成"黑白线稿"的提示词。 +关键要求： +1. 风格必须是：简单的黑白线稿、简笔画、图标风格 (Line art, Sketch, Icon style)。 +2. 画面必须清晰、线条粗壮，适合低分辨率热敏打印机打印。 +3. 绝对不要有复杂的阴影、渐变、彩色描述。 +4. 背景必须是纯白 (White background)。 +5. 提示词内容请使用英文描述，因为绘图模型对英文理解更好，但在描述中强调 "black and white line art", "simple lines", "vector style"。 +6. 尺寸比例遵循宽48mm:高30mm (约 1.6:1)。 +7. 直接输出优化后的提示词，不要包含任何解释。""" try: if progress_callback: progress_callback(10, "正在调用AI优化提示词...") + print(f"Calling AI with prompt: {system_prompt}\n\n用户语音识别结果：{asr_text}\n\n优化后的提示词：") response = Generation.call( model='qwen-turbo',