add more

app.py

@@ -75,7 +75,7 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str,
     client_ip = request.client.host
     if client_ip != '127.0.0.1' and client_ip != 'localhost' and client_ip != '0.0.0.0':
         if time_client_map.get(client_ip):
-            return "Too many requests from this IP, please try again later."
+            return None, "Too many requests, please try again later."
         else:
             time_client_map.put(client_ip, 1, 10)  # 添加一个过期时间为 10 秒的项
 
@@ -85,7 +85,7 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str,
 
     count += 1
     if count > 5:
-        return "Too many requests from this IP, please try again later."
+        return None, "Too many requests, please try again later."
     else:
         count_client_map.put(client_ip, count, 24 * 60 * 60)  # 添加一个过期时间为 24 小时的项
 
@@ -165,7 +165,7 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str,
 
     images.append(output[0])
 
-    return images
+    return images, ""
 
 
 css = """
@@ -467,7 +467,7 @@ with block:
                         #    samples = gr.Slider(label="Images", minimum=1, maximum=4, value=4, step=1)
                         #    steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=45, step=1)
                         guidance_scale = gr.Slider(
-                            label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1
+                            label="Guidance Scale", minimum=0, maximum=40, value=9, step=0.1
                         )
 
                     with gr.Row(elem_id="generate-container", elem_classes="generate-container").style(height="100"):
@@ -481,6 +481,8 @@ with block:
                     label="Generated images", show_label=False, elem_id="gallery"
                 ).style()
 
+                result = gr.Textbox(label="Run Status")
+
         # with gr.Group(elem_id="container-advanced-btns"):
         #     # advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
         #     with gr.Group(elem_id="share-btn-container"):
@@ -490,16 +492,16 @@ with block:
 
         ex = gr.Examples(examples=examples, fn=infer,
                          inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
-                         outputs=[gallery],
+                         outputs=[gallery, result],
                          examples_per_page=5,
                          cache_examples=False)
         ex.dataset.headers = [""]
         negative.submit(infer, inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
-                        outputs=[gallery], postprocess=False)
+                        outputs=[gallery, result], postprocess=False)
         text.submit(infer, inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
-                    outputs=[gallery], postprocess=False)
+                    outputs=[gallery, result], postprocess=False)
         btn.click(infer, inputs=[text, negative, width, height, sampler, steps, seed, guidance_scale],
-                  outputs=[gallery], postprocess=False)
+                  outputs=[gallery, result], postprocess=False)
 
 block.queue(concurrency_count=5,
             max_size=100).launch(