Add load complete QMessageBox

Add load complete QMessageBox

Author: JE-Chen

je_editor/pyside_ui/dialog/ai_dialog/set_ai_dialog.py

@@ -1,4 +1,4 @@
-from PySide6.QtWidgets import QWidget, QBoxLayout, QLineEdit, QHBoxLayout, QPushButton, QMessageBox
+from PySide6.QtWidgets import QWidget, QBoxLayout, QLineEdit, QHBoxLayout, QPushButton, QMessageBox, QGridLayout, QLabel
 
 from je_editor.pyside_ui.main_ui.ai_widget.ai_config import ai_config
 from je_editor.utils.logging.loggin_instance import jeditor_logger
@@ -10,21 +10,25 @@ class SetAIDialog(QWidget):
     def __init__(self):
         jeditor_logger.info("Init SetAIDialog")
         super().__init__()
-        self.box_layout = QBoxLayout(QBoxLayout.Direction.TopToBottom)
+        self.base_url_label = QLabel(language_wrapper.language_word_dict.get("base_url_label"))
         self.base_url_input = QLineEdit()
+        self.api_key_label = QLabel(language_wrapper.language_word_dict.get("api_key_label"))
         self.api_key_input = QLineEdit()
+        self.chat_model_label = QLabel(language_wrapper.language_word_dict.get("ai_model_label"))
         self.chat_model_input = QLineEdit()
         self.add_ai_info_button = QPushButton()
         self.add_ai_info_button.setText(language_wrapper.language_word_dict.get("add_ai_model_pushbutton"))
         self.add_ai_info_button.clicked.connect(self.update_ai_config)
-        self.box_h_layout = QHBoxLayout()
-        self.box_h_layout.addWidget(self.add_ai_info_button)
-        self.box_layout.addWidget(self.base_url_input)
-        self.box_layout.addWidget(self.api_key_input)
-        self.box_layout.addWidget(self.chat_model_input)
-        self.box_layout.addLayout(self.box_h_layout)
+        self.grid_layout = QGridLayout()
+        self.grid_layout.addWidget(self.base_url_label, 0, 0)
+        self.grid_layout.addWidget(self.base_url_input, 0, 1)
+        self.grid_layout.addWidget(self.api_key_label, 1, 0)
+        self.grid_layout.addWidget(self.api_key_input, 1, 1)
+        self.grid_layout.addWidget(self.chat_model_label, 2, 0)
+        self.grid_layout.addWidget(self.chat_model_input, 2, 1)
+        self.grid_layout.addWidget(self.add_ai_info_button, 3, 1)
         self.setWindowTitle(language_wrapper.language_word_dict.get("add_ai_model_title"))
-        self.setLayout(self.box_layout)
+        self.setLayout(self.grid_layout)
 
     def update_ai_config(self):
         base_url = self.base_url_input.text().strip()

je_editor/pyside_ui/main_ui/ai_widget/chat_ui.py

@@ -19,12 +19,6 @@
     from je_editor.pyside_ui.main_ui.main_editor import EditorMain
 
 
-def set_ai_config():
-    # Set and output AI a config file
-    set_ai_config_dialog = SetAIDialog()
-    set_ai_config_dialog.show()
-
-
 class ChatUI(QWidget):
 
     def __init__(self, main_window: EditorMain):
@@ -53,9 +47,9 @@ def __init__(self, main_window: EditorMain):
         self.font_size_combobox.currentTextChanged.connect(self.update_panel_text_size)
         # Buttons
         self.set_ai_config_button = QPushButton(language_wrapper.language_word_dict.get("chat_ui_set_ai_button"))
-        self.set_ai_config_button.clicked.connect(set_ai_config)
+        self.set_ai_config_button.clicked.connect(self.set_ai_config)
         self.load_ai_config_button = QPushButton(language_wrapper.language_word_dict.get("chat_ui_load_ai_button"))
-        self.load_ai_config_button.clicked.connect(self.load_ai_config)
+        self.load_ai_config_button.clicked.connect(lambda: self.load_ai_config(show_load_complete=True))
         self.call_ai_model_button = QPushButton(language_wrapper.language_word_dict.get("chat_ui_call_ai_model_button"))
         self.call_ai_model_button.clicked.connect(self.call_ai_model)
         # Add to layout
@@ -70,6 +64,8 @@ def __init__(self, main_window: EditorMain):
         # Variable
         self.ai_config: AIConfig = ai_config
         self.lang_chain_interface: Union[LangChainInterface, None] = None
+        self.set_ai_config_dialog = None
+        # Timer to pop queue
         self.pull_message_timer = QTimer(self)
         self.pull_message_timer.setInterval(1000)
         self.pull_message_timer.timeout.connect(self.pull_message)
@@ -84,7 +80,7 @@ def update_panel_text_size(self):
         self.chat_panel.setFont(
             QFontDatabase.font(self.font().family(), "", int(self.font_size_combobox.currentText())))
 
-    def load_ai_config(self):
+    def load_ai_config(self, show_load_complete: bool = False):
         ai_config_file = Path(str(Path.cwd()) + "/" + ".jeditor/ai_config.json")
         if ai_config_file.exists():
             with open(ai_config_file, "r", encoding="utf-8"):
@@ -101,6 +97,12 @@ def load_ai_config(self):
                         chat_model=ai_info.get("chat_model"),
                         prompt_template=ai_info.get("prompt_template"),
                     )
+            if show_load_complete:
+                load_complete = QMessageBox(self)
+                load_complete.setWindowTitle(language_wrapper.language_word_dict.get("load_ai_messagebox_title"))
+                load_complete.setText(language_wrapper.language_word_dict.get("load_ai_messagebox_text"))
+                load_complete.exec()
+
 
     def call_ai_model(self):
         if isinstance(self.lang_chain_interface, LangChainInterface):
@@ -121,3 +123,8 @@ def pull_message(self):
             ai_response = ai_config.message_queue.get_nowait()
             self.chat_panel.appendPlainText(ai_response)
             self.chat_panel.appendPlainText("\n")
+
+    def set_ai_config(self):
+        # Set and output AI a config file
+        self.set_ai_config_dialog = SetAIDialog()
+        self.set_ai_config_dialog.show()

je_editor/utils/multi_language/english.py

@@ -136,4 +136,9 @@
     "chat_ui_set_ai_button": "Set AI setting",
     "chat_ui_load_ai_button": "Load AI setting",
     "chat_ui_call_ai_model_button": "Send prompt",
+    "base_url_label": "AI server URL",
+    "api_key_label": "AI server API Key",
+    "ai_model_label": "AI Model",
+    "load_ai_messagebox_title": "Load complete",
+    "load_ai_messagebox_text": "Load complete",
 }

je_editor/utils/multi_language/traditional_chinese.py

@@ -134,4 +134,9 @@
     "chat_ui_set_ai_button": "設定 AI 設定",
     "chat_ui_load_ai_button": "載入 AI 設定",
     "chat_ui_call_ai_model_button": "傳送 prompt",
+    "base_url_label":"AI 伺服器 URL",
+    "api_key_label": "AI 伺服器 API Key",
+    "ai_model_label": "AI Model",
+    "load_ai_messagebox_title": "載入成功",
+    "load_ai_messagebox_text": "載入成功",
 }