Update app_local.py

app_local.py

@@ -258,6 +258,28 @@ def update_prompt_preview(preset_type, base_prompt):
         return preview_text
     else:
         return "Select a preset above to see how your base prompt will be modified for batch generation."
+        
+def toggle_output_count(preset_type):
+    """Control output count slider interactivity and show/hide preset editor"""
+    if preset_type and preset_type in PRESETS:
+        # When preset is selected, disable manual output count and show editor
+        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), 
+            gr.Slider(interactive=False, value=preset["count"]),  # Disable slider
+            prompts[0], prompts[1], prompts[2], prompts[3]  # Populate preset prompts
+        )
+    else:
+        # When no preset is selected, enable manual output count and hide editor
+        return (
+            gr.Group(visible=False), 
+            gr.Slider(interactive=True),  # Enable slider
+            "", "", "", ""  # Clear preset prompts
+        )
 
         
 @spaces.GPU()
@@ -464,10 +486,10 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
             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)
+                preset_prompt_1 = gr.Textbox(label="Prompt 1", lines=1, value="")
+                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")
             
             rewrite_toggle = gr.Checkbox(
@@ -517,14 +539,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,
-                    interactive=True  ### NEW ### - Default to interactive
+                    value=2,
+                    interactive=True
                 )
                 
         # Output Column
@@ -602,14 +624,15 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
             return update_prompt_preview(preset_type, "your subject")  # Use placeholder for preview
         return "Select a preset first to edit its prompts."
     
-        
+
+    # Set up preset dropdown change handler - now controls output count slider
     preset_dropdown.change(
-        fn=toggle_output_count,  ### CHANGED ### - Using new function
+        fn=toggle_output_count,
         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
+    # Set up prompt preview updates
     preset_dropdown.change(
         fn=update_prompt_preview,
         inputs=[preset_dropdown, prompt],
@@ -654,5 +677,4 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
         outputs=outputs
     )
 
-
 demo.queue(max_size=5).launch()