DawnC commited on
Commit
64b10ca
·
verified ·
1 Parent(s): bb16ad7

Update ui_manager.py

Browse files
Files changed (1) hide show
  1. ui_manager.py +9 -0
ui_manager.py CHANGED
@@ -754,10 +754,19 @@ class UIManager:
754
  self.inpainting_history.pop(0)
755
 
756
  quality_score = result.get('quality_score', 0)
 
 
 
757
  status = f"Inpainting complete in {elapsed:.1f}s"
758
  if quality_score > 0:
759
  status += f" | Quality: {quality_score:.0f}/100"
760
 
 
 
 
 
 
 
761
  return (
762
  result.get('combined_image'),
763
  result.get('preview_image'),
 
754
  self.inpainting_history.pop(0)
755
 
756
  quality_score = result.get('quality_score', 0)
757
+ metadata = result.get('metadata', {})
758
+ actual_steps = metadata.get('num_inference_steps', num_steps)
759
+
760
  status = f"Inpainting complete in {elapsed:.1f}s"
761
  if quality_score > 0:
762
  status += f" | Quality: {quality_score:.0f}/100"
763
 
764
+ # Notify user if steps were reduced (Spaces first-load scenario)
765
+ if actual_steps < num_steps:
766
+ status += f"\n⚠️ Steps reduced: {num_steps} → {actual_steps} (Spaces optimization)"
767
+ if actual_steps <= 7:
768
+ status += f"\n💡 First request - reduced quality. Click 'Generate' again for better results!"
769
+
770
  return (
771
  result.get('combined_image'),
772
  result.get('preview_image'),