Robustness change

app.py

@@ -142,12 +142,12 @@ def parse_lset_prompt(lset_prompt):
 def handle_lora_selection_change(preset_name: str, current_prompt: str):
     """
     Appends the selected LoRA's trigger words to the current prompt text
-    and controls the visibility of the weight slider.
+    and controls the visibility of the weight slider. Ensures slider is only
+    visible on success.
     """
-    # If "None" is selected, do nothing to the prompt and hide the slider.
+    # If "None" is selected, hide the slider and return the prompt unchanged.
     if not preset_name or preset_name == "None":
         gr.Info("LoRA cleared.")
-        # Return the prompt unchanged, and hide the slider.
         return gr.update(value=current_prompt), gr.update(visible=False, interactive=False)
 
     try:
@@ -167,24 +167,23 @@ def handle_lora_selection_change(preset_name: str, current_prompt: str):
         except json.JSONDecodeError:
             lset_prompt_raw = lset_content
 
+        # Only if we successfully get trigger words, we update the prompt and show the slider.
         if lset_prompt_raw:
-            # Append the new trigger words to the current prompt.
             trigger_words = parse_lset_prompt(lset_prompt_raw)
             separator = ", " if current_prompt and not current_prompt.endswith((",", " ")) else ""
             new_prompt = f"{current_prompt}{separator}{trigger_words}".strip()
             gr.Info(f"✅ Appended triggers from '{preset_name}'. You can now edit them.")
-            # Return the updated prompt and show the slider.
             return gr.update(value=new_prompt), gr.update(visible=True, interactive=True)
         else:
-            # If the .lset file has no prompt, do nothing.
+            # If the .lset file has no prompt, don't change the prompt and ensure the slider is hidden.
             gr.Info(f"ℹ️ No prompt found in '{preset_name}.lset'. Prompt unchanged.")
-            return gr.update(value=current_prompt), gr.update(visible=True, interactive=True)
+            return gr.update(value=current_prompt), gr.update(visible=False, interactive=False)
 
     except Exception as e:
         print(f"Info: Could not process .lset for '{preset_name}'. Reason: {e}")
         gr.Warning(f"⚠️ Could not load triggers for '{preset_name}'.")
-        # On error, return the prompt unchanged but still show the slider.
-        return gr.update(value=current_prompt), gr.update(visible=True, interactive=True)
+        # On any error, don't change the prompt and ensure the slider is hidden.
+        return gr.update(value=current_prompt), gr.update(visible=False, interactive=False)
 
 
 def _manage_lora_state(pipe, selected_lora: str, lora_weight: float) -> bool:
@@ -503,3 +502,4 @@ if __name__ == "__main__":
     app_ui = build_ui(t2v_pipe, enhancer_pipe, available_loras)
     app_ui.queue(max_size=10).launch()
 
+