Bugfix #2

Author: yjg30737

pyqt_openai/chat_widget/center/commandSuggestionWidget.py

@@ -1,9 +1,12 @@
+from PySide6.QtCore import Signal
 from PySide6.QtWidgets import QListWidget, QWidget, QVBoxLayout, QLabel
 
 from pyqt_openai.lang.translations import LangClass
 
 
 class CommandSuggestionWidget(QWidget):
+    toggleCommandSuggestion = Signal(bool)
+
     def __init__(self, parent=None):
         super().__init__(parent)
         self.__initUi()
@@ -29,3 +32,9 @@ def onCountChanged(self):
         scrollbarHeight = self.__commandList.verticalScrollBar().sizeHint().height()
         totalHeight = contentHeight + itemHeight
         self.setMaximumHeight(totalHeight + scrollbarHeight)
+
+    def showEvent(self, event):
+        self.toggleCommandSuggestion.emit(True)
+
+    def hideEvent(self, event):
+        self.toggleCommandSuggestion.emit(False)

pyqt_openai/chat_widget/center/prompt.py

@@ -87,6 +87,9 @@ def __initUi(self):
 
         # Create the command suggestion list
         self.__suggestionWidget = CommandSuggestionWidget()
+        self.__suggestionWidget.toggleCommandSuggestion.connect(
+            self.__setCommandSuggestionEnabled
+        )
         self.__suggestion_list = self.__suggestionWidget.getCommandList()
 
         self.__uploadedImageFileWidget = UploadedImageFileWidget()
@@ -247,6 +250,13 @@ def __getEveryPromptCommands(self, get_name_only=False):
             return [obj["name"] for obj in self.__p_grp]
         return self.__p_grp
 
+    def __setCommandSuggestionEnabled(self, f):
+        for w in self.__textEditGroup.getGroup().values():
+            if isinstance(w, JSONEditor):
+                pass
+            else:
+                w.setCommandSuggestionEnabled(f)
+
     def __updateSuggestions(self):
         w = self.__textEditGroup.getCurrentTextEdit()
         if w and self.__commandEnabled:
@@ -257,7 +267,6 @@ def __updateSuggestions(self):
                 input_text_chunk = input_text_chunk[-1]
                 starts_with_f = input_text_chunk.startswith("/")
                 self.__suggestionWidget.setVisible(starts_with_f)
-                w.setCommandSuggestionEnabled(starts_with_f)
                 if starts_with_f:
                     command_word = input_text_chunk[1:]
                     # Set every prompt commands first
@@ -339,7 +348,6 @@ def __showEnding(self, f):
 
     def __supportPromptCommand(self, f):
         self.__commandEnabled = f
-        self.__textEditGroup.setCommandEnabled(f)
 
     def __readingFiles(self):
         filenames = QFileDialog.getOpenFileNames(

pyqt_openai/chat_widget/center/textEditPromptGroup.py

@@ -95,11 +95,6 @@ def getCurrentTextEdit(self):
             if w.hasFocus():
                 return w
 
-    def setCommandEnabled(self, f: bool):
-        for w in self.__textGroup.values():
-            if isinstance(w, TextEditPrompt):
-                w.setCommandSuggestionEnabled(f)
-
     def adjustHeight(self) -> int:
         """
         Adjust overall height of text edit group based on their contents and return adjusted height