improve latex plugin

crazy_functions/Latex_Function_Wrap.py

@@ -0,0 +1,34 @@
+
+from crazy_functions.Latex_Function import Latex翻译中文并重新编译PDF
+from crazy_functions.plugin_template.plugin_class_template import GptAcademicPluginTemplate, ArgProperty
+
+
+class Arxiv_Localize(GptAcademicPluginTemplate):
+    def __init__(self):
+        pass
+
+    def define_arg_selection_menu(self):
+        """
+        定义插件的二级选项菜单
+        """
+        gui_definition = {
+            "main_input":
+                ArgProperty(title="ArxivID", description="输入Arxiv的ID或者网址", default_value="", type="string").model_dump_json(), # 主输入，自动从输入框同步
+            "advanced_arg":
+                ArgProperty(title="额外的翻译提示词",
+                            description=r"如果有必要, 请在此处给出自定义翻译命令, 解决部分词汇翻译不准确的问题。 "
+                                        r"例如当单词'agent'翻译不准确时, 请尝试把以下指令复制到高级参数区: "
+                                        r'If the term "agent" is used in this section, it should be translated to "智能体". ',
+                            default_value="", type="string").model_dump_json(), # 高级参数输入区，自动同步
+            "allow_cache":
+                ArgProperty(title="是否允许从缓存中调取结果", options=["允许缓存", "从头执行"], default_value="允许缓存", description="无", type="dropdown").model_dump_json(),
+        }
+        return gui_definition
+
+    def execute(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_request):
+
+        allow_cache = plugin_kwargs["allow_cache"]
+        advanced_arg = plugin_kwargs["advanced_arg"]
+
+        if allow_cache == "从头执行": plugin_kwargs["advanced_arg"] = "--no-cache " + plugin_kwargs["advanced_arg"]
+        yield from Latex翻译中文并重新编译PDF(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_request)

crazy_functions/PDF_Translate_Wrap.py

@@ -7,6 +7,9 @@ class PDF_Tran(GptAcademicPluginTemplate):
         pass
 
     def define_arg_selection_menu(self):
+        """
+        定义插件的二级选项菜单
+        """
         gui_definition = {
             "main_input":
                 ArgProperty(title="PDF文件路径", description="请上传文件后，再点击该插件", default_value="", type="string").model_dump_json(), # 主输入，自动从输入框同步

crazy_functions/plugin_template/plugin_class_template.py

@@ -7,7 +7,7 @@ class ArgProperty(BaseModel): # PLUGIN_ARG_MENU
     title: str = Field(description="The title", default="")
     description: str = Field(description="The description", default="")
     default_value: str|float = Field(description="The default value", default="")
-    type: str = Field(description="The type", default="")
+    type: str = Field(description="The type", default="")   # currently we support ['string', 'dropdown']
     options: List[str] = Field(default=[], description="List of options available for the argument")
 
 class GptAcademicPluginTemplate():