Update waifuc_gui/interface.py

waifuc_gui/interface.py

@@ -40,7 +40,10 @@ class Interface:
                 "config_status": "配置状态",
                 "view_logs": "查看日志",
                 "download_log": "下载日志",
-                "local_source_info": "上传包含图片的zip文件用于LocalSource"
+                "local_source_info": "上传包含图片的zip文件用于LocalSource",
+                "pixiv_word": "搜索关键词（Pixiv）",
+                "missing_params": "请填写所有必填参数（如搜索关键词）",
+                "invalid_source": "无效的数据源配置"
             },
             "en": {
                 "title": "Waifuc Data Collection Tool",
@@ -66,7 +69,10 @@ class Interface:
                 "config_status": "Configuration Status",
                 "view_logs": "View Logs",
                 "download_log": "Download Log",
-                "local_source_info": "Upload a ZIP file containing images for LocalSource"
+                "local_source_info": "Upload a ZIP file containing images for LocalSource",
+                "pixiv_word": "Search Keyword (Pixiv)",
+                "missing_params": "Please fill in all required parameters (e.g., search keyword)",
+                "invalid_source": "Invalid data source configuration"
             }
         }
 
@@ -100,6 +106,16 @@ class Interface:
         self, selected_source, params, selected_actions, action_params, dataset_name, selected_exporter,
         source_manager, action_manager, exporter_manager, file_handler
     ):
+        if not selected_source or not dataset_name or not selected_exporter:
+            return (
+                self.get_text("missing_params"),
+                None, ""
+            )
+        if selected_source == "PixivSearchSource" and (not params.get("word") or not params.get("refresh_token")):
+            return (
+                self.get_text("missing_params"),
+                None, ""
+            )
         output_dir = f"/tmp/user_{source_manager.config_manager.session_id}/{dataset_name}"
         try:
             logger = logging.getLogger("waifuc_gui")
@@ -214,17 +230,17 @@ class Interface:
                 self.config_manager.set_config("language", language)
                 updates = {
                     language_dropdown: gr.update(label=self.get_text("language")),
-                    source_dropdown: gr.update(label=self.get_text("select_source"), info="Choose a data source like DanbooruSource or LocalSource" if self.config_manager.get_config("language") == "en" else "选择数据源，如DanbooruSource或LocalSource"),
-                    action_checkboxes: gr.update(label=self.get_text("select_actions"), info="Select actions to process images, e.g., NoMonochromeAction" if self.config_manager.get_config("language") == "en" else "选择处理图片的动作，例如NoMonochromeAction"),
-                    dataset_name_input: gr.update(label=self.get_text("dataset_name"), info="Name for the output dataset" if self.config_manager.get_config("language") == "en" else "输出数据集的名称"),
-                    exporter_dropdown: gr.update(label=self.get_text("select_exporter"), info="Choose an exporter邢器，如TextualInversionExporter" if self.config_manager.get_config("language") == "en" else "选择导出器，如TextualInversionExporter"),
+                    source_dropdown: gr.update(label=self.get_text("select_source"), info=self.get_text("select_source_info")),
+                    action_checkboxes: gr.update(label=self.get_text("select_actions"), info=self.get_text("select_actions_info")),
+                    dataset_name_input: gr.update(label=self.get_text("dataset_name"), info=self.get_text("dataset_name_info")),
+                    exporter_dropdown: gr.update(label=self.get_text("select_exporter"), info=self.get_text("select_exporter_info")),
                     start_btn: gr.update(value=self.get_text("start_collection")),
                     status: gr.update(label=self.get_text("status")),
                     output_file: gr.update(label=self.get_text("download_data")),
-                    pixiv_token_input: gr.update(label=self.get_text("pixiv_token"), info="Required for PixivSearchSource, stored in session only" if self.config_manager.get_config("language") == "en" else "PixivSearchSource所需，仅存储在会话中"),
-                    output_dir_input: gr.update(label=self.get_text("output_dir"), info="Directory for output data, session-specific" if self.config_manager.get_config("language") == "en" else "输出数据的目录，会话专用"),
-                    num_items_input: gr.update(label=self.get_text("num_items"), info="Default number of images to collect" if self.config_manager.get_config("language") == "en" else "默认收集的图片数量"),
-                    resize_size_input: gr.update(label=self.get_text("resize_size"), info="Default size for ResizeAction" if self.config_manager.get_config("language") == "en" else "ResizeAction的默认尺寸"),
+                    pixiv_token_input: gr.update(label=self.get_text("pixiv_token"), info=self.get_text("pixiv_token_info")),
+                    output_dir_input: gr.update(label=self.get_text("output_dir"), info=self.get_text("output_dir_info")),
+                    num_items_input: gr.update(label=self.get_text("num_items"), info=self.get_text("num_items_info")),
+                    resize_size_input: gr.update(label=self.get_text("resize_size"), info=self.get_text("resize_size_info")),
                     config_export_btn: gr.update(value=self.get_text("export_config")),
                     config_import_file: gr.update(label=self.get_text("import_config")),
                     save_config_btn: gr.update(value=self.get_text("save_config")),
@@ -313,8 +329,8 @@ class Interface:
                 outputs=log_download_file
             )
             
-            cleanup_btn = gr.Button("Cleanup Session" if self.config_manager.get_config("language") == "en" else "清理会话")
-            cleanup_status = gr.Textbox(label="Cleanup Status" if self.config_manager.get_config("language") == "en" else "清理状态")
+            cleanup_btn = gr.Button(self.get_text("cleanup_session"))
+            cleanup_status = gr.Textbox(label=self.get_text("cleanup_status"), interactive=False)
             
             cleanup_btn.click(
                 fn=self.cleanup_session,
@@ -339,4 +355,4 @@ class Interface:
             "log_download_file": log_download_file,
             "cleanup_btn": cleanup_btn,
             "cleanup_status": cleanup_status
-        }
+        }