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()