Format, ready for paste txt file feature

yjg30737 · yjg30737 · commit 3f0e3f0dc35d · 2024-11-05T18:01:13.000+09:00
diff --git a/pyqt_openai/__init__.py b/pyqt_openai/__init__.py
@@ -76,13 +76,15 @@ def get_config_directory():
 
     return config_dir
 
+
 BIN_DIR = get_config_directory()
 
 UPDATER_NAME = "Updater.exe" if sys.platform == "win32" else "Updater"
 
 # The default updater path (relative to the application's root directory) - For Windows
 UPDATER_PATH = os.path.join(BIN_DIR, UPDATER_NAME)
 
+
 # Move the binary file to the config folder to prevent "file not found" error
 def move_bin(filename, dst_dir):
     original_path = os.path.join(ROOT_DIR, filename)
@@ -91,6 +93,7 @@ def move_bin(filename, dst_dir):
             os.remove(dst_dir)
         shutil.move(original_path, dst_dir)
 
+
 move_bin(UPDATER_NAME, UPDATER_PATH)
 
 CONTACT = "yjg30737@gmail.com"
@@ -240,7 +243,6 @@ def move_bin(filename, dst_dir):
 # DEFAULT_MARKDOWN_a_color = '#000'
 
 
-
 command_key = "Ctrl"
 if sys.platform == "darwin":
     command_key = "Cmd"
@@ -431,13 +433,20 @@ def move_bin(filename, dst_dir):
     {"display_name": "OpenAI", "env_var_name": "OPENAI_API_KEY", "api_key": ""},
     {"display_name": "Gemini", "env_var_name": "GEMINI_API_KEY", "api_key": ""},
     {"display_name": "Claude", "env_var_name": "CLAUDE_API_KEY", "api_key": ""},
-
     # For G4F only
     {"display_name": "DeepInfra", "env_var_name": "DEEPINFRA_API_KEY", "api_key": ""},
     {"display_name": "Groq", "env_var_name": "GROQ_API_KEY", "api_key": ""},
-    {"display_name": "HuggingFace", "env_var_name": "HUGGINGFACE_API_KEY", "api_key": ""},
+    {
+        "display_name": "HuggingFace",
+        "env_var_name": "HUGGINGFACE_API_KEY",
+        "api_key": "",
+    },
     {"display_name": "OpenRouter", "env_var_name": "OPENROUTER_API_KEY", "api_key": ""},
-    {"display_name": "Perplexity API", "env_var_name": "PERPLEXITY_API_KEY", "api_key": ""}
+    {
+        "display_name": "Perplexity API",
+        "env_var_name": "PERPLEXITY_API_KEY",
+        "api_key": "",
+    },
 ]
 
 # This has to be managed separately since some of the arguments are different with usual models
@@ -455,7 +464,7 @@ def move_bin(filename, dst_dir):
 PROVIDER_MODEL_DICT = {
     "OpenAI": ["gpt-4o", "gpt-4o-mini"] + O1_MODELS,
     "Gemini": ["gemini-1.5-flash", "gemini-1.5-pro"],
-    "Claude": ["claude-3-5-sonnet-20240620"]
+    "Claude": ["claude-3-5-sonnet-20240620"],
 }
 
 # Constants related to the number of messages LLM will store
@@ -769,11 +778,13 @@ def move_bin(filename, dst_dir):
     },
 }
 
+
 # Dynamically add the API keys to the configuration data
 def update_general_config_with_api_keys(config_data, api_configs):
     for config in api_configs:
         config_data["General"][config["env_var_name"]] = config["api_key"]
 
+
 update_general_config_with_api_keys(CONFIG_DATA, DEFAULT_API_CONFIGS)
 
 # Set the default llama index cache directory for preventing any issues such as PermissionError
diff --git a/pyqt_openai/chat_widget/center/aiChatUnit.py b/pyqt_openai/chat_widget/center/aiChatUnit.py
@@ -6,7 +6,8 @@
     ICON_INFO,
     ICON_FAVORITE_YES,
     ICON_SPEAKER,
-    WHISPER_TTS_MODEL, ICON_FILE,
+    WHISPER_TTS_MODEL,
+    ICON_FILE,
 )
 from pyqt_openai.config_loader import CONFIG_MANAGER
 from pyqt_openai.chat_widget.center.chatUnit import ChatUnit
diff --git a/pyqt_openai/chat_widget/center/chatHome.py b/pyqt_openai/chat_widget/center/chatHome.py
@@ -8,7 +8,8 @@
     DEFAULT_APP_NAME,
     HOW_TO_GET_OPENAI_API_KEY_URL,
     LARGE_LABEL_PARAM,
-    MEDIUM_LABEL_PARAM, QUICKSTART_MANUAL_URL,
+    MEDIUM_LABEL_PARAM,
+    QUICKSTART_MANUAL_URL,
 )
 from pyqt_openai.lang.translations import LangClass
 from pyqt_openai.widgets.linkLabel import LinkLabel
