je-editor 0.0.204__py3-none-any.whl → 0.0.205__py3-none-any.whl

je_editor/pyside_ui/code/code_process/code_exec.py

@@ -111,7 +111,7 @@ class ExecManager(object):
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
                 stdin=subprocess.PIPE,
-                shell=True
+                shell=False
             )
             self.still_run_program = True
             # program output message queue thread

je_editor/pyside_ui/code/shell_process/shell_exec.py

@@ -44,8 +44,8 @@ class ShellManager(object):
         self.read_program_output_from_thread = None
         self.main_window: EditorWidget = main_window
         self.compiler_path = None
-        self.code_result: [QTextEdit, None] = None
-        self.timer: [QTimer, None] = None
+        self.code_result: Union[QTextEdit, None] = None
+        self.timer: Union[QTimer, None] = None
         self.still_run_shell: bool = True
         self.process = None
         self.run_output_queue: queue = queue.Queue()
@@ -157,7 +157,7 @@ class ShellManager(object):
         if self.after_done_function is not None:
             self.after_done_function()
 
-    # exit program change run flag to false and clean read thread and queue and process
+    # exit program change runs flag to false and clean read thread and queue and process
     def exit_program(self) -> None:
         jeditor_logger.info("ShellManager exit_program")
         self.still_run_shell = False

je_editor/pyside_ui/dialog/ai_dialog/set_ai_dialog.py

@@ -0,0 +1,44 @@
+from PySide6.QtWidgets import QWidget, QLineEdit, 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
+from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
+
+
+class SetAIDialog(QWidget):
+
+    def __init__(self):
+        jeditor_logger.info("Init SetAIDialog")
+        super().__init__()
+        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.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.grid_layout)
+
+    def update_ai_config(self):
+        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"))

je_editor/pyside_ui/dialog/file_dialog/create_file_dialog.py

@@ -12,13 +12,13 @@ class CreateFileDialog(QWidget):
         self.box_layout = QBoxLayout(QBoxLayout.Direction.TopToBottom)
         self.file_name_input = QLineEdit()
         self.create_file_button = QPushButton()
-        self.create_file_button.setText(language_wrapper.language_word_dict.get("dialog_create_file"))
+        self.create_file_button.setText(language_wrapper.language_word_dict.get("create_file_dialog_pushbutton"))
         self.create_file_button.clicked.connect(self.create_file)
         self.box_h_layout = QHBoxLayout()
         self.box_h_layout.addWidget(self.create_file_button)
         self.box_layout.addWidget(self.file_name_input)
         self.box_layout.addLayout(self.box_h_layout)
-        self.setWindowTitle(language_wrapper.language_word_dict.get("dialog_create_file"))
+        self.setWindowTitle(language_wrapper.language_word_dict.get("create_file_dialog_pushbutton"))
         self.setLayout(self.box_layout)
 
     def create_file(self):
@@ -26,7 +26,8 @@ class CreateFileDialog(QWidget):
         file_name = self.file_name_input.text().strip()
         if file_name == "":
             create_file_message_box = QMessageBox(self)
-            create_file_message_box.setText(language_wrapper.language_word_dict.get("dialog_input_file_name"))
+            create_file_message_box.setText(
+                language_wrapper.language_word_dict.get("input_file_name_dialog_pushbutton"))
             create_file_message_box.show()
         else:
             with open(file_name, "w+") as file:

je_editor/pyside_ui/dialog/file_dialog/save_file_dialog.py

@@ -42,3 +42,4 @@ def choose_file_get_save_file_path(parent_qt_instance: EditorMain) -> bool:
             widget.rename_self_tab()
             return True
         return False
+    return False

je_editor/pyside_ui/dialog/search_ui/search_error_box.py

@@ -12,9 +12,9 @@ class SearchResultBox(QWidget):
         self.box_layout = QBoxLayout(QBoxLayout.Direction.TopToBottom)
         self.search_input = QLineEdit()
         self.search_next_button = QPushButton()
-        self.search_next_button.setText(language_wrapper.language_word_dict.get("dialog_search_next"))
+        self.search_next_button.setText(language_wrapper.language_word_dict.get("search_next_dialog_pushbutton"))
         self.search_back_button = QPushButton()
