Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from random import randint
 from all_models import models
 from datetime import datetime
+import time
 
 def get_current_time():
     now = datetime.now()
@@ -35,7 +36,7 @@ def update_imgbox(choices):
 
 def gen_fn(model_str, prompt, negative_prompt, max_retries=10, progress=gr.Progress()):
     if model_str == 'NA':
-        return None
+        return None, ""
     
     retries = 0
     feedback = ""
@@ -87,10 +88,11 @@ def make_me():
     
     with gr.Row():
         output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models]
-        progress_bars = [gr.Progress(label=f"Progress for {m}", visible=(m != 'NA')) for m in default_models]
+        progress_bars = [gr.Progress(visible=(m != 'NA')) for m in default_models]
+        progress_labels = [gr.Label(value=f"Progress for {m}", visible=(m != 'NA')) for m in default_models]
         feedback_box = gr.Textbox(label="Feedback", lines=10, interactive=False, elem_id="custom_feedback_box")
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
-        for m, o, p, c in zip(current_models, output, progress_bars, default_models):
+        for m, o, p, l, c in zip(current_models, output, progress_bars, progress_labels, default_models):
             gen_event = gen_button.click(gen_fn, [m, txt_input, negative_txt_input, gr.Slider(1, 10, value=10, label="Max Retries"), p], [o, feedback_box], cancels=[gen_event])
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])