Hide screen capture with ESC

nrl-ai · Dec 13, 2024 · ca8052d · ca8052d
1 parent e88e94d
commit ca8052d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/llama_assistant/screen_capture_widget.py b/llama_assistant/screen_capture_widget.py
@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING
 from PyQt5.QtWidgets import QApplication, QWidget, QDesktopWidget, QPushButton
 from PyQt5.QtCore import Qt, QRect
-from PyQt5.QtGui import QPainter, QColor, QPen
+from PyQt5.QtGui import QPainter, QColor, QPen, QKeyEvent
 
 from llama_assistant import config
 from llama_assistant.ocr_engine import OCREngine
@@ -74,6 +74,11 @@ def hide(self):
         self.button_widget.hide()
         super().hide()
 
+    def keyPressEvent(self, event: QKeyEvent):
+        if event.key() == Qt.Key_Escape:
+            self.hide()
+            self.parent.show()
+
     def mousePressEvent(self, event):
         if event.button() == Qt.LeftButton:
             self.start_point = event.pos()  # Capture start position