Add chat ui

JE-Chen · JE-Chen · commit ac020e58393e · 2025-05-29T15:09:30.000+08:00
Add chat ui
* Add Chat with AI ui and classes
diff --git a/je_editor/pyside_ui/dialog/ai_dialog/set_ai_dialog.py b/je_editor/pyside_ui/dialog/ai_dialog/set_ai_dialog.py
@@ -1,5 +1,6 @@
-from PySide6.QtWidgets import QWidget, QBoxLayout, QLineEdit, QHBoxLayout, QPushButton
+from PySide6.QtWidgets import QWidget, QBoxLayout, QLineEdit, QHBoxLayout, QPushButton, QMessageBox
 
+from je_editor.pyside_ui.main_ui.ai_widget.ai_config import ai_config
 from je_editor.utils.logging.loggin_instance import jeditor_logger
 from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
 
@@ -26,4 +27,14 @@ def __init__(self):
         self.setLayout(self.box_layout)
 
     def update_ai_config(self):
-        pass
+        base_url = self.base_url_input.text().strip()
+        api_key = self.api_key_input.text().strip()
+        chat_model = self.chat_model_input.text().strip()
+        if base_url and chat_model:
+            ai_config.choosable_ai.update(
+                {"AI_model": {"base_url": base_url, "api_key": api_key, "chat_model": chat_model}}
+            )
+        else:
+            QMessageBox.warning(self,
+                                language_wrapper.language_word_dict.get("set_ai_model_warring_title"),
+                                language_wrapper.language_word_dict.get("set_ai_model_warring_text"))
diff --git a/je_editor/pyside_ui/main_ui/ai_widget/ai_config.py b/je_editor/pyside_ui/main_ui/ai_widget/ai_config.py
@@ -1,13 +1,19 @@
-from pathlib import Path
-
-from je_editor.utils.json.json_file import read_json
+from queue import Queue
 
 
 class AIConfig(object):
 
     def __init__(self):
         self.current_ai_model_system_prompt: str = ""
         self.choosable_ai: dict[str, dict[str, str]] = {
-            "AI_model": {"ai_base_url": "", "ai_api_key": "", "chat_model": ""}
+            "AI_model": {
+                "ai_base_url": "",
+                "ai_api_key": "",
+                "chat_model": "",
+                "prompt_template": "",
+            }
         }
+        self.message_queue = Queue()
+
 
+ai_config = AIConfig()
diff --git a/je_editor/pyside_ui/main_ui/ai_widget/ask_thread.py b/je_editor/pyside_ui/main_ui/ai_widget/ask_thread.py
@@ -0,0 +1,17 @@
+from threading import Thread
+
+from je_editor.pyside_ui.main_ui.ai_widget.ai_config import ai_config
+from je_editor.pyside_ui.main_ui.ai_widget.langchain_interface import LangChainInterface
+
+
+class AskThread(Thread):
+
+    def __init__(self, lang_chain_interface: LangChainInterface, prompt):
+        super().__init__()
+        self.lang_chain_interface = lang_chain_interface
+        self.prompt = prompt
+
+
+    def run(self):
+        ai_response = self.lang_chain_interface.call_ai_model(prompt=self.prompt)
+        ai_config.message_queue.put(ai_response)
diff --git a/je_editor/pyside_ui/main_ui/ai_widget/chat_ui.py b/je_editor/pyside_ui/main_ui/ai_widget/chat_ui.py
@@ -1,20 +1,30 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Union
 
-from PySide6.QtCore import Qt
+from PySide6.QtCore import Qt, QTimer
 from PySide6.QtGui import QFontDatabase
-from PySide6.QtWidgets import QWidget, QPlainTextEdit, QScrollArea, QLabel, QComboBox, QGridLayout, QMessageBox
+from PySide6.QtWidgets import QWidget, QPlainTextEdit, QScrollArea, QLabel, QComboBox, QGridLayout, QPushButton, \
+    QMessageBox, QSizePolicy, QLineEdit
 
-from je_editor.pyside_ui.main_ui.ai_widget.ai_config import AIConfig
+from je_editor.pyside_ui.dialog.ai_dialog.set_ai_dialog import SetAIDialog
+from je_editor.pyside_ui.main_ui.ai_widget.ai_config import AIConfig, ai_config
+from je_editor.pyside_ui.main_ui.ai_widget.ask_thread import AskThread
+from je_editor.pyside_ui.main_ui.ai_widget.langchain_interface import LangChainInterface
 from je_editor.utils.json.json_file import read_json
 from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
 
 if TYPE_CHECKING:
     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):
