Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from random import randint
 
-# Пример списка моделей
+# Список моделей (пример)
 models = ["Model A", "Model B", "Model C"]
 
 # Загрузка моделей
@@ -21,11 +21,11 @@ load_fn(models)
 num_models = len(models)
 default_models = models[:num_models]
 
-# Расширение списка выбора моделей
+# Расширение выбора моделей
 def extend_choices(choices):
     return choices + (num_models - len(choices)) * ['NA']
 
-# Обновление списка изображений
+# Обновление изображений
 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]
@@ -41,32 +41,48 @@ def gen_fn(model_str, prompt):
 def make_me():
     with gr.Row():
         with gr.Column(scale=1):
-            txt_input = gr.Textbox(label='Your prompt:', lines=3, elem_id="custom_textbox", placeholder="Prompt")
+            txt_input = gr.Textbox(
+                label='Your prompt:', 
+                lines=3, 
+                elem_id="custom_textbox", 
+                placeholder="Enter your prompt"
+            )
             with gr.Row():
                 gen_button = gr.Button('Generate images', elem_id="custom_gen_button")
                 stop_button = gr.Button('Stop', variant='secondary', interactive=False, elem_id="custom_stop_button")
 
                 # Логика кнопок
-                gen_button.click(lambda: [gen_button, stop_button], None, [gen_button, stop_button])
-                stop_button.click(lambda: [gen_button, stop_button], None, [gen_button, stop_button])
+                def on_generate_click():
+                    return gen_button, stop_button.interactive(True)
+
+                def on_stop_click():
+                    return gen_button, stop_button.interactive(False)
+
+                gen_button.click(on_generate_click, None, [gen_button, stop_button])
+                stop_button.click(on_stop_click, None, [gen_button, stop_button])
 
-    # Вывод изображений
     with gr.Row():
         output = [gr.Image(label=m, elem_id="custom_image", tool="editor") for m in default_models]
         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], o)
-            stop_button.click(lambda: [gen_button, stop_button], None, [gen_button, stop_button], cancels=[gen_event])
+            stop_button.click(on_stop_click, None, [gen_button, stop_button], cancels=[gen_event])
 
-    # Выбор моделей
     with gr.Accordion('Model selection', elem_id="custom_accordion"):
-        model_choice = gr.CheckboxGroup(models, label=f'{num_models} different models selected', 
-                                        value=default_models, elem_id="custom_checkbox_group")
+        model_choice = gr.CheckboxGroup(
+            models, 
+            label=f'{num_models} models selected', 
+            value=default_models, 
+            elem_id="custom_checkbox_group"
+        )
         model_choice.change(update_imgbox, model_choice, output)
         model_choice.change(extend_choices, model_choice, current_models)
 
+    with gr.Row():
+        gr.HTML("")
+
+# Кастомный CSS для интерфейса
 custom_css = """
-/* Основные стили */
 body, html {
     overflow-y: auto;
     height: 100%;
@@ -81,20 +97,7 @@ body, html {
     flex-direction: column;
 }
 
-/* Прокручиваемое изображение */
-.custom_image {
-    max-width: 100%;
-    height: auto;
-    object-fit: contain;
-    overflow: auto;
-    border: 1px solid #3b4252;
-    background-color: #2d343f;
-    border-radius: 4px;
-    cursor: grab;
-    margin: 10px;
-}
-
-/* Стили текста и кнопок */
+/* Стили для текста */
 .custom_textbox {
     background-color: #2d343f;
     border: 1px solid #3b4252;
@@ -105,6 +108,7 @@ body, html {
     box-sizing: border-box;
 }
 
+/* Стили для кнопок */
 .custom_gen_button, .custom_stop_button {
     padding: 15px;
     border-radius: 4px;
@@ -116,6 +120,20 @@ body, html {
     transform: scale(1.05);
 }
 
+/* Стили для изображений */
+.custom_image {
+    max-width: 100%;
+    height: auto;
+    object-fit: contain;
+    overflow: auto;
+    border: 1px solid #3b4252;
+    background-color: #2d343f;
+    border-radius: 4px;
+    cursor: grab;
+    margin: 10px;
+}
+
+/* Стили для выбора моделей */
 .custom_accordion {
     margin-top: 20px;
     border: 1px solid #3b4252;
@@ -124,8 +142,14 @@ body, html {
 """
 
 # Создание интерфейса
-with gr.Blocks(css=custom_css) as demo: 
+with gr.Blocks(css=custom_css) as demo:
     make_me()
+demo.queue(default_concurrency_limit=240, max_size=240)
+demo.launch(max_threads=400, ssr_mode=True)
+# https://github.com/gradio-app/gradio/issues/6339
+
+
 
+# Очередь и запуск интерфейса
 demo.queue()
 demo.launch()