Update app_local.py

app_local.py

@@ -46,16 +46,18 @@ rewriter_model = AutoModelForCausalLM.from_pretrained(
     device_map="auto",
     quantization_config=bnb_config,
 )
+
+# Store original presets for reference
+ORIGINAL_PRESETS = deepcopy(PRESETS)
+
 def get_fresh_presets():
-    return copy.deepcopy(PRESETS)
+    return ORIGINAL_PRESETS
 
 preset_state = gr.State(value=get_fresh_presets())
 
 def reset_presets():
     return get_fresh_presets()
     
-# Store original presets for reference
-ORIGINAL_PRESETS = deepcopy(PRESETS)
 # Preload enhancement model at startup
 logger.info("🔄 Loading prompt enhancement model...")
 rewriter_tokenizer = AutoTokenizer.from_pretrained(REWRITER_MODEL)
@@ -504,7 +506,7 @@ def infer(
 with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Lightning Mode]") as demo:
     preset_prompts_state = gr.State(value=[])
     # preset_prompts_state = gr.State(value=["", "", "", ""])
-    preset_state = gr.State(value=get_fresh_presets())
+    preset_state = gr.State(value=ORIGINAL_PRESETS)
     gr.Markdown("## ⚡️ Qwen-Image-Edit Lightning Presets")
     
     with gr.Row():
@@ -538,8 +540,9 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
                 preset_prompt_2 = gr.Textbox(label="Prompt 2", lines=1, value="")
                 preset_prompt_3 = gr.Textbox(label="Prompt 3", lines=1, value="")
                 preset_prompt_4 = gr.Textbox(label="Prompt 4", lines=1, value="")
-                update_preset_button = gr.Button("Update Preset", variant="secondary")
-                reset_button = gr.Button("Reset Presets", variant="stop")
+                
+                update_preset_button = gr.Button("Update Preset", variant="secondary", visible=False)
+                reset_button = gr.Button("Reset Presets", variant="stop", visible=False)