@@ -30,41 +40,84 @@ def __init__(self, main_window: EditorMain):
         self.chat_panel_scroll_area.setViewportMargins(0, 0, 0, 0)
         self.chat_panel_scroll_area.setWidget(self.chat_panel)
         self.chat_panel.setFont(QFontDatabase.font(self.font().family(), "", 16))
+        # Prompt input
+        self.prompt_input = QLineEdit()
+        self.prompt_input.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Minimum)
+        self.prompt_input.returnPressed.connect(self.call_ai_model)
         # Font size combobox
         self.font_size_label = QLabel(language_wrapper.language_word_dict.get("font_size"))
         self.font_size_combobox = QComboBox()
         for font_size in range(2, 101, 2):
             self.font_size_combobox.addItem(str(font_size))
         self.font_size_combobox.setCurrentText("16")
         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.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.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
         self.grid_layout = QGridLayout()
-        self.grid_layout.addWidget(self.chat_panel_scroll_area, 1, 0, -1, -1)
-        self.grid_layout.addWidget(self.font_size_combobox, 0, 1)
-        ai_config = AIConfig()
-        # Load AI config if exists
+        self.grid_layout.addWidget(self.chat_panel_scroll_area, 0, 0, 1, 4)
+        self.grid_layout.addWidget(self.call_ai_model_button, 1, 0)
+        self.grid_layout.addWidget(self.font_size_combobox, 1, 1)
+        self.grid_layout.addWidget(self.set_ai_config_button, 1, 2)
+        self.grid_layout.addWidget(self.load_ai_config_button, 1, 3)
+        self.grid_layout.addWidget(self.prompt_input, 2, 0, 1, 4)
+
+        # Variable
+        self.ai_config: AIConfig = ai_config
+        self.lang_chain_interface: Union[LangChainInterface, None] = None
+        self.pull_message_timer = QTimer(self)
+        self.pull_message_timer.setInterval(1000)
+        self.pull_message_timer.timeout.connect(self.pull_message)
+        self.pull_message_timer.start()
+
+        # Set layout
+        self.setLayout(self.grid_layout)
+
+        self.load_ai_config()
+
+    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):
         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") as file:
+            with open(ai_config_file, "r", encoding="utf-8"):
                 json_data: dict = read_json(str(ai_config_file))
             if json_data:
-                if json_data.get("AI_model") and len(json_data.get("AI_model")) == 3:
+                if json_data.get("AI_model") and len(json_data.get("AI_model")) == 4:
                     ai_info: dict = json_data.get("AI_model")
                     if ai_info.get("ai_base_url") and ai_info.get("chat_model"):
                         ai_config.choosable_ai.update(json_data)
-                    else:
-                        QMessageBox.warning(self.main_window,
-                                            language_wrapper.language_word_dict.get("set_ai_model_warring_title"),
-                                            language_wrapper.language_word_dict.get("set_ai_model_warring_text"))
-
-    def update_panel_text_size(self):
-        self.chat_panel.setFont(
-            QFontDatabase.font(self.font().family(), "", int(self.font_size_combobox.currentText())))
+                    self.lang_chain_interface = LangChainInterface(
+                        main_window=self,
+                        api_key=ai_info.get("ai_api_key"),
+                        base_url=ai_info.get("ai_base_url"),
+                        chat_model=ai_info.get("chat_model"),
+                        prompt_template=ai_info.get("prompt_template"),
+                    )
 
-    def set_ai_config(self):
-        # Set and output AI a config file
-        pass
+    def call_ai_model(self):
+        if isinstance(self.lang_chain_interface, LangChainInterface):
+            thread = AskThread(lang_chain_interface=self.lang_chain_interface, prompt=self.prompt_input.text())
+            thread.start()
+        else:
+            ai_info = ai_config.choosable_ai.get('AI_model')
+            QMessageBox.warning(self,
+                                language_wrapper.language_word_dict.get("call_ai_model_error_title"),
+                                language_wrapper.language_word_dict.get(
+                                    f"ai_api_key: {ai_info.get('ai_api_key')}, \n"
+                                    f"ai_base_url: {ai_info.get('ai_base_url')}, \n"
+                                    f"chat_model: {ai_info.get('chat_model')}, \n"
+                                    f"prompt_template: {ai_info.get('prompt_template')}"))
 
-    def load_ai_config(self):
-        # Load exists an AI config file
-        pass
+    def pull_message(self):
+        if not ai_config.message_queue.empty():
+            ai_response = ai_config.message_queue.get_nowait()
+            self.chat_panel.appendPlainText(ai_response)
+            self.chat_panel.appendPlainText("\n")
diff --git a/je_editor/pyside_ui/main_ui/ai_widget/langchain_interface.py b/je_editor/pyside_ui/main_ui/ai_widget/langchain_interface.py
@@ -12,11 +12,11 @@
 from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
 
 if TYPE_CHECKING:
