import modes as mc import tkinter as tk from tkinter import ttk import threading import base_mode as bm IS_BOARD = True try: from pinpong.board import Board, Pin, NeoPixel from unihiker import GUI gui = GUI() except ModuleNotFoundError: print("警告：非行空板环境") IS_BOARD = False import time class XingkongBoardBase: def __init__(self): pass def on_btn_click(self, callback): """ 按下机外的那个按钮。 该按钮按键流程（必须辩论模式）： 1. 进入立论模式（若是正方，则直接说话；若是反方，则直接开始录音） 2. 当本方为正方时，开始录（反方）音；为反方时，停止录（正方的）音 3. 当本方为正方时，停止录（反方）音；为反方时，开始说话 4. 当本方为正方时，进入进入自由辩论模式；为反方时，也是进入自由辩论模式（请确保灯光为绿） 5. 关闭自由辩论模式 6. 进入总结陈词模式。为正方时将直接开始录音；为反方时将直接输出语音，完成后辩论模式结束 7. 正方停止录音 8. 正方输出语音。完成后辩论模式结束 :param callback: 当按下时的回调函数 :return: """ pass def on_btn_dialog_click(self, callback): """ 按下“对话模式”按钮 :param callback: 当按下时的回调函数 :return: """ pass def on_btn_debate_click(self, callback): """ 按下“辩论模式”按钮 :param callback: 当按下时的回调函数 :return: """ pass def on_btn_stop_click(self, callback): """ 按下“停止”按钮 :param callback: 当按下时的回调函数 :return: """ pass def set_light_color(self, color): """ 设置灯光颜色。 在对话模式和自由辩论模式不设置灯光颜色。 red: 正在处理一些事件，此时不能点击机外按钮 green: 准备就绪，可以点击机外按钮 blue: 正在录音 orange: 成功进入辩论模式和自由辩论 :param color: "red"|"green"|"blue"|"orange" :return: """ pass SYSTEM = \ """ 你是人大附中经开学校金鹏科技团开发的辩论智能体机器人，名叫人开一号。 这个机器人运用了流式语音识别、大语言模型、流式语音合成等技术。 当你被要求介绍自己时，请简要介绍你的功能和技术实现。 请注意，你的回答将被直接通过语音合成输出，因此你不能输出不适合TTS的内容。 """ class Unihiker_Board(XingkongBoardBase): def __init__(self): super().__init__() Board().begin() # self.gui=GUI() self.btn21 = Pin(Pin.P21, Pin.IN) self.pin23 = Pin(Pin.D23) self.np1 = NeoPixel(self.pin23, 22) self.np1.brightness(150) gui.add_button(x=0, y=290, w=80, h=30, text="对话", origin='nw', onclick=self._trigger_dialog_click) gui.add_button(x=80, y=290, w=80, h=30, text="辩论", origin='nw', onclick=self._trigger_debate_click) gui.add_button(x=160, y=290, w=80, h=30, text="停止", origin='nw', onclick=self._trigger_stop_click) self.btn_count = 0 self.btn_callback = None self.dialog_callback = None self.debate_callback = None self.stop_callback = None def on_btn_click(self, callback): self.btn_callback = callback def on_btn_dialog_click(self, callback): self.dialog_callback = callback def on_btn_debate_click(self, callback): self.debate_callback = callback def on_btn_stop_click(self, callback): self.stop_callback = callback def set_light_color(self, color): if color == "red": self.np1.range_color(0, 21, 0xFF0000) if color == "green": self.np1.range_color(0, 21, 0x00FF00) if color == "blue": self.np1.range_color(0, 21, 0x0000FF) if color == "orange": self.np1.range_color(0, 21, 0xFFFF00) def _trigger_btn_click(self): if self.btn_callback: threading.Thread(target=self.btn_callback, daemon=True).start() def _trigger_dialog_click(self): if self.dialog_callback: threading.Thread(target=self.dialog_callback, daemon=True).start() def _trigger_debate_click(self): if self.debate_callback: threading.Thread(target=self.debate_callback, daemon=True).start() def _trigger_stop_click(self): if self.stop_callback: threading.Thread(target=self.stop_callback, daemon=True).start() def btn_click_check(self): while True: if self.btn21.read_digital() == 0: while self.btn21.read_digital() == 0: pass self._trigger_btn_click() def run(self): threading.Thread(target=self.btn_click_check, daemon=True).start() while True: time.sleep(1) class Robot: def __init__(self, board: XingkongBoardBase): self.modes = dict() self.btn_count = 0 self.board = board self.side = "" self.topic = "" self.opinion = "" self.last_prompt = "" self.session_id = None def register_mode(self, modename: str, mode: bm.Mode): self.modes[modename] = mode def run_mode(self, modename): self.modes[modename].run() def stop_mode(self, modename): self.modes[modename].stop() del self.modes[modename] def get_mode(self, modename): return self.modes[modename] def stop_all(self): for mode in self.modes.values(): mode.stop() self.modes.clear() def dialog(self): self.stop_all() self.board.set_light_color("orange") dialog_mode = mc.DialogMode(system_prompt=SYSTEM) self.register_mode("dialog", dialog_mode) self.run_mode("dialog") def set_debate_config(self, side, topic, opinion): self.side = side self.topic = topic self.opinion = opinion def debate(self): self.stop_all() self.board.set_light_color("orange") self.btn_count = 0 if self.side == "正方": self.board.on_btn_click(self._btn_right_callback) else: self.board.on_btn_click(self._btn_against_callback) def on_cancel(self): self.stop_all() self.board.set_light_color("green") def _btn_right_callback(self): self.btn_count += 1 if self.btn_count == 1: self.board.set_light_color("red") self.register_mode("make_point", mc.MakePointMode(side=self.side, topic=self.topic, opinion=self.opinion)) self.run_mode("make_point") self.board.set_light_color("green") elif self.btn_count == 2: md: mc.MakePointMode = self.get_mode("make_point") self.board.set_light_color("blue") md.start_record() elif self.btn_count == 3: md: mc.MakePointMode = self.get_mode("make_point") self.board.set_light_color("red") md.stop_record() self.board.set_light_color("green") self.last_prompt = md.ready_next() elif self.btn_count == 4: self.board.set_light_color("red") md: mc.MakePointMode = self.get_mode("make_point") self.session_id = md.session_id self.stop_mode("make_point") self.register_mode("free_debate", mc.FreeDebateMode(side=self.side, topic=self.topic, opinion=self.opinion, last_prompt=self.last_prompt, session_id=self.session_id)) self.run_mode("free_debate") self.board.set_light_color("orange") elif self.btn_count == 5: self.board.set_light_color("red") md: mc.FreeDebateMode = self.get_mode("free_debate") self.session_id = md.session_id self.stop_mode("free_debate") self.board.set_light_color("green") elif self.btn_count == 6: self.board.set_light_color("red") self.register_mode("end_debate", mc.EndDebateMode(side=self.side, topic=self.topic, opinion=self.opinion, session_id=self.session_id)) self.run_mode("end_debate") self.board.set_light_color("blue") elif self.btn_count == 7: md: mc.EndDebateMode = self.get_mode("end_debate") md.stop_record() self.board.set_light_color("green") elif self.btn_count == 8: self.board.set_light_color("red") md: mc.EndDebateMode = self.get_mode("end_debate") md.start_talk() self.board.set_light_color("green") self.stop_mode("end_debate") def _btn_against_callback(self): self.btn_count += 1 if self.btn_count == 1: self.board.set_light_color("blue") self.register_mode("make_point", mc.MakePointMode(side=self.side, topic=self.topic, opinion=self.opinion)) self.run_mode("make_point") elif self.btn_count == 2: md: mc.MakePointMode = self.get_mode("make_point") self.board.set_light_color("red") md.stop_record() self.board.set_light_color("green") elif self.btn_count == 3: self.board.set_light_color("red") md: mc.MakePointMode = self.get_mode("make_point") md.ready_next() self.board.set_light_color("green") elif self.btn_count == 4: self.board.set_light_color("red") md: mc.MakePointMode = self.get_mode("make_point") self.session_id = md.session_id self.stop_mode("make_point") self.register_mode("free_debate", mc.FreeDebateMode(side=self.side, topic=self.topic, opinion=self.opinion, session_id=self.session_id)) self.run_mode("free_debate") self.board.set_light_color("green") elif self.btn_count == 5: self.board.set_light_color("red") md: mc.FreeDebateMode = self.get_mode("free_debate") self.session_id = md.session_id self.stop_mode("free_debate") self.board.set_light_color("green") elif self.btn_count == 6: self.board.set_light_color("red") self.register_mode("end_debate", mc.EndDebateMode(side=self.side, topic=self.topic, opinion=self.opinion, session_id=self.session_id)) self.run_mode("end_debate") md: mc.EndDebateMode = self.get_mode("end_debate") md.start_talk() def run(self): self.board.on_btn_dialog_click(self.dialog) self.board.on_btn_debate_click(self.debate) self.board.on_btn_stop_click(self.on_cancel) class MockXingkongBoard(XingkongBoardBase): def __init__(self): super().__init__() self.root = tk.Tk() self.root.title("模拟控制台") self.root.geometry("300x200") # 灯光显示 self.light = tk.Label(self.root, width=10, height=5, bg="gray") self.light.pack(pady=10) # 按钮布局 btn_frame = ttk.Frame(self.root) btn_frame.pack(pady=10) # 机外按钮 ttk.Button(btn_frame, text="机外按钮", command=self._trigger_btn_click).grid(row=0, column=0, padx=5) # 功能按钮 ttk.Button(btn_frame, text="对话模式", command=self._trigger_dialog_click).grid(row=1, column=0, padx=5) ttk.Button(btn_frame, text="辩论模式", command=self._trigger_debate_click).grid(row=1, column=1, padx=5) ttk.Button(btn_frame, text="停止", command=self._trigger_stop_click).grid(row=1, column=2, padx=5) # 回调存储 self.btn_callback = None self.dialog_callback = None self.debate_callback = None self.stop_callback = None def on_btn_click(self, callback): self.btn_callback = callback def on_btn_dialog_click(self, callback): self.dialog_callback = callback def on_btn_debate_click(self, callback): self.debate_callback = callback def on_btn_stop_click(self, callback): self.stop_callback = callback def set_light_color(self, color): # 仅设置颜色，不添加任何模式判断 self.light.config(bg=color) def _trigger_btn_click(self): # 直接触发回调，高耗时操作放线程 if self.btn_callback: threading.Thread(target=self.btn_callback, daemon=True).start() def _trigger_dialog_click(self): if self.dialog_callback: threading.Thread(target=self.dialog_callback, daemon=True).start() def _trigger_debate_click(self): if self.debate_callback: threading.Thread(target=self.debate_callback, daemon=True).start() def _trigger_stop_click(self): if self.stop_callback: threading.Thread(target=self.stop_callback, daemon=True).start() def run(self): self.root.mainloop() if __name__ == '__main__': board = Unihiker_Board() if IS_BOARD else MockXingkongBoard() robot = Robot(board) robot.set_debate_config( "反方", "利用基因技术复活已灭绝物种（反灭绝工程）是生态责任还是资源浪费", "利用基因技术复活已灭绝物种（反灭绝工程）是资源浪费" ) if IS_BOARD else robot.set_debate_config( "反方", "量子计算机的实用化是否会首先对现有加密体系构成致命威胁", "量子计算机的实用化是否不会首先对现有加密体系构成致命威胁" ) robot.run() board.run()