初次移植

src/heurams/interface/screens/memory.py

@@ -0,0 +1,140 @@
+
+class MemScreen(Screen):
+    BINDINGS = [
+        ("d", "toggle_dark", "改变色调"),
+        ("q", "pop_screen", "返回主菜单"),
+        ("v", "play_voice", "朗读"),
+    #    ("p", "precache_current", "预缓存当前单元集"),  # 新增预缓存快捷键
+    ]
+    if config.get("quick_pass"):
+        BINDINGS.append(("k", "quick_pass", "快速通过[调试]"))
+    btn = dict()
+
+    def __init__(
+        self,
+        nucleon_file: pt.NucleonUnion,
+        electron_file: pt.ElectronUnion,
+        tasked_num,
+    ):
+        super().__init__(name=None, id=None, classes=None)
+        self.nucleon_file = nucleon_file
+        self.electron_file = electron_file
+        self.reactor = Reactor(nucleon_file, electron_file, self, tasked_num)
+        self.stage = 1
+        self.stage += self.reactor.set_round_templated(self.stage)
+        first_forward = self.reactor.forward()
+        print(first_forward)
+        if first_forward == -1:
+            self.stage = 3
+            self.reactor.set_round_templated(3)
+            print(self.reactor.forward())
+        #self._forward_judge(first_forward)
+        self.compo = next(self.reactor.current_appar)
+        self.feedback_state = 0 # 默认状态
+        self.feedback_state_map = {
+            0: "",
+            255: "回答有误, 请重试. 或者重新学习此单元",
+        }
+
+    def compose(self) -> ComposeResult:
+        if type(self.compo).__name__ == "Recognition":
+            self.action_play_voice()
+        yield Header(show_clock=True)
+        with Center():
+            yield Static(
+                f"当前进度: {len(self.reactor.procession) - self.reactor.index}/{len(self.reactor.procession)}"
+            )
+        yield Label(self.feedback_state_map[self.feedback_state])
+        yield from self.compo.compose()
+        if self.feedback_state == 255:
+            yield Button("重新学习此单元", id="re-recognize", variant="warning")
+        yield Footer()
+
+    def on_mount(self):
+        pass
+
+    def on_button_pressed(self, event):
+        try:
+            if event.button.id == "re-recognize":
+                return
+        except:
+            pass
+        ret = self.compo.handler(event, "button")
+        self._forward_judge(ret)
+    def _forward_judge(self, ret):
+        self.feedback_state = 0
+        if ret == -1:
+            return
+        if ret == 0:
+            try:
+                self.compo = next(self.reactor.current_appar)
+                self.refresh_ui()
+            except StopIteration:
+                nxt = self.reactor.forward(1)
+                try:
+                    self.compo = next(self.reactor.current_appar)
+                except:
+                    pass
+                if nxt == -1:
+                    if self.reactor.round_set == 0:
+                        if self.stage == 4:
+                            if config.get("save"):
+                                self.reactor.save()
+                            self.compo = compo.Finished(
+                                self, None, pt.Atom.placeholder()
+                            )
+                            self.refresh_ui()
+                        else:
+                            self.reactor.set_round_templated(self.stage)
+                            self.reactor.forward(1)
+                            self.stage += 1
+                            self.compo = next(self.reactor.current_appar)
+                            self.refresh_ui()
+                        return
+                    return
+                else:
+                    self.refresh_ui()
+                    return
+        if ret >= 1:
+            if ret == 2:
+                self.feedback_state = 255 # 表示错误
+            else:
+                self.feedback_state = 0
+            self.refresh_ui()
+            return
+
+    def refresh_ui(self):
+        self.call_later(self.recompose)
+        print(type(self.compo).__name__)
+
+    def action_play_voice(self):
+        def play():
+            cache_dir = pathlib.Path(f"./cache/voice/")
+            cache_dir.mkdir(parents=True, exist_ok=True)
+            cache = cache_dir / f"{aux.get_md5(self.reactor.current_atom[1].content.replace('/',''))}.wav"
+            if not cache.exists():
+                import edge_tts as tts
+                communicate = tts.Communicate(
+                    self.reactor.current_atom[1].content.replace("/", ""),
+                    "zh-CN-XiaoxiaoNeural",
+                )
+                communicate.save_sync(
+                    f"./cache/voice/{aux.get_md5(self.reactor.current_atom[1].content.replace('/',''))}.wav"
+                )
+            playsound(str(cache))
+
+        threading.Thread(target=play).start()
+
+    def action_precache_current(self):
+        """预缓存当前单元集的音频"""
+        precache_screen = PrecachingScreen(self.nucleon_file)
+        self.app.push_screen(precache_screen)
+
+    def action_quick_pass(self):
+        self.reactor.report(self.reactor.current_atom, 5)
+        self._forward_judge(0)
+    def action_toggle_dark(self):
+        self.app.action_toggle_dark()
+
+    def action_pop_screen(self):
+        self.app.pop_screen()