Update app_local.py

app_local.py

@@ -325,7 +325,7 @@ def infer(
     seed=42,
     randomize_seed=False,
     true_guidance_scale=4.0,
-    num_inference_steps=4,
+    num_inference_steps=3,
     rewrite_prompt=True,
     num_images_per_prompt=1,
     preset_type=None,
@@ -391,7 +391,7 @@ def infer(
             batch_prompts = [f"{original_prompt}, {preset_prompt}" for preset_prompt in non_empty_preset_prompts]
             num_images_per_prompt = len(non_empty_preset_prompts)  # Use actual count of non-empty prompts
             prompt_info = (
-                f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #2196F3; background: #f0f8ff'>"
+                f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #2196F3;>"
                 f"<h4 style='margin-top: 0;'>🎨 Preset: {preset_type}</h4>"
                 f"<p>{preset['description']}</p>"
                 f"<p><strong>Base Prompt:</strong> {original_prompt}</p>"
@@ -403,7 +403,7 @@ def infer(
             # Fallback to manual if no valid prompts
             batch_prompts = [prompt]
             prompt_info = (
-                f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #FF9800; background: #fff8f0'>"
+                f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #FF9800;>"
                 f"<h4 style='margin-top: 0;'>⚠️ Invalid Preset</h4>"
                 f"<p>No valid prompts found. Using manual prompt.</p>"
                 f"<p><strong>Prompt:</strong> {original_prompt}</p>"
@@ -418,7 +418,7 @@ def infer(
                 enhanced_instruction = polish_prompt(original_prompt)
                 if enhanced_instruction and enhanced_instruction != original_prompt:
                     prompt_info = (
-                        f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #4CAF50; background: #f5f9fe'>"
+                        f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #4CAF50;>"
                         f"<h4 style='margin-top: 0;'>🚀 Prompt Enhancement</h4>"
                         f"<p><strong>Original:</strong> {original_prompt}</p>"
                         f"<p><strong style='color:#2E7D32;'>Enhanced:</strong> {enhanced_instruction}</p>"
@@ -427,23 +427,23 @@ def infer(
                     batch_prompts = [enhanced_instruction]
                 else:
                     prompt_info = (
-                        f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #FF9800; background: #fff8f0'>"
+                        f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #FF9800;>"
                         f"<h4 style='margin-top: 0;'>📝 Prompt Enhancement</h4>"
                         f"<p>No enhancement applied or enhancement failed</p>"
                         f"</div>"
                     )
             except Exception as e:
-                print(f"Prompt enhancement error: {str(e)}")  # Debug logging
+                logger.warning(f"Prompt enhancement error: {str(e)}")  # Debug logging
                 gr.Warning(f"Prompt enhancement failed: {str(e)}")
                 prompt_info = (
-                    f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #FF5252; background: #fef5f5'>"
+                    f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #FF5252;>"
                     f"<h4 style='margin-top: 0;'>⚠️ Enhancement Not Applied</h4>"
                     f"<p>Using original prompt. Error: {str(e)[:100]}</p>"
                     f"</div>"
                 )
         else:
             prompt_info = (
-                f"<div style='margin:10px; padding:10px; border-radius:8px; background: #f8f9fa'>"
+                f"<div style='margin:10px; padding:10px; border-radius:8px;>"
                 f"<h4 style='margin-top: 0;'>📝 Original Prompt</h4>"
                 f"<p>{original_prompt}</p>"
                 f"</div>"
@@ -489,7 +489,7 @@ def infer(
             gc.collect()
         gr.Error(f"Image generation failed: {str(e)}")
         return [], base_seed, (
-            f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #dd2c00; background: #fef5f5'>"
+            f"<div style='margin:10px; padding:15px; border-radius:8px; border-left:4px solid #dd2c00;>"
             f"<h4 style='margin-top: 0;'>⚠️ Processing Error</h4>"
             f"<p>{str(e)[:200]}</p>"
             f"</div>"
@@ -607,9 +607,10 @@ with gr.Blocks(title="'Qwen Image Edit' Model Playground & Showcase [4-Step Ligh
             result = gr.Gallery(
                 label="Edited Images",
                 columns=lambda x: min(x, 2),
-                height=500,
-                object_fit="cover",
-                preview=True
+                height="80vh",
+                object_fit="contain",
+                preview=True,
+                container=True
             )
             prompt_info = gr.HTML(
                 value="<div style='padding:15px; margin-top:15px'>"