Add clipboard cleanup option for clipboard backend

src/desktop_x11.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import logging
 import threading
+import time
 import warnings
 from typing import Callable, Iterable
 
@@ -33,6 +34,7 @@ MOD_MAP = {
     "cmd": X.Mod4Mask,
     "command": X.Mod4Mask,
 }
+CLIPBOARD_RESTORE_DELAY_SEC = 0.15
 
 
 class X11Adapter:
@@ -70,17 +72,35 @@ class X11Adapter:
         thread = threading.Thread(target=self._listen, args=("Escape", callback), daemon=True)
         thread.start()
 
-    def inject_text(self, text: str, backend: str) -> None:
+    def inject_text(
+        self,
+        text: str,
+        backend: str,
+        *,
+        remove_transcription_from_clipboard: bool = False,
+    ) -> None:
         backend = (backend or "").strip().lower()
         if backend in ("", "clipboard"):
+            previous_clipboard = None
+            if remove_transcription_from_clipboard:
+                previous_clipboard = self._read_clipboard_text()
             self._write_clipboard(text)
             self._paste_clipboard()
+            if remove_transcription_from_clipboard:
+                time.sleep(CLIPBOARD_RESTORE_DELAY_SEC)
+                self._restore_clipboard_text(previous_clipboard)
             return
         if backend == "injection":
             self._type_text(text)
             return
         raise ValueError(f"unknown injection backend: {backend}")
 
+    def _read_clipboard_text(self) -> str | None:
+        Gtk.init([])
+        clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)
+        text = clipboard.wait_for_text()
+        return str(text) if text is not None else None
+
     def run_tray(self, state_getter: Callable[[], str], on_quit: Callable[[], None]) -> None:
         self.menu = Gtk.Menu()
         quit_item = Gtk.MenuItem(label="Quit")
@@ -165,6 +185,14 @@ class X11Adapter:
         while Gtk.events_pending():
             Gtk.main_iteration()
 
+    def _restore_clipboard_text(self, text: str | None) -> None:
+        Gtk.init([])
+        clipboard = Gtk.Clipboard.get(Gdk.SELECTION_CLIPBOARD)
+        clipboard.set_text(text or "", -1)
+        clipboard.store()
+        while Gtk.events_pending():
+            Gtk.main_iteration()
+
     def _paste_clipboard(self) -> None:
         dpy = display.Display()
         self._send_combo(dpy, ["Control_L", "Shift_L", "v"])