refactor: prompt logic

app_local.py

@@ -6,7 +6,7 @@ import spaces
 from PIL import Image
 from diffusers import QwenImageEditPipeline, FlowMatchEulerDiscreteScheduler
 from diffusers.utils import is_xformers_available
-from presets import PRESETS, get_preset_choices, get_preset_info
+from presets import PRESETS, get_preset_choices, get_preset_info, update_preset_prompt
 
 import os
 import sys
@@ -426,8 +426,9 @@ def infer(
             f"</div>"
         )
 
-
 with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
+    preset_prompts_state = gr.State(value=["", "", "", ""])
+
     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>
@@ -459,6 +460,16 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                 label="Preset Batch Generation",
                 interactive=True
             )
+            # Add editable preset prompts (initially hidden)
+            preset_editor = gr.Group(visible=False)
+            with preset_editor:
+                gr.Markdown("### 🎨 Edit Preset Prompts")
+                preset_prompt_1 = gr.Textbox(label="Prompt 1", lines=1)
+                preset_prompt_2 = gr.Textbox(label="Prompt 2", lines=1)
+                preset_prompt_3 = gr.Textbox(label="Prompt 3", lines=1)
+                preset_prompt_4 = gr.Textbox(label="Prompt 4", lines=1)
+                update_preset_button = gr.Button("Update Preset", variant="secondary")
+            
             rewrite_toggle = gr.Checkbox(
                 label="Enable Prompt Enhancement",
                 value=True,
@@ -506,12 +517,14 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                         step=1,
                         value=4
                     )
+                ### CHANGED ### - Added interactive parameter
                 num_images_per_prompt = gr.Slider(
                     label="Output Count (Manual)",
                     minimum=1,
                     maximum=4,
                     step=1,
-                    value=1
+                    value=1,
+                    interactive=True  ### NEW ### - Default to interactive
                 )
                 
         # Output Column
@@ -527,7 +540,75 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                 value="<div style='padding:15px; margin-top:15px'>"
                 "Prompt details will appear after generation. Ability to edit Preset Prompts on the fly will be implemented shortly.</div>"
             )
+
+    def toggle_output_count(preset_type):
+        if preset_type and preset_type in PRESETS:
+            # When preset is selected, disable manual output count
+            preset = PRESETS[preset_type]
+            return (
+                gr.Group(visible=True), 
+                gr.Slider(interactive=False),  # Disable slider
+                preset_prompt_1.update(value=preset["prompts"][0] if len(preset["prompts"]) > 0 else ""),
+                preset_prompt_2.update(value=preset["prompts"][1] if len(preset["prompts"]) > 1 else ""),
+                preset_prompt_3.update(value=preset["prompts"][2] if len(preset["prompts"]) > 2 else ""),
+                preset_prompt_4.update(value=preset["prompts"][3] if len(preset["prompts"]) > 3 else "")
+            )
+        else:
+            # When no preset is selected, enable manual output count
+            return (
+                gr.Group(visible=False), 
+                gr.Slider(interactive=True),  # Enable slider
+                gr.Textbox(value=""), gr.Textbox(value=""), gr.Textbox(value=""), gr.Textbox(value="")
+            )
+            
+    # Function to show preset editor when a preset is selected
+    def show_preset_editor(preset_type):
+        if preset_type and preset_type in PRESETS:
+            preset = PRESETS[preset_type]
+            prompts = preset["prompts"]
+            # Pad prompts to 4 items if needed
+            while len(prompts) < 4:
+                prompts.append("")
+            return gr.Group(visible=True), prompts[0], prompts[1], prompts[2], prompts[3]
+        return gr.Group(visible=False), "", "", "", ""
+    # Function to update prompt preview when preset is selected
+    def update_prompt_preview(preset_type, base_prompt):
+        """Update the prompt preview display based on selected preset and base prompt"""
+        if preset_type and preset_type in PRESETS:
+            preset = PRESETS[preset_type]
+            preview_text = f"**Preset: {preset_type}**\n\n"
+            preview_text += f"*{preset['description']}*\n\n"
+            preview_text += "**Generated Prompts:**\n"
+            
+            for i, preset_prompt in enumerate(preset["prompts"], 1):
+                full_prompt = f"{base_prompt}, {preset_prompt}"
+                preview_text += f"{i}. {full_prompt}\n"
+            
+            return preview_text
+        else:
+            return "Select a preset above to see how your base prompt will be modified for batch generation."
+    
+    # Function to update individual preset prompts
+    def update_preset_prompt_textbox(preset_type, prompt_1, prompt_2, prompt_3, prompt_4):
+        """Update preset prompts based on user input"""
+        if preset_type and preset_type in PRESETS:
+            # Update each prompt in the preset
+            new_prompts = [prompt_1, prompt_2, prompt_3, prompt_4]
+            for i, new_prompt in enumerate(new_prompts):
+                if new_prompt and i < len(PRESETS[preset_type]["prompts"]):
+                    PRESETS[preset_type]["prompts"][i] = new_prompt
             
+            # Return updated preset info for preview
+            return update_prompt_preview(preset_type, "your subject")  # Use placeholder for preview
+        return "Select a preset first to edit its prompts."
+    
+        
+    preset_dropdown.change(
+        fn=toggle_output_count,  ### CHANGED ### - Using new function
+        inputs=preset_dropdown,
+        outputs=[preset_editor, num_images_per_prompt, preset_prompt_1, preset_prompt_2, preset_prompt_3, preset_prompt_4]
+    )
+    
     # Set up prompt preview update
     preset_dropdown.change(
         fn=update_prompt_preview,
@@ -541,6 +622,13 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
         outputs=prompt_preview
     )
     
+    # Set up preset editor update button
+    update_preset_button.click(
+        fn=update_preset_prompt_textbox,
+        inputs=[preset_dropdown, preset_prompt_1, preset_prompt_2, preset_prompt_3, preset_prompt_4],
+        outputs=prompt_preview
+    )
+    
     # Set up processing
     inputs = [
         input_image,