-    from je_editor.pyside_ui.main_ui.main_editor import EditorMain
+    from je_editor.pyside_ui.main_ui.ai_widget.chat_ui import ChatUI
 
 class LangChainInterface(object):
 
-    def __init__(self, main_window: EditorMain, prompt_template: str, base_url: str, api_key: Union[SecretStr, str],
+    def __init__(self, main_window: ChatUI, prompt_template: str, base_url: str, api_key: Union[SecretStr, str],
                  chat_model: str):
         self.system_message_prompt = SystemMessagePromptTemplate.from_template(prompt_template)
         self.base_url = base_url
@@ -41,5 +41,5 @@ def call_ai_model(self, prompt: str) -> str | None:
         except Exception as error:
             QMessageBox.warning(self.main_window,
                                 language_wrapper.language_word_dict.get("call_ai_model_error_title"),
-                                language_wrapper.language_word_dict.get(repr(error)))
+                                str(error))
         return message
diff --git a/je_editor/pyside_ui/main_ui/menu/dock_menu/build_dock_menu.py b/je_editor/pyside_ui/main_ui/menu/dock_menu/build_dock_menu.py
@@ -9,6 +9,7 @@
 from frontengine import FrontEngineMainUI
 
 from je_editor.pyside_ui.browser.browser_widget import BrowserWidget
+from je_editor.pyside_ui.main_ui.ai_widget.chat_ui import ChatUI
 from je_editor.pyside_ui.main_ui.dock.destroy_dock import DestroyDock
 from je_editor.pyside_ui.main_ui.editor.editor_widget_dock import FullEditorWidget
 from je_editor.pyside_ui.main_ui.ipython_widget.rich_jupyter import IpythonWidget
@@ -59,7 +60,13 @@ def set_dock_menu(ui_we_want_to_set: EditorMain) -> None:
         lambda: add_dock_widget(ui_we_want_to_set, "ipython")
     )
     ui_we_want_to_set.dock_menu.addAction(ui_we_want_to_set.dock_menu.new_ipython)
-
+    # ChatUI
+    ui_we_want_to_set.dock_menu.new_chat_ui = QAction(
+        language_wrapper.language_word_dict.get("chat_ui_dock_label"))
+    ui_we_want_to_set.dock_menu.new_chat_ui.triggered.connect(
+        lambda: add_dock_widget(ui_we_want_to_set, "chat_ui")
+    )
+    ui_we_want_to_set.dock_menu.addAction(ui_we_want_to_set.dock_menu.new_chat_ui)
 
 def add_dock_widget(ui_we_want_to_set: EditorMain, widget_type: str = None):
     jeditor_logger.info("build_dock_menu.py add_dock_widget "
@@ -90,6 +97,9 @@ def add_dock_widget(ui_we_want_to_set: EditorMain, widget_type: str = None):
     elif widget_type == "ipython":
         dock_widget.setWindowTitle(language_wrapper.language_word_dict.get("dock_ipython_title"))
         dock_widget.setWidget(IpythonWidget(ui_we_want_to_set))
+    elif widget_type == "chat_ui":
+        dock_widget.setWindowTitle(language_wrapper.language_word_dict.get("chat_ui_dock_label"))
+        dock_widget.setWidget(ChatUI(ui_we_want_to_set))
     else:
         dock_widget.setWindowTitle(language_wrapper.language_word_dict.get("dock_browser_title"))
         dock_widget.setWidget(BrowserWidget())
diff --git a/je_editor/pyside_ui/main_ui/menu/tab_menu/build_tab_menu.py b/je_editor/pyside_ui/main_ui/menu/tab_menu/build_tab_menu.py
@@ -5,6 +5,7 @@
 from frontengine import FrontEngineMainUI
 
 from je_editor.pyside_ui.browser.browser_widget import BrowserWidget
+from je_editor.pyside_ui.main_ui.ai_widget.chat_ui import ChatUI
 from je_editor.pyside_ui.main_ui.editor.editor_widget import EditorWidget
 from je_editor.pyside_ui.main_ui.ipython_widget.rich_jupyter import IpythonWidget
 from je_editor.utils.logging.loggin_instance import jeditor_logger
@@ -55,6 +56,13 @@ def set_tab_menu(ui_we_want_to_set: EditorMain) -> None:
         lambda: add_ipython_tab(ui_we_want_to_set)
     )
     ui_we_want_to_set.tab_menu.addAction(ui_we_want_to_set.tab_menu.add_ipython_action)
