Disable sending messages until response/error received

Author: Cubicpath

src/chatgpt_gui/gui/widgets/conversation_view.py

@@ -11,6 +11,7 @@
 from PySide6.QtGui import *
 from PySide6.QtWidgets import *
 
+from ...models import DistributedCallable
 from ...network.client import Conversation
 from ...network.client import Message
 from ...utils import init_layouts
@@ -29,6 +30,7 @@ def __init__(self, conversation: Conversation | None = None, *args, **kwargs) ->
         super().__init__(*args, **kwargs)
         self.conversation = conversation if conversation is not None else Conversation()
 
+        self.is_waiting: bool = False
         self._init_ui()
 
     def _init_ui(self) -> None:
@@ -111,6 +113,10 @@ def toggle_multiline():
         self.output.setPlaceholderText(tr('gui.output_text.placeholder'))
 
         app().client.receivedMessage.connect(self.receive_message)
+        app().client.receivedError.connect(DistributedCallable((
+            lambda _: setattr(self, 'is_waiting', False),
+            lambda _: self.send_button.setDisabled(False)
+        )))
 
     def append_to_view(self, text: str) -> None:
         """Append some new text to the output.
@@ -128,13 +134,15 @@ def send_message(self) -> None:
 
         Clears the input and disables after sending.
         """
-        message: str = self.input_line.text() or self.input_multi.toPlainText()
-        self.input_line.clear()
-        self.input_multi.clear()
-        self.send_button.setDisabled(True)
-        self.append_to_view(tr('gui.output_text.you_prompt', message, key_eval=False))  # type: ignore
+        if not self.is_waiting:
+            message: str = self.input_line.text() or self.input_multi.toPlainText()
+            self.input_line.clear()
+            self.input_multi.clear()
+            self.is_waiting = True
+            self.send_button.setDisabled(True)
+            self.append_to_view(tr('gui.output_text.you_prompt', message, key_eval=False))  # type: ignore
 
-        app().client.send_message(message, self.conversation)  # type: ignore
+            app().client.send_message(message, self.conversation)  # type: ignore
 
     def receive_message(self, message: Message, conversation: Conversation) -> None:
         """Receive a response from the client.
@@ -143,5 +151,6 @@ def receive_message(self, message: Message, conversation: Conversation) -> None:
         :param conversation: Conversation received in.
         """
         if conversation is self.conversation:
+            self.is_waiting = False
             self.send_button.setDisabled(False)
             self.append_to_view(tr('gui.output_text.ai_prompt', message.text, key_eval=False))

src/chatgpt_gui/resources/lang/en.default.json

@@ -28,6 +28,8 @@
     "warnings.empty_token.title": "No Session",
     "errors.authentication_failed.description": "Failed to login to <b>%s</b>.<br><br>Error: %s",
     "errors.authentication_failed.title": "Authentication Failed",
+    "errors.client_http_error.description": "Error %i: %s",
+    "errors.client_http_error.title": "Client HTTP Error: %s",
     "errors.missing_package.cancel": "Cancel",
     "errors.missing_package.description": "The action \"%s\" requires the \"{0}\" package, which is not currently installed.\n\nPressing \"{errors.missing_package.install}\" will install it using the following command:\n\n\"%s -m pip install {0}\"",
     "errors.missing_package.install": "Install",
@@ -85,6 +87,8 @@
     "gui.menus.account.sign_in": "Sign In...",
     "gui.menus.account.sign_out": "Sign Out...",
     "gui.menus.account.signed_in_as": "       Signed-in as: %s",
+    "gui.menus.conversations.import": "Import Conversation From...",
+    "gui.menus.conversations.export": "Export Conversation To...",
     "gui.menus.help": "Help",
     "gui.menus.help.about": "About",
     "gui.menus.help.about_qt": "About Qt",
@@ -129,6 +133,7 @@
     "network.http.codes.403.description": "Request forbidden -- You cannot get this resource with or without an access token.",
     "network.http.codes.404.description": "No resource found at the given location.",
     "network.http.codes.405.description": "Invalid method -- GET requests are not accepted for this resource.",
-    "network.http.codes.406.description": "Client does not support the given resource format."
+    "network.http.codes.406.description": "Client does not support the given resource format.",
+    "network.http.codes.429.description": "You have sent too many requests, and have been rate limited by OpenAI."
   }
 }