-        self.search_back_button.setText(language_wrapper.language_word_dict.get("dialog_search_back"))
+        self.search_back_button.setText(language_wrapper.language_word_dict.get("search_back_dialog_pushbutton"))
         self.box_h_layout = QHBoxLayout()
         self.box_h_layout.addWidget(self.search_back_button)
         self.box_h_layout.addWidget(self.search_next_button)

je_editor/pyside_ui/dialog/search_ui/search_text_box.py

@@ -12,13 +12,13 @@ class SearchBox(QWidget):
         self.box_layout = QBoxLayout(QBoxLayout.Direction.TopToBottom)
         self.search_input = QLineEdit()
         self.search_next_button = QPushButton()
-        self.search_next_button.setText(language_wrapper.language_word_dict.get("dialog_search_next"))
+        self.search_next_button.setText(language_wrapper.language_word_dict.get("search_next_dialog_pushbutton"))
         self.search_back_button = QPushButton()
-        self.search_back_button.setText(language_wrapper.language_word_dict.get("dialog_search_back"))
+        self.search_back_button.setText(language_wrapper.language_word_dict.get("search_back_dialog_pushbutton"))
         self.box_h_layout = QHBoxLayout()
         self.box_h_layout.addWidget(self.search_back_button)
         self.box_h_layout.addWidget(self.search_next_button)
         self.box_layout.addWidget(self.search_input)
         self.box_layout.addLayout(self.box_h_layout)
-        self.setWindowTitle("Search Text")
+        self.setWindowTitle(language_wrapper.language_word_dict.get("search_box_dialog_title"))
         self.setLayout(self.box_layout)

je_editor/pyside_ui/main_ui/ai_widget/ai_config.py

@@ -0,0 +1,19 @@
+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": "",
+                "prompt_template": "",
+            }
+        }
+        self.message_queue = Queue()
+
+
+ai_config = AIConfig()

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)

je_editor/pyside_ui/main_ui/ai_widget/chat_ui.py

@@ -0,0 +1,130 @@
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING, Union
+
+from PySide6.QtCore import Qt, QTimer
+from PySide6.QtGui import QFontDatabase
+from PySide6.QtWidgets import QWidget, QPlainTextEdit, QScrollArea, QLabel, QComboBox, QGridLayout, QPushButton, \
+    QMessageBox, QSizePolicy, QLineEdit
+
+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
+
+
+class ChatUI(QWidget):
+
+    def __init__(self, main_window: EditorMain):
+        super().__init__()
+        self.setAttribute(Qt.WidgetAttribute.WA_DeleteOnClose)
+        self.main_window = main_window
+        # Chat panel
+        self.chat_panel = QPlainTextEdit()
+        self.chat_panel.setLineWrapMode(self.chat_panel.LineWrapMode.NoWrap)
+        self.chat_panel.setReadOnly(True)
+        self.chat_panel_scroll_area = QScrollArea()
+        self.chat_panel_scroll_area.setWidgetResizable(True)
+        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(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(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
+        self.grid_layout = QGridLayout()
+        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.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)
+        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, 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"):
+                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")) == 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)
+                    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"),
+                    )
+            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):
+            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 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")
+
+    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/pyside_ui/main_ui/ai_widget/langchain_interface.py

@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+import os
+import re
+from typing import Union, TYPE_CHECKING
+
+from PySide6.QtWidgets import QMessageBox
+from langchain.prompts.chat import SystemMessagePromptTemplate
+from langchain_openai import ChatOpenAI
+from pydantic import SecretStr
+
+from je_editor.utils.multi_language.multi_language_wrapper import language_wrapper
+
+if TYPE_CHECKING:
+    from je_editor.pyside_ui.main_ui.ai_widget.chat_ui import ChatUI
+
+class LangChainInterface(object):
+
+    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
+        self.api_key = api_key
+        self.chat_model = chat_model
+        self.main_window = main_window
+        os.environ["OPENAI_BASE_URL"] = self.base_url
+        os.environ["OPENAI_API_KEY"] = self.api_key
+        os.environ["CHAT_MODEL"] = self.chat_model
+        self.chat_ai = ChatOpenAI(base_url=self.base_url, api_key=self.api_key, model=self.chat_model)
+
+    def call_ai_model(self, prompt: str) -> str | None:
+        message = None
+        try:
+            message = self.chat_ai.invoke(prompt).text()
+            match = re.search(r"</think>\s*(.*)", message, re.DOTALL)
+            if match:
+                message = match.group(1).strip()
+            else:
+                message = message
+
+        except Exception as error:
+            QMessageBox.warning(self.main_window,
+                                language_wrapper.language_word_dict.get("call_ai_model_error_title"),
+                                str(error))
+        return message

