fix: 改进

src/heurams/interface/widgets/finished.py

@@ -1,5 +1,6 @@
 from textual.widgets import (
     Label,
+    Button,
 )
 from textual.widget import Widget
 
@@ -9,4 +10,10 @@ class Finished(Widget):
         super().__init__(*children, name=name, id=id, classes=classes, disabled=disabled, markup=markup)
 
     def compose(self):
-        yield Label("本次记忆进程结束, 下次再会!", id="finished_msg")
+        yield Label("本次记忆进程结束", id="finished_msg")
+        yield Button("返回上一级", id="back-to-menu")
+
+    def on_button_pressed(self, event):
+        button_id = event.button.id
+        if button_id == 'back-to-menu':
+            self.app.pop_screen()