Update app_local.py

app_local.py

@@ -173,10 +173,10 @@ def polish_prompt(original_prompt: str) -> str:
     with torch.no_grad():
         generated_ids = rewriter_model.generate(
             **model_inputs,
-            max_new_tokens=256,
+            max_new_tokens=512,
             do_sample=True,
-            temperature=0.7,
-            top_p=0.8,
+            temperature=0.75,
+            top_p=0.85,
             repetition_penalty=1.1,
             no_repeat_ngram_size=3,
             pad_token_id=rewriter_tokenizer.eos_token_id
@@ -486,7 +486,7 @@ def infer(
                 num_images_per_prompt=1
             ).images
             edited_images.extend(result)
-            print(f"Generated image {i+1}/{len(batch_prompts)} with prompt: {current_prompt[:75]}...")
+            print(f"Generated image {i+1}/{len(batch_prompts)} with prompt: {current_prompt}...")
         # Clear cache after generation
         # if device == "cuda":
         #     torch.cuda.empty_cache()
@@ -505,7 +505,7 @@ def infer(
             f"</div>"
         )
 
-with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
+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=["", "", "", ""])
 
@@ -524,7 +524,7 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                 label="Source Image",
                 type="pil",
                 height=300
-            )            
+            )
 
             prompt = gr.Textbox(
                 label="Edit Instructions / Base Prompt",
@@ -550,11 +550,7 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                 preset_prompt_4 = gr.Textbox(label="Prompt 4", lines=1, value="")
                 update_preset_button = gr.Button("Update Preset", variant="secondary")
             
-            rewrite_toggle = gr.Checkbox(
-                label="Enable Prompt Enhancement",
-                value=True,
-                interactive=True
-            )
+
             
             # Add prompt preview component
             prompt_preview = gr.Textbox(
@@ -565,6 +561,14 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                 value="Enter a base prompt and select a preset above to see how your prompt will be modified for batch generation.",
                 placeholder="Prompt preview will appear here..."
             )
+            
+            rewrite_toggle = gr.Checkbox(
+                label="Additional Prompt Enhancement",
+                info="Setting this to true will pass the basic prompt(s) generated via the static preset template to a secondary LLM tasked with improving the overall cohesiveness and details of the final generation prompt.",
+                value=True,
+                interactive=True
+            )
+            
             run_button = gr.Button(
                 "Generate Edit(s)",
                 variant="primary"
@@ -588,14 +592,14 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
                         minimum=1.0,
                         maximum=10.0,
                         step=0.1,
-                        value=1.0
+                        value=1.1
                     )
                     num_inference_steps = gr.Slider(
                         label="Inference Steps",
-                        minimum=2,
+                        minimum=1,
                         maximum=16,
                         step=1,
-                        value=4
+                        value=3
                     )
                     
                 num_images_per_prompt = gr.Slider(
@@ -618,7 +622,7 @@ with gr.Blocks(title="Qwen Image Edit - Fast Lightning Mode w/ Batch") as demo:
             )
             prompt_info = gr.HTML(
                 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>"
+                "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>"
             )
 
     # Fix the show_preset_editor function to use ORIGINAL_PRESETS: