Update app.py

app.py

@@ -108,10 +108,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
  pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
  else:
  pipe.load_lora_weights(lora_path)
- 
- #with calculateDuration(f"Fusing LoRA weights for {selected_lora['title']}"):
- #pipe.fuse_lora(lora_scale)
- 
+
  # Set random seed for reproducibility
  with calculateDuration("Randomizing seed"):
  if randomize_seed:
@@ -121,15 +118,16 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
  
  # Consume the generator to get the final image
  final_image = None
+ step_counter = 0
  for image in image_generator:
+ step_counter+=1
  final_image = image
- yield image, seed # Yield intermediate images and seed
-
- with calculateDuration("Unload"):
- #pipe.unfuse_lora()
+ progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
+ yield image, seed, gr.update(value=progress_bar, visible=True)
+ 
+ yield final_image, seed, gr.update(value=progress_bar, visible=False)
+ with calculateDuration("Unloading LoRA"):
  pipe.unload_lora_weights()
- 
- return final_image, seed # Return the final image and seed
 
 def get_huggingface_safetensors(link):
  split_link = link.split("/")
@@ -218,6 +216,10 @@ css = '''
 .card_internal{display: flex;height: 100px;margin-top: .5em}
 .card_internal img{margin-right: 1em}
 .styler{--form-gap-width: 0px !important}
+#progress{height:30px}
+#progress .generating{display:none}
+.progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px}
+.progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
 '''
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
  title = gr.HTML(
@@ -246,6 +248,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
  custom_lora_info = gr.HTML(visible=False)
  custom_lora_button = gr.Button("Remove custom LoRA", visible=False)
  with gr.Column(scale=4):
+ progress_bar = gr.Markdown(elem_id="progress",visible=False)
  result = gr.Image(label="Generated Image")
 
  with gr.Row():
@@ -282,8 +285,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
  triggers=[generate_button.click, prompt.submit],
  fn=run_lora,
  inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
- outputs=[result, seed]
+ outputs=[result, seed, progress_bar]
  )
 
 app.queue()
-app.launch(show_error=True)
+app.launch()