Update app_local.py

app_local.py

@@ -15,7 +15,7 @@ import math
 import json  # Added json import
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import logging
-from copy import deepcopy
+import copy
 
 #############################
 os.environ.setdefault('GRADIO_ANALYTICS_ENABLED', 'False')
@@ -46,14 +46,16 @@ rewriter_model = AutoModelForCausalLM.from_pretrained(
     device_map="auto",
     quantization_config=bnb_config,
 )
-
 def get_fresh_presets():
-    """Return a fresh copy of presets to avoid persistence across users"""
-    return deepcopy(PRESETS)
+    return copy.deepcopy(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)
-session_presets = get_fresh_presets()
 # Preload enhancement model at startup
 logger.info("🔄 Loading prompt enhancement model...")
 rewriter_tokenizer = AutoTokenizer.from_pretrained(REWRITER_MODEL)
@@ -300,11 +302,15 @@ def update_prompt_preview(preset_type, base_prompt):
         
 def update_preset_prompt_textbox(preset_type, p1, p2, p3, p4):
     if preset_type and preset_type in preset_state.value:
-        preset_state.value[preset_type]["prompts"] = [p1, p2, p3, p4]
-        # Re‑render preview with updated data
+        # Build new preset instead of mutating in place
+        new_preset = {
+            **preset_state.value[preset_type],
+            "prompts": [p1, p2, p3, p4]
+        }
+        preset_state.value[preset_type] = new_preset
         return update_prompt_preview_with_presets(preset_type, prompt.value, preset_state.value)
     return "Select a preset first."
-
+    
 def update_prompt_preview_with_presets(preset_type, base_prompt, custom_presets):
     if preset_type and preset_type in custom_presets:
         preset = custom_presets[preset_type]
@@ -499,14 +505,7 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
     preset_prompts_state = gr.State(value=[])
     # preset_prompts_state = gr.State(value=["", "", "", ""])
     preset_state = gr.State(value=get_fresh_presets())
-    
-    gr.Markdown("""
-    <div style="text-align: center; background: linear-gradient(to right, #3a7bd5, #00d2ff); color: white; padding: 20px; border-radius: 8px;">
-        <h1 style="margin-bottom: 5px;">⚡️ Qwen-Image-Edit Lightning</h1>
-        <p>✨ 4-step inferencing with lightx2v's LoRA.</p>
-        <p>📝 Local Prompt Enhancement, Batched Multi-image Generation, 🎨 Preset Batches</p>
-    </div>
-    """)
+    gr.Markdown("## ⚡️ Qwen-Image-Edit Lightning Presets")
     
     with gr.Row():
         # Input Column
@@ -540,7 +539,8 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
                 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")
+
 
             
             # Add prompt preview component
@@ -609,14 +609,21 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
                 columns=2,
                 height="60vh",
                 object_fit="contain",
-                preview=True,
                 container=True
             )
             prompt_info = gr.HTML(
                 value="<div style='padding:15px; margin-top:15px'>"
                 "Hint: depending on the original image, prompt quality, and complexity, you can often get away with 3 steps, even 2 steps without much loss in quality. </div>"
             )
-
+        
+        
+    gr.Markdown("""
+    <div style="text-align: center; background: linear-gradient(to right, #3a7bd5, #00d2ff); color: white; padding: 20px; border-radius: 8px;">
+        <p>✨ 4-step inferencing with lightx2v's LoRA.</p>
+        <p>📝 Local Prompt Enhancement, Batched Multi-image Generation, 🎨 Preset Batches</p>
+    </div>
+    """)
+    
     def show_preset_editor(preset_type):
         if preset_type and preset_type in preset_state.value:
             preset = preset_state.value[preset_type]
@@ -718,10 +725,14 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
         inputs=inputs,
         outputs=outputs
     )
+    # .then(
+    #     fn=reset_presets, outputs=preset_state
+    # )
     prompt.submit(
         fn=infer,
         inputs=inputs,
         outputs=outputs
     )
+    reset_button.click(fn=reset_presets, outputs=preset_state)
 
 demo.queue(max_size=5).launch()