Update app.py

app.py

@@ -33,31 +33,39 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA'), elem_id="custom_image") for m in choices_plus]
 
-def gen_fn(model_str, prompt, negative_prompt, max_retries=10):
+def gen_fn(model_str, prompt, negative_prompt, max_retries=10, progress=gr.Progress()):
     if model_str == 'NA':
         return None
     
     retries = 0
+    feedback = ""
     while retries < max_retries:
         try:
             noise = str(randint(0, 9999999))
+            with progress(tracker=True) as pbar:
+                for step in range(1, 101):
+                    time.sleep(0.05)  # Simulate work being done
+                    pbar(step / 100)
+            
             if hasattr(models_load[model_str], 'negative_prompt'):
                 result = models_load[model_str](f'{prompt} {noise}', negative_prompt=negative_prompt)
             else:
                 result = models_load[model_str](f'{prompt} {noise}')
-            return result
+            feedback += f"{get_current_time()} - Attempt {retries + 1}: Success for {model_str}\n"
+            return result, feedback
         except Exception as e:
             # Check for specific error messages or status codes
             if "CUDA out of memory" in str(e) or "500" in str(e):
-                print(f"CUDA out of memory or server error: {e}")
+                feedback += f"{get_current_time()} - CUDA out of memory or server error for {model_str}: {e}\n"
             else:
-                print(f"Error generating image: {e}")
+                feedback += f"{get_current_time()} - Error generating image for {model_str}: {e}\n"
             
             retries += 1
             if retries >= max_retries:
-                raise Exception(f"Failed to generate image after {max_retries} retries.")
+                feedback += f"{get_current_time()} - Failed to generate image after {max_retries} retries for {model_str}\n"
+                raise Exception(f"Failed to generate image after {max_retries} retries for {model_str}.")
     
-    return None
+    return None, feedback
 
 def make_me():
     with gr.Row():
@@ -79,9 +87,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]
+        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 in zip(current_models, output):
-            gen_event = gen_button.click(gen_fn, [m, txt_input, negative_txt_input], o)
+        for m, o, p, c in zip(current_models, output, progress_bars, 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])
     
     with gr.Accordion('Model selection', elem_id="custom_accordion"):
@@ -230,6 +240,17 @@ body {
     width: 100%;
     box-sizing: border-box;
 }
+.custom_feedback_box {
+    background-color: #2d343f;
+    border: 1px solid #3b4252;
+    color: #7f8184;
+    padding: 10px;
+    border-radius: 4px;
+    margin-top: 10px;
+    width: 100%;
+    box-sizing: border-box;
+    overflow-y: auto;
+}
 @media (max-width: 768px) {
    .gradio-container {
         width: 100%;