Update waifuc_gui/interface.py

waifuc_gui/interface.py

@@ -77,9 +77,9 @@ class Interface:
         updates = []
         for source, source_params in self.params.items():
             for param_name, param_input in source_params.items():
-                updates.append(param_input.update(visible=(source == selected_source)))
+                updates.append(gr.update(visible=(source == selected_source)))
         local_source_info_visible = selected_source == "LocalSource"
-        return updates + [gr.Markdown.update(visible=local_source_info_visible)]
+        return updates + [gr.update(visible=local_source_info_visible)]
 
     def collect_params(self, selected_source, *param_values):
         collected = {}
@@ -102,7 +102,6 @@ class Interface:
     ):
         output_dir = f"/tmp/user_{source_manager.config_manager.session_id}/{dataset_name}"
         try:
-            output_dir = f"/tmp/user_{source_manager.config_manager.session_id}/{dataset_name}"
             logger = logging.getLogger("waifuc_gui")
             logger.info(f"Starting collection for {selected_source} with dataset {dataset_name}")
             os.makedirs(output_dir, exist_ok=True)
@@ -217,30 +216,33 @@ class Interface:
                 new_param_components = [param_input for source_params in new_params.values() for param_input in source_params.values()]
                 action_params = self.action_manager.create_action_param_inputs(action_checkboxes.value or [])
                 new_action_param_inputs = [param_input for action_params_dict in action_params.values() for param_input in action_params_dict.values()]
-                return {
-                    language_dropdown: gr.Dropdown(label=self.get_text("language")),
-                    source_dropdown: gr.Dropdown(label=self.get_text("select_source")),
-                    action_checkboxes: gr.CheckboxGroup(label=self.get_text("select_actions")),
-                    dataset_name_input: gr.Textbox(label=self.get_text("dataset_name")),
-                    exporter_dropdown: gr.Dropdown(label=self.get_text("select_exporter")),
-                    start_btn: gr.Button(self.get_text("start_collection")),
-                    status: gr.Textbox(label=self.get_text("status")),
-                    output_file: gr.File(label=self.get_text("download_data")),
-                    pixiv_token_input: gr.Textbox(label=self.get_text("pixiv_token")),
-                    output_dir_input: gr.Textbox(label=self.get_text("output_dir")),
-                    num_items_input: gr.Number(label=self.get_text("num_items")),
-                    resize_size_input: gr.Number(label=self.get_text("resize_size")),
-                    config_export_btn: gr.Button(self.get_text("export_config")),
-                    config_import_file: gr.File(label=self.get_text("import_config")),
-                    save_config_btn: gr.Button(self.get_text("save_config")),
-                    config_status: gr.Textbox(label=self.get_text("config_status")),
-                    log_output: gr.Textbox(label=self.get_text("view_logs")),
-                    log_download_btn: gr.Button(self.get_text("download_log")),
-                    log_download_file: gr.File(label=self.get_text("download_log")),
-                    local_source_info: gr.Markdown(self.get_text("local_source_info"), visible=source_dropdown.value == "LocalSource"),
-                    **{comp: new_comp for comp, new_comp in zip(param_components, new_param_components)},
-                    **{comp: new_comp for comp, new_comp in zip(action_param_inputs, new_action_param_inputs)}
+                updates = {
+                    language_dropdown: gr.update(label=self.get_text("language")),
+                    source_dropdown: gr.update(label=self.get_text("select_source")),
+                    action_checkboxes: gr.update(label=self.get_text("select_actions")),
+                    dataset_name_input: gr.update(label=self.get_text("dataset_name")),
+                    exporter_dropdown: gr.update(label=self.get_text("select_exporter")),
+                    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")),
+                    output_dir_input: gr.update(label=self.get_text("output_dir")),
+                    num_items_input: gr.update(label=self.get_text("num_items")),
+                    resize_size_input: gr.update(label=self.get_text("resize_size")),
+                    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")),
+                    config_status: gr.update(label=self.get_text("config_status")),
+                    log_output: gr.update(label=self.get_text("view_logs")),
+                    log_download_btn: gr.update(value=self.get_text("download_log")),
+                    log_download_file: gr.update(label=self.get_text("download_log")),
+                    local_source_info: gr.update(value=self.get_text("local_source_info"), visible=source_dropdown.value == "LocalSource"),
                 }
+                for old_comp, new_comp in zip(param_components, new_param_components):
+                    updates[old_comp] = new_comp
+                for old_comp, new_comp in zip(action_param_inputs, new_action_param_inputs):
+                    updates[old_comp] = new_comp
+                return updates
             
             language_dropdown.change(
                 fn=update_language,