je_editor/pyside_ui/main_ui/main_editor.py

@@ -7,7 +7,7 @@ from typing import Dict, Type
 import jedi.settings
 from PySide6.QtCore import QTimer, QEvent
 from PySide6.QtGui import QFontDatabase, QIcon, Qt
-from PySide6.QtWidgets import QMainWindow, QWidget, QTabWidget
+from PySide6.QtWidgets import QMainWindow, QWidget, QTabWidget, QMessageBox
 from frontengine import FrontEngineMainUI
 from qt_material import QtStyleTools
 

je_editor/pyside_ui/main_ui/menu/dock_menu/build_dock_menu.py

@@ -9,6 +9,7 @@ from PySide6.QtWidgets import QFileDialog
 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())

je_editor/pyside_ui/main_ui/menu/tab_menu/build_tab_menu.py

@@ -5,6 +5,7 @@ from typing import TYPE_CHECKING
 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()}")
+
+

je_editor/utils/multi_language/english.py

@@ -14,10 +14,13 @@ english_word_dict = {
     "browser_reload_button": "Reload",
     "browser_search_button": "Search",
     # Dialog
-    "dialog_create_file": "Create File",
-    "dialog_input_file_name": "Please enter right file name",
-    "dialog_search_next": "Search next",
-    "dialog_search_back": "Search back",
+    "create_file_dialog_pushbutton": "Create File",
+    "input_file_name_dialog_pushbutton": "Please enter right file name",
+    "search_next_dialog_pushbutton": "Search next",
+    "search_back_dialog_pushbutton": "Search back",
+    "search_box_dialog_title": "Search Text",
+    "add_ai_model_title": "Please fill in AI model information",
+    "add_ai_model_pushbutton": "Add AI Model",
     # Editor
     "editor_code_result": "Code result",
     "editor_format_check": "Format checker",
@@ -106,6 +109,7 @@ please make sure that the current encoding is consistent with the default encodi
     "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",
@@ -122,4 +126,19 @@ please make sure that the current encoding is consistent with the default encodi
     "language_menu_bar_please_restart_messagebox": "Please restart",
     # Qtconsole
     "please_install_qtcontsole_label": "Please python -m pip install qtconsole first.",
+    # Chat UI
+    "font_size": "Font Size",
+    "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]",
+    "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",
+    "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

@@ -14,10 +14,13 @@ traditional_chinese_word_dict = {
     "browser_reload_button": "重新載入",
     "browser_search_button": "搜尋",
     # Dialog
-    "dialog_create_file": "建立檔案",
-    "dialog_input_file_name": "請輸入正確格式的檔案名稱",
-    "dialog_search_next": "搜尋下一個",
-    "dialog_search_back": "搜尋上一個",
+    "create_file_dialog_pushbutton": "建立檔案",
+    "input_file_name_dialog_pushbutton": "請輸入正確格式的檔案名稱",
+    "search_next_dialog_pushbutton": "搜尋下一個",
+    "search_back_dialog_pushbutton": "搜尋上一個",
+    "search_box_dialog_title": "搜尋文字",
+    "add_ai_model_title": "請填寫 AI 模型資訊",
+    "add_ai_model_pushbutton": "新增 AI 模型",
     # Editor
     "editor_code_result": "程式運行結果",
     "editor_format_check": "程式格式檢查",
@@ -104,6 +107,7 @@ traditional_chinese_word_dict = {
     "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": "字體",
@@ -120,4 +124,19 @@ traditional_chinese_word_dict = {
     "language_menu_bar_please_restart_messagebox": "請重啟以應用更改",
     # Qtconsole
     "please_install_qtcontsole_label": "請先 python -m pip install qtconsole",
+    # Chat UI
+    "font_size": "字體大小",
+    "set_ai_model_waring_title": "無法設置 AI Model",
+    "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",
+    "base_url_label":"AI 伺服器 URL",
+    "api_key_label": "AI 伺服器 API Key",
+    "ai_model_label": "AI Model",
+    "load_ai_messagebox_title": "載入成功",
+    "load_ai_messagebox_text": "載入成功",
 }

je_editor/utils/redirect_manager/redirect_manager_class.py

@@ -53,7 +53,7 @@ class RedirectManager(object):
         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

{je_editor-0.0.204.dist-info → je_editor-0.0.205.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: je_editor
-Version: 0.0.204
+Version: 0.0.205
 Summary: JEditor is basic but powerful editor include GPT
 Author-email: JE-Chen <jechenmailman@gmail.com>
 Project-URL: Homepage, https://github.com/JE-Chen/je_editor
@@ -23,6 +23,9 @@ Requires-Dist: frontengine
 Requires-Dist: pycodestyle
 Requires-Dist: jedi
 Requires-Dist: qtconsole
+Requires-Dist: langchain_openai
+Requires-Dist: langchain
+Requires-Dist: pydantic
 Dynamic: license-file
 
 # je_editor

{je_editor-0.0.204.dist-info → je_editor-0.0.205.dist-info}/RECORD

@@ -15,26 +15,33 @@ je_editor/pyside_ui/code/auto_save/auto_save_thread.py,sha256=V_A8_4JZdHdF9e2E_r
 je_editor/pyside_ui/code/code_format/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/code/code_format/pep8_format.py,sha256=zrCkUdz2SG4bnMSkXfccvpHAmUOxXLedSBxL_nkdaoE,3574
 je_editor/pyside_ui/code/code_process/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/code/code_process/code_exec.py,sha256=mdkHe2MBBvgZbRjmSRgcu9KEFG_d0oyPWedYFUCZaxU,9840
+je_editor/pyside_ui/code/code_process/code_exec.py,sha256=yrkWz7lqWmbpfuNUHavFfyisKuVfhHSHS6D95Oq52Dk,9841
 je_editor/pyside_ui/code/plaintext_code_edit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/code/plaintext_code_edit/code_edit_plaintext.py,sha256=lcwmBemFenH-uDGe6fo70MS47emzPFgy9XL7cipwryY,14249
 je_editor/pyside_ui/code/shell_process/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/code/shell_process/shell_exec.py,sha256=WpWORZ-1T4G_SdnJ4K_6THycNOpQbsIFgjSQFQOJ_1M,8730
+je_editor/pyside_ui/code/shell_process/shell_exec.py,sha256=y1SxB2NnblB3U7qOwQxuJv7aMgBiBSMTY9gP5X8-j4E,8741
 je_editor/pyside_ui/code/syntax/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/code/syntax/python_syntax.py,sha256=9rK7_DYAANgf28PUdVl9xs42b76Re_SWwIpijEGBuzQ,3054
 je_editor/pyside_ui/code/syntax/syntax_setting.py,sha256=oaLRF_83Oa6bcAorf7AlZG0Mrt4FmsK7b4aA-LS_bRo,2284
 je_editor/pyside_ui/code/textedit_code_result/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/code/textedit_code_result/code_record.py,sha256=pT-CmcADEaXLZ0ih6D6rpe1XywgszrEpcJxtJYgopJE,2162
 je_editor/pyside_ui/dialog/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+je_editor/pyside_ui/dialog/ai_dialog/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+je_editor/pyside_ui/dialog/ai_dialog/set_ai_dialog.py,sha256=w_cku2dQthtk0TDSxEeDB-tvzdHV4vmUhFVDlb2ynO8,2392
 je_editor/pyside_ui/dialog/file_dialog/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/dialog/file_dialog/create_file_dialog.py,sha256=XwVmVYv5SdZXHb_dHD2qfGf6q8LzxmD-6PD16W8cLm0,1599
+je_editor/pyside_ui/dialog/file_dialog/create_file_dialog.py,sha256=3NvGVld_If2TsM_Xt0pgWDTcj51szS8MxG1raU-tP2A,1650
 je_editor/pyside_ui/dialog/file_dialog/open_file_dialog.py,sha256=wTI4TwMfJ-960c_C0cLy2EhWbZosewSmpyWOrcf9yLE,3806
-je_editor/pyside_ui/dialog/file_dialog/save_file_dialog.py,sha256=ON8brmfaBn7ySFob-dU58zn5BFPy-7mFtCDe9RaFWko,1763
+je_editor/pyside_ui/dialog/file_dialog/save_file_dialog.py,sha256=pyPfwgrDiDpMKk8T_4oBukgJLxdWnzIKSziwFt4YuUU,1781
 je_editor/pyside_ui/dialog/search_ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/dialog/search_ui/search_error_box.py,sha256=GzMXMnIL1k8ZwG3A2a_oMgiUiXXqd7HKW45YlVlL_eo,1168
-je_editor/pyside_ui/dialog/search_ui/search_text_box.py,sha256=H2MbC-RGpKg9TzjV4MZj7Bw7qs9a9E4cjA4GI5zsv4U,1154
+je_editor/pyside_ui/dialog/search_ui/search_error_box.py,sha256=jSa423cQW_xQ2WAj46CCWPsCP0Gy7g-UDe5o8LQyKQk,1190
+je_editor/pyside_ui/dialog/search_ui/search_text_box.py,sha256=ul-98FVByq_TNkkLZcBSF4VSwIQNBBj-4e6jYlh6YEA,1229
 je_editor/pyside_ui/main_ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/main_ui/main_editor.py,sha256=S2NOZ7NaLtDM-sesxcPaN_FAYu69e92ZkKdBUILY1PM,11062
+je_editor/pyside_ui/main_ui/main_editor.py,sha256=NCyNb20qLg-PRgjuyHB3UxOcIWfSDgvO0mn7SF2Jdgk,11075
+je_editor/pyside_ui/main_ui/ai_widget/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+je_editor/pyside_ui/main_ui/ai_widget/ai_config.py,sha256=Uc9UkrIeALqjmZCF5ukXtSgU1EB4xa5223gk3CfYXX4,459
+je_editor/pyside_ui/main_ui/ai_widget/ask_thread.py,sha256=ma5Cnwlr0RbU-sytmodcYabWueyGE8DKKISZ_0zf3Eo,571
+je_editor/pyside_ui/main_ui/ai_widget/chat_ui.py,sha256=yBoWXvGvgUCWjmPoBcOhj0mtOm0R42mYEs7FokYQmnE,6763
+je_editor/pyside_ui/main_ui/ai_widget/langchain_interface.py,sha256=OgBaUcUMR0Z2VQUO9QfCX2wJc_qBFw3IvoMCiEGBTug,1768
 je_editor/pyside_ui/main_ui/dock/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/main_ui/dock/destroy_dock.py,sha256=MTN45BykNm6FA4gMW7gI4Kr9orTdcxVTm7mch3DUhaw,604
 je_editor/pyside_ui/main_ui/editor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -48,7 +55,7 @@ je_editor/pyside_ui/main_ui/menu/set_menu_bar.py,sha256=I12DXLyRO4cKe17fQY-QDazK
 je_editor/pyside_ui/main_ui/menu/check_style_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/main_ui/menu/check_style_menu/build_check_style_menu.py,sha256=1wD6FI21Dw3IcQONuTkHP9rUPGwY70OvCtSc-ZcTKMk,3793
 je_editor/pyside_ui/main_ui/menu/dock_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/main_ui/menu/dock_menu/build_dock_menu.py,sha256=IBRpoZxsyrbeWdz86KNyfmkwa-q4_tThf9IpCy-Aqa0,4983
+je_editor/pyside_ui/main_ui/menu/dock_menu/build_dock_menu.py,sha256=74djXCmevdbGs61EBiIuvWUnHrN41DCZJw6aPEkLiK8,5600
 je_editor/pyside_ui/main_ui/menu/file_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/main_ui/menu/file_menu/build_file_menu.py,sha256=YT6kvXShRbfM4Qo51VveO_IK8uDqlNrGwHZUiM6_Tes,6881
 je_editor/pyside_ui/main_ui/menu/help_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -67,7 +74,7 @@ je_editor/pyside_ui/main_ui/menu/run_menu/under_run_menu/utils.py,sha256=zk5adQ7
 je_editor/pyside_ui/main_ui/menu/style_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/main_ui/menu/style_menu/build_style_menu.py,sha256=OjfcjQ7fA1Z1rMwIS9oGkIkC-jy-e0jdGoYo1btTqUQ,1852
 je_editor/pyside_ui/main_ui/menu/tab_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/pyside_ui/main_ui/menu/tab_menu/build_tab_menu.py,sha256=-vi3E259R8c9RQPjudAr7SoJ_X2ocBHPEAlTeihPoG0,4877
+je_editor/pyside_ui/main_ui/menu/tab_menu/build_tab_menu.py,sha256=ZP_HHXmQYguSVQpzK-WI0Jb-PgLvKRMW5GlfGXBPQcA,5691
 je_editor/pyside_ui/main_ui/menu/text_menu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/pyside_ui/main_ui/menu/text_menu/build_text_menu.py,sha256=k3pOpK4i8ccs3D8VIrNx8g3kUDp1fZdqv9H5uEvNl5c,3558
 je_editor/pyside_ui/main_ui/save_settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -94,15 +101,15 @@ je_editor/utils/json_format/json_process.py,sha256=tszo48OnVivVkuuPrl-FtGaaq-1YO
 je_editor/utils/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/utils/logging/loggin_instance.py,sha256=OFdWgra201B4f1vDM35M9L-47DK-8j-5HQEAq_ySSKM,841
 je_editor/utils/multi_language/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/utils/multi_language/english.py,sha256=gR0utQPeyNhcpRNaaaa05mFC1ErGWPvC4nEQzAL_3D8,5642
+je_editor/utils/multi_language/english.py,sha256=A9I54gmcCNPEcFwNH_EMUkGYasSusiNUPLbFUTJewsE,6637
 je_editor/utils/multi_language/multi_language_wrapper.py,sha256=_MtYrE51poQ1p_iWvT2eg_604uj6IsiBgGpifWpYLGc,1050
-je_editor/utils/multi_language/traditional_chinese.py,sha256=Vf6tpdE0Rb8hiVBGHLpSioi4QrRXSsP5-iktePZGk0w,5668
+je_editor/utils/multi_language/traditional_chinese.py,sha256=MPCBCwd7KnSHgq86z77jS1EECnlFnpmRMKeqRO_fUM4,6667
 je_editor/utils/redirect_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-je_editor/utils/redirect_manager/redirect_manager_class.py,sha256=2tLTOnxpvcYvtO3R91iHaaeN7EOE3uTgLg-UlASy5Ac,2218
+je_editor/utils/redirect_manager/redirect_manager_class.py,sha256=CHASOUD0Uob5jZrxK_lK9SE1TE1WCZEUfDPRGeJNSEs,2269
 je_editor/utils/venv_check/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 je_editor/utils/venv_check/check_venv.py,sha256=oCrMdue4NYUUGrVifh_iHFwIgxVx9azYN4jz3Xiulgg,999
-je_editor-0.0.204.dist-info/licenses/LICENSE,sha256=KMhUHh6pnIUvmXDW-49L_Sz63bqkOlPDqsecaqKiitU,1091
-je_editor-0.0.204.dist-info/METADATA,sha256=DkmoT7tw_70DnSWlAx_ZlXcOkMnWkPRdfHOpSyCCrpA,3314
-je_editor-0.0.204.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-je_editor-0.0.204.dist-info/top_level.txt,sha256=_9YA7BgxpkmdLs-5V_UQILxClcMRgPyG1a3qaE-Bkcs,10
-je_editor-0.0.204.dist-info/RECORD,,
+je_editor-0.0.205.dist-info/licenses/LICENSE,sha256=KMhUHh6pnIUvmXDW-49L_Sz63bqkOlPDqsecaqKiitU,1091
+je_editor-0.0.205.dist-info/METADATA,sha256=7oCIeF9Yky6gvTDFWrgkxwTSWHNjkWC_9iEJY1tg6dk,3398
+je_editor-0.0.205.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+je_editor-0.0.205.dist-info/top_level.txt,sha256=_9YA7BgxpkmdLs-5V_UQILxClcMRgPyG1a3qaE-Bkcs,10
+je_editor-0.0.205.dist-info/RECORD,,

{je_editor-0.0.204.dist-info → je_editor-0.0.205.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any