fix: clear preview before processing

Author: goodroot

lib/main.py

@@ -1445,6 +1445,7 @@ def _stop_recording(self):
         print("Recording stopped", flush=True)
 
         try:
+            self._clear_mic_osd_preview_text()
 
             # Set visualizer to processing state (keep it visible during transcription)
             self._set_visualizer_state('processing')

lib/mic_osd/main.py

@@ -313,13 +313,17 @@ def _poll_state_file(self):
                     state = f.read().strip()
                     if state and state != self._last_visualizer_state:
                         self._last_visualizer_state = state
+                        if self.window and hasattr(self.window, 'set_visualizer_state'):
+                            self.window.set_visualizer_state(state)
                         # Update visualization state if it has the set_state method
                         if hasattr(self.visualization, 'set_state'):
                             self.visualization.set_state(state)
             else:
                 # No state file means default to recording state
                 if self._last_visualizer_state != 'recording':
                     self._last_visualizer_state = 'recording'
+                    if self.window and hasattr(self.window, 'set_visualizer_state'):
+                        self.window.set_visualizer_state('recording')
                     if hasattr(self.visualization, 'set_state'):
                         self.visualization.set_state('recording')
 

lib/mic_osd/window.py

@@ -51,6 +51,7 @@ def __init__(self, visualization, width=300, height=60):
         self._width = width
         self._height = height
         self._preview_text = ""
+        self._visualizer_state = "recording"
 
         # Layer shell MUST be initialized immediately after window creation
         # and BEFORE any other window configuration
@@ -137,14 +138,19 @@ def set_preview_text(self, text: str):
         """Set compact transcript preview text."""
         self._preview_text = (text or "").rstrip('\r\n')
         self.drawing_area.queue_draw()
+
+    def set_visualizer_state(self, state: str):
+        """Track visualizer state so partial previews only render while recording."""
+        self._visualizer_state = (state or "recording").lower()
+        self.drawing_area.queue_draw()
 
     def set_visualization(self, visualization):
         """Change the visualization type."""
         self.visualization = visualization
         self.drawing_area.queue_draw()
 
     def _draw_preview_text(self, cr: cairo.Context, width: int, height: int):
-        if not self._preview_text:
+        if not self._preview_text or self._visualizer_state != "recording":
             return
 
         padding = 14

tests/test_main_startup_safety.py

@@ -58,6 +58,37 @@ def test_show_mic_osd_clears_preview_before_showing(self):
         self.assertIsNotNone(show_line)
         self.assertLess(clear_line, show_line)
 
+    def test_stop_recording_clears_preview_before_processing_state(self):
+        tree = ast.parse((ROOT / "lib" / "main.py").read_text(encoding="utf-8"))
+
+        stop_func = None
+        for node in ast.walk(tree):
+            if isinstance(node, ast.FunctionDef) and node.name == "_stop_recording":
+                stop_func = node
+                break
+
+        self.assertIsNotNone(stop_func)
+
+        clear_line = None
+        processing_line = None
+        for node in ast.walk(stop_func):
+            if not isinstance(node, ast.Call):
+                continue
+            if isinstance(node.func, ast.Attribute) and node.func.attr == "_clear_mic_osd_preview_text":
+                clear_line = node.lineno
+            elif (
+                isinstance(node.func, ast.Attribute)
+                and node.func.attr == "_set_visualizer_state"
+                and node.args
+                and isinstance(node.args[0], ast.Constant)
+                and node.args[0].value == "processing"
+            ):
+                processing_line = node.lineno
+
+        self.assertIsNotNone(clear_line)
+        self.assertIsNotNone(processing_line)
+        self.assertLess(clear_line, processing_line)
+
 
 if __name__ == "__main__":
     unittest.main()

tests/test_mic_osd_main_fallback.py

@@ -82,6 +82,25 @@ def test_cleanup_uses_preview_file_helper(self):
         )
         self.assertTrue(calls_helper)
 
+    def test_state_poll_updates_window_visualizer_state(self):
+        tree = ast.parse((ROOT / "lib" / "mic_osd" / "main.py").read_text(encoding="utf-8"))
+
+        poll_func = None
+        for node in ast.walk(tree):
+            if isinstance(node, ast.FunctionDef) and node.name == "_poll_state_file":
+                poll_func = node
+                break
+
+        self.assertIsNotNone(poll_func)
+
+        calls_window_state = any(
+            isinstance(node, ast.Call)
+            and isinstance(node.func, ast.Attribute)
+            and node.func.attr == "set_visualizer_state"
+            for node in ast.walk(poll_func)
+        )
+        self.assertTrue(calls_window_state)
+
 
 if __name__ == "__main__":
     unittest.main()

tests/test_mic_osd_runner.py

@@ -33,6 +33,35 @@ def fire(self):
         self.callback(*self.args)
 
 
+class FakeCairoContext:
+    def __init__(self):
+        self.shown_text = []
+
+    def select_font_face(self, *args):
+        pass
+
+    def set_font_size(self, *args):
+        pass
+
+    def text_extents(self, text):
+        return (0, 0, len(text) * 5, 10, 0, 0)
+
+    def set_source_rgba(self, *args):
+        pass
+
+    def rectangle(self, *args):
+        pass
+
+    def fill(self):
+        pass
+
+    def move_to(self, *args):
+        pass
+
+    def show_text(self, text):
+        self.shown_text.append(text)
+
+
 class MicOSDRunnerTests(unittest.TestCase):
     def _import_window_with_stubs(self):
         for module_name in ("mic_osd.window",):
@@ -219,6 +248,22 @@ def text_extents(self, text):
         self.assertEqual(window._text_width(ObjectContext(), "abcd"), 20)
         self.assertEqual(window._text_height(ObjectContext(), "abcd"), 10)
 
+    def test_preview_text_draws_only_while_recording(self):
+        window_module, _ = self._import_window_with_stubs()
+        window = object.__new__(window_module.OSDWindow)
+        window._preview_text = "live partial"
+        window._visualizer_state = "processing"
+
+        processing_cr = FakeCairoContext()
+        window._draw_preview_text(processing_cr, 400, 68)
+
+        window._visualizer_state = "recording"
+        recording_cr = FakeCairoContext()
+        window._draw_preview_text(recording_cr, 400, 68)
+
+        self.assertEqual(processing_cr.shown_text, [])
+        self.assertEqual(recording_cr.shown_text, ["live partial"])
+
     def test_preview_text_preserves_spaces_but_trims_newlines(self):
         with tempfile.TemporaryDirectory() as tmp:
             preview_file = Path(tmp) / "hyprwhspr" / "transcript_preview"