diff --git a/pyqt_openai/chat_widget/center/chatUnit.py b/pyqt_openai/chat_widget/center/chatUnit.py
@@ -6,7 +6,8 @@
     QVBoxLayout,
     QHBoxLayout,
     QSpacerItem,
-    QSizePolicy, QLabel,
+    QSizePolicy,
+    QLabel,
 )
 
 from pyqt_openai import DEFAULT_ICON_SIZE, ICON_COPY
@@ -26,7 +27,7 @@ def __initUi(self):
 
         self._icon = RoundedImage()
         self._icon.setMaximumSize(*DEFAULT_ICON_SIZE)
-        self._icon.setStyleSheet('margin-right: 3px;')
+        self._icon.setStyleSheet("margin-right: 3px;")
 
         self._nameLbl = QLabel()
 
diff --git a/pyqt_openai/chat_widget/center/textEditPromptGroup.py b/pyqt_openai/chat_widget/center/textEditPromptGroup.py
@@ -10,7 +10,7 @@
     PROMPT_END_KEY_NAME,
     PROMPT_JSON_KEY_NAME,
     CONTEXT_DELIMITER,
-    IMAGE_FILE_EXT_LIST,
+    IMAGE_FILE_EXT_LIST, TEXT_FILE_EXT_LIST,
 )
 from pyqt_openai.chat_widget.center.textEditPrompt import TextEditPrompt
 from pyqt_openai.lang.translations import LangClass
@@ -22,6 +22,7 @@ class TextEditPromptGroup(QWidget):
     onUpdateSuggestion = Signal()
     onSendKeySignalToSuggestion = Signal(str)
     onPasteFile = Signal(QByteArray)
+    onPasteText = Signal(str)
 
     def __init__(self, parent=None):
         super().__init__(parent)
@@ -153,6 +154,10 @@ def handleDrop(self, urls):
                 with open(url, "rb") as f:
                     data = f.read()
                     self.onPasteFile.emit(data)
+            elif Path(url).suffix in TEXT_FILE_EXT_LIST:
+                with open(url, "r") as f:
+                    data = f.read()
+                    self.onPasteText.emit(data)
 
     def handlePaste(self):
         clipboard = QApplication.clipboard()
@@ -181,6 +186,9 @@ def handlePaste(self):
 
             # Emit the image data
             self.onPasteFile.emit(image_data)
+        elif mime_data.hasText():
+            text = mime_data.text()
+            self.onPasteText.emit(text)
         else:
             self.__textEdit.paste()
 
diff --git a/pyqt_openai/settings_dialog/apiWidget.py b/pyqt_openai/settings_dialog/apiWidget.py
@@ -64,11 +64,11 @@ def __initVal(self):
                 "Gemini": HOW_TO_GET_GEMINI_API_KEY_URL,
                 "Llama": HOW_TO_GET_LLAMA_API_KEY_URL,
                 "Replicate": HOW_TO_REPLICATE,
-                "DeepInfra": '',
-                "Groq": '',
-                "HuggingFace": '',
-                "OpenRouter": '',
-                "Perplexity API": '',
+                "DeepInfra": "",
+                "Groq": "",
+                "HuggingFace": "",
+                "OpenRouter": "",
+                "Perplexity API": "",
             }[obj["display_name"]]
 
     def __initUi(self):
diff --git a/pyqt_openai/settings_dialog/voiceSettingsWidget.py b/pyqt_openai/settings_dialog/voiceSettingsWidget.py
@@ -13,7 +13,8 @@
 from pyqt_openai import (
     WHISPER_TTS_VOICE_TYPE,
     WHISPER_TTS_VOICE_SPEED_RANGE,
-    EDGE_TTS_VOICE_TYPE, DEFAULT_HIGHLIGHT_TEXT_COLOR,
+    EDGE_TTS_VOICE_TYPE,
+    DEFAULT_HIGHLIGHT_TEXT_COLOR,
 )
 from pyqt_openai.config_loader import CONFIG_MANAGER
 from pyqt_openai.lang.translations import LangClass
diff --git a/pyqt_openai/shortcutDialog.py b/pyqt_openai/shortcutDialog.py
@@ -15,7 +15,9 @@
     DEFAULT_SHORTCUT_SETTING,
     DEFAULT_SHORTCUT_SEND,
     DEFAULT_SHORTCUT_SHOW_SECONDARY_TOOLBAR,
-    DEFAULT_SHORTCUT_FOCUS_MODE, DEFAULT_SWITCH_PROMPT_UP, DEFAULT_SWITCH_PROMPT_DOWN,
+    DEFAULT_SHORTCUT_FOCUS_MODE,
+    DEFAULT_SWITCH_PROMPT_UP,
+    DEFAULT_SWITCH_PROMPT_DOWN,
 )
 from pyqt_openai.lang.translations import LangClass
 
diff --git a/pyqt_openai/sqlite.py b/pyqt_openai/sqlite.py
@@ -15,7 +15,8 @@
     PROMPT_GROUP_TABLE_NAME,
     PROMPT_ENTRY_TABLE_NAME,
     get_config_directory,
