Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import PIL.Image as Image
 import random
 import time
+import threading
 
 # Список моделей с отображаемыми названиями и внутренними именами
 models = {
@@ -27,10 +28,10 @@ models = {
 # Очередь на генерации
 queue = []
 
-def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, gallery, params_text):
+def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, output_image, params_textbox):
     # Проверка очереди
     if len(queue) >= 150:
-        return "Очередь переполнена. Пожалуйста, подождите.", gallery, params_text, seed
+        return "Очередь переполнена. Пожалуйста, подождите.", None, None
     
     # Добавляем задачу в очередь
     task_id = random.randint(100000, 999999)
@@ -50,7 +51,9 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         f"Гуидансе скейл: {guidance_scale}"
     )
     
-    images = []
+    # Обновление параметров
+    params_textbox.update(params_text)
+    
     for step in range(steps):
         # Имитация генерации изображения
         time.sleep(random.uniform(0.1, 0.5))  # Таймаут до 0.5 секунды
@@ -59,14 +62,13 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         image_array = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
         image = Image.fromarray(image_array)
         
-        # Добавление изображения в галерею
-        images.append(image)
-        gallery = images
+        # Обновление изображения
+        output_image.update(image)
     
     # Удаляем задачу из очереди
     queue.remove(task_id)
     
-    return gallery, params_text, seed
+    return output_image, params_text, seed
 
 # Создание интерфейса
 with gr.Blocks() as demo:
@@ -93,7 +95,7 @@ with gr.Blocks() as demo:
         guidance_scale_slider = gr.Slider(0.1, 20.0, value=7.5, step=0.1, label="Гуидансе скейл")
     
     generate_button = gr.Button("Сгенерировать изображение")
-    gallery = gr.Gallery(label="Сгенерированные изображения")
+    output_image = gr.Image(label="Сгенерированное изображение")
     params_textbox = gr.Textbox(label="Параметры генерации", interactive=False)
     
     generate_button.click(
@@ -107,10 +109,10 @@ with gr.Blocks() as demo:
             steps_slider,
             guidance_scale_slider,
             random_seed_checkbox,
-            gallery,
+            output_image,
             params_textbox
         ],
-        outputs=[gallery, params_textbox, seed_slider]
+        outputs=[output_image, params_textbox, seed_slider]
     )
     
     demo.queue(max_size=150, api_open=False)