Update app.py

app.py

@@ -876,8 +876,9 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                 uploads = gr.Files(label="Subir datasets locales (csv/json/jsonl/txt)", file_count="multiple", type="binary")
     
     with gr.Accordion("⚙️ Parámetros Avanzados", open=False):
-        with gr.Tabs():
-            with gr.TabItem("Arquitectura (Desde Cero)"):
+        with gr.Tabs() as tabs_advanced:
+            
+            with gr.TabItem("Arquitectura (Desde Cero)", id="tab_scratch"):
                 scratch_group = gr.Group(visible=False)
                 with scratch_group:
                     auto_config_scratch = gr.Checkbox(
@@ -917,7 +918,7 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                     with gr.Row():
                         scratch_tie_word_embeddings = gr.Checkbox(label="tie_word_embeddings", value=False)
             
-            with gr.TabItem("Opciones Generales (Trainer)"):
+            with gr.TabItem("Opciones Generales (Trainer)", id="tab_general"):
                 general_group = gr.Group(visible=True)
                 with general_group:
                     with gr.Row():
@@ -941,7 +942,7 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                         quantization = gr.Textbox(label="quantization", value="int4", placeholder="none, int8 o int4", info="Ignorado si 'Entrenar desde Cero' está activo.")
                         scheduler = gr.Textbox(label="scheduler", value="cosine", placeholder="linear, cosine o constant")
 
-            with gr.TabItem("Hiperparámetros Numéricos"):
+            with gr.TabItem("Hiperparámetros Numéricos", id="tab_numeric"):
                 numeric_group = gr.Group(visible=True)
                 with numeric_group:
                     with gr.Row():
@@ -960,7 +961,7 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                         warmup_ratio = gr.Number(value=0.05, label="warmup_ratio")
                         weight_decay = gr.Number(value=0.01, label="weight_decay")
 
-            with gr.TabItem("PEFT / LoRA"):
+            with gr.TabItem("PEFT / LoRA", id="tab_peft"):
                 peft_group = gr.Group(visible=True)
                 with peft_group:
                     gr.Markdown("#### PEFT / LoRA (Ignorado si 'Entrenar desde Cero' está activo o `peft` está deshabilitado)")
@@ -970,7 +971,7 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                         lora_dropout = gr.Number(value=0.05, label="lora_dropout")
                         target_modules = gr.Textbox(value="q_proj,k_proj,v_proj,o_proj", placeholder="Ej: all-linear o q_proj,v_proj", label="target_modules (Separados por coma)")
             
-            with gr.TabItem("Configuración del Modelo (Carga)"):
+            with gr.TabItem("Configuración del Modelo (Carga)", id="tab_load"):
                 load_config_group = gr.Group(visible=True)
                 with load_config_group:
                     gr.Markdown("#### Configuración del Modelo (Carga - Ignorado si 'Entrenar desde Cero' está activo)")
@@ -980,7 +981,7 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                     with gr.Row():
                         new_special_tokens_input = gr.Textbox(label="Nuevos tokens especiales (coma-separado)", placeholder="<|im_start|>, <|im_end|>, <|system|>", info="Añadir tokens al vocabulario. Redimensiona embeddings.")
             
-            with gr.TabItem("Formato de Datos (Razonamiento)"):
+            with gr.TabItem("Formato de Datos (Razonamiento)", id="tab_cot"):
                 cot_group = gr.Group(visible=True)
                 with cot_group:
                     with gr.Row():
@@ -992,14 +993,14 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
                     with gr.Row():
                         response_col_input = gr.Textbox(label="Columna de Respuesta", value="response", info="Nombre de la columna con la respuesta final.")
 
-            with gr.TabItem("Normalización de Texto (Datos)"):
+            with gr.TabItem("Normalización de Texto (Datos)", id="tab_norm"):
                 normalization_group = gr.Group(visible=True)
                 with normalization_group:
                     with gr.Row():
                         apply_lowercase_input = gr.Checkbox(label="Aplicar minúsculas", value=False, info="Convierte todo el texto a minúsculas antes de tokenizar.")
                         remove_punctuation_input = gr.Checkbox(label="Eliminar puntuación", value=False, info="Elimina signos de puntuación del texto antes de tokenizar.")
             
-            with gr.TabItem("Logging (Weights & Biases)"):
+            with gr.TabItem("Logging (Weights & Biases)", id="tab_wandb"):
                 wandb_group = gr.Group(visible=True)
                 with wandb_group:
                     with gr.Row():
@@ -1010,12 +1011,15 @@ with gr.Blocks(title="HuggingFace SFT Trainer Studio", theme="gradio/soft", css=
     def toggle_scratch_config(scratch_checked):
         scratch_visible = scratch_checked
         finetune_visible = not scratch_checked
-        return {
+        
+        updates = {
             model_base_input: gr.Textbox(visible=finetune_visible),
             scratch_group: gr.Group(visible=scratch_visible),
             peft_group: gr.Group(visible=finetune_visible),
             load_config_group: gr.Group(visible=finetune_visible),
         }
+        
+        return updates
 
     train_from_scratch.change(
         fn=toggle_scratch_config,