-    DEFAULT_DATETIME_FORMAT, CHAT_FILE_TABLE_NAME,
+    DEFAULT_DATETIME_FORMAT,
+    CHAT_FILE_TABLE_NAME,
 )
 from pyqt_openai.config_loader import CONFIG_MANAGER
 from pyqt_openai.models import (
diff --git a/pyqt_openai/updateSoftwareDialog.py b/pyqt_openai/updateSoftwareDialog.py
@@ -55,7 +55,9 @@ def __initUi(self):
         self.__updateManualLbl = QLabel()
 
         lay.addWidget(self.releaseNoteBrowser)
-        if sys.platform == "win32" and not CONFIG_MANAGER.get_general_property('manual_update'):
+        if sys.platform == "win32" and not CONFIG_MANAGER.get_general_property(
+            "manual_update"
+        ):
             update_url = f"https://github.com/{self.__owner}/{self.__repo}/releases/download/{self.__recent_version}/VividNode.zip"
 
             buttonBox = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
@@ -70,10 +72,12 @@ def __initUi(self):
             self.__updateManualLbl.setText(
                 f'<b>{LangClass.TRANSLATIONS["Update Available"]}</b>'
                 + f"""<br> 
-            For manual updates, please click the link for the latest version and install the file appropriate for your operating system.<br>
+            For manual updates, please click the link for the latest version and install the file appropriate for your operating system.<br><br>
             Windows - Install via exe or zip<br>
             Linux - Install via tar.gz<br>
             macOS - Install via dmg<br>
+            <br>
+            If you want to enable automatic updates, please go to the settings and enable the option.
             """
             )
             self.__updateManualLbl.setWordWrap(True)
diff --git a/pyqt_openai/util/script.py b/pyqt_openai/util/script.py
@@ -65,7 +65,7 @@
     OPENAI_CHAT_ENDPOINT,
     STT_MODEL,
     DEFAULT_DATETIME_FORMAT,
-    DEFAULT_TOKEN_CHUNK_SIZE
+    DEFAULT_TOKEN_CHUNK_SIZE,
 )
 from pyqt_openai.config_loader import CONFIG_MANAGER
 from pyqt_openai.globals import (
@@ -797,6 +797,7 @@ def get_g4f_image_models() -> list:
     models = [model["image_model"] for model in image_models]
     return models
 
+
 def get_g4f_image_providers(including_auto=False) -> list:
     """
     Get all the providers that support image generation
@@ -809,15 +810,11 @@ def get_providers():
         The function get from g4f/gui/server/api.py
         """
         return {
-            provider.__name__: (provider.label
-                                if hasattr(provider, "label")
-                                else provider.__name__) +
-                               (" (WebDriver)"
-                                if "webdriver" in provider.get_parameters()
-                                else "") +
-                               (" (Auth)"
-                                if provider.needs_auth
-                                else "")
+            provider.__name__: (
+                provider.label if hasattr(provider, "label") else provider.__name__
+            )
+            + (" (WebDriver)" if "webdriver" in provider.get_parameters() else "")
+            + (" (Auth)" if provider.needs_auth else "")
             for provider in __providers__
             if provider.working
         }
@@ -844,15 +841,31 @@ def get_provider_models(provider: str) -> list[dict]:
         if provider in __map__:
             provider: ProviderType = __map__[provider]
             if issubclass(provider, ProviderModelMixin):
-                return [{"model": model, "default": model == provider.default_model} for model in provider.get_models()]
+                return [
+                    {"model": model, "default": model == provider.default_model}
+                    for model in provider.get_models()
+                ]
             elif provider.supports_gpt_35_turbo or provider.supports_gpt_4:
                 return [
-                    *([{"model": "gpt-4", "default": not provider.supports_gpt_4}] if provider.supports_gpt_4 else []),
-                    *([{"model": "gpt-3.5-turbo",
-                        "default": not provider.supports_gpt_4}] if provider.supports_gpt_35_turbo else [])
+                    *(
+                        [{"model": "gpt-4", "default": not provider.supports_gpt_4}]
+                        if provider.supports_gpt_4
+                        else []
+                    ),
+                    *(
+                        [
+                            {
+                                "model": "gpt-3.5-turbo",
+                                "default": not provider.supports_gpt_4,
+                            }
+                        ]
+                        if provider.supports_gpt_35_turbo
+                        else []
+                    ),
                 ]
             else:
                 return []
+
     return [model["model"] for model in get_provider_models(provider)]
 
 
@@ -1168,7 +1181,7 @@ def run(self):
                             f"--voice={self.input_args['voice']}",
                             f"--text={self.input_args['input']}",
                         ],
-                        creationflags=subprocess.CREATE_NO_WINDOW
+                        creationflags=subprocess.CREATE_NO_WINDOW,
                     ) as process:
                         process.communicate()
                 else:
diff --git a/pyqt_openai/widgets/fileTableDialog.py b/pyqt_openai/widgets/fileTableDialog.py
@@ -7,4 +7,4 @@ def __init__(self, parent=None):
         self.__initUi()
 
     def __initUi(self):
-        pass
+        pass