+    # ChatUI
+    ui_we_want_to_set.tab_menu.add_chat_ui_action = QAction(
+        language_wrapper.language_word_dict.get("tab_menu_chat_ui_tab_name"))
+    ui_we_want_to_set.tab_menu.add_chat_ui_action.triggered.connect(
+        lambda: add_chat_ui_tab(ui_we_want_to_set)
+    )
+    ui_we_want_to_set.tab_menu.addAction(ui_we_want_to_set.tab_menu.add_chat_ui_action)
 
 
 def add_editor_tab(ui_we_want_to_set: EditorMain):
@@ -98,3 +106,12 @@ def add_ipython_tab(ui_we_want_to_set: EditorMain):
         f"{language_wrapper.language_word_dict.get('tab_menu_ipython_tab_name')} "
         f"{ui_we_want_to_set.tab_widget.count()}")
 
+
+def add_chat_ui_tab(ui_we_want_to_set: EditorMain):
+    jeditor_logger.info(f"build_tab_menu.py add_ipython_tab ui_we_want_to_set: {ui_we_want_to_set}")
+    ui_we_want_to_set.tab_widget.addTab(
+        ChatUI(ui_we_want_to_set),
+        f"{language_wrapper.language_word_dict.get('tab_menu_chat_ui_tab_name')} "
+        f"{ui_we_want_to_set.tab_widget.count()}")
+
+
diff --git a/je_editor/utils/multi_language/english.py b/je_editor/utils/multi_language/english.py
@@ -109,6 +109,7 @@
     "tab_menu_web_tab_name": "WEB Browser",
     "tab_menu_stackoverflow_tab_name": "Stackoverflow",
     "tab_menu_ipython_tab_name": "IPython(Jupyter)",
+    "tab_menu_chat_ui_tab_name": "ChatUI",
     # Text Menu
     "text_menu_label": "Text",
     "text_menu_label_font": "Font",
@@ -130,5 +131,9 @@
     "set_ai_model_waring_title": "Can not set AI Model",
     "set_ai_model_waring_text": "Please check [ai_base_url, ai_api_key, chat_model]",
     "call_ai_model_error_title": "Can't connect to AI Model",
-    "call_ai_model_error_text": "Please check [ai_base_url, ai_api_key, chat_model]"
+    "call_ai_model_error_text": "Please check [ai_base_url, ai_api_key, chat_model]",
+    "chat_ui_dock_label": "Chat UI",
+    "chat_ui_set_ai_button": "Set AI setting",
+    "chat_ui_load_ai_button": "Load AI setting",
+    "chat_ui_call_ai_model_button": "Send prompt",
 }
diff --git a/je_editor/utils/multi_language/traditional_chinese.py b/je_editor/utils/multi_language/traditional_chinese.py
@@ -107,6 +107,7 @@
     "tab_menu_web_tab_name": "瀏覽器",
     "tab_menu_stackoverflow_tab_name": "Stackoverflow",
     "tab_menu_ipython_tab_name": "IPython(Jupyter)",
+    "tab_menu_chat_ui_tab_name": "ChatUI",
     # Text Menu
     "text_menu_label": "文字",
     "text_menu_label_font": "字體",
@@ -129,4 +130,8 @@
     "set_ai_model_warring_text": "請檢查 [ai_base_url, ai_api_key, chat_model]",
     "call_ai_model_error_title": "無法連線到 AI 模型",
     "call_ai_model_error_text": "請檢查 [ai_base_url, ai_api_key, chat_model]",
+    "chat_ui_dock_label": "聊天 UI",
+    "chat_ui_set_ai_button": "設定 AI 設定",
+    "chat_ui_load_ai_button": "載入 AI 設定",
+    "chat_ui_call_ai_model_button": "傳送 prompt",
 }
diff --git a/je_editor/utils/redirect_manager/redirect_manager_class.py b/je_editor/utils/redirect_manager/redirect_manager_class.py
@@ -53,7 +53,7 @@ def set_redirect() -> None:
         default_logger = logging.getLogger("RedirectManager")
         default_logger.addHandler(redirect_err)
         skip_logger_list = ["JEditor", "FrontEngine",
-                            "AutomationIDE", "TestPioneer"]
+                            "AutomationIDE", "TestPioneer", "langchain", "langchain_core", "langchain_openai"]
         for name in logging.root.manager.loggerDict.keys():
             if name in skip_logger_list:
                 continue
