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,7 +28,7 @@ models = {
 # Очередь на генерации
 queue = []
 
-def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed):
+def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, gallery, params_textbox):
     # Проверка очереди
     if len(queue) >= 150:
         return "Очередь переполнена. Пожалуйста, подождите.", None, None
@@ -40,16 +41,6 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
     if random_seed:
         seed = random.randint(0, 1000000)
     
-    # Имитация генерации изображения
-    time.sleep(random.uniform(0.1, 1.0))  # Таймаут до 1 секунды
-    
-    # Удаляем задачу из очереди
-    queue.remove(task_id)
-    
-    # Имитация изображения (шум)
-    image_array = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
-    image = Image.fromarray(image_array)
-    
     # Формирование текста параметров
     params_text = (
         f"Промпт: {prompt}\n"
@@ -60,7 +51,26 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         f"Гуидансе скейл: {guidance_scale}"
     )
     
-    return image, params_text, seed
+    # Обновление параметров
+    params_textbox.update(params_text)
+    
+    images = []
+    for step in range(steps):
+        # Имитация генерации изображения
+        time.sleep(random.uniform(0.1, 0.5))  # Таймаут до 0.5 секунды
+        
+        # Имитация изображения (шум)
+        image_array = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
+        image = Image.fromarray(image_array)
+        
+        # Добавление изображения в галерею
+        images.append(image)
+        gallery.update(images)
+    
+    # Удаляем задачу из очереди
+    queue.remove(task_id)
+    
+    return images, params_text, seed
 
 # Создание интерфейса
 with gr.Blocks() as demo:
@@ -87,7 +97,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("Сгенерировать изображение")
-    output_image = gr.Image(label="Сгенерированное изображение")
+    gallery = gr.Gallery(label="Сгенерированные изображения")
     params_textbox = gr.Textbox(label="Параметры генерации", interactive=False)
     
     generate_button.click(
@@ -100,9 +110,11 @@ with gr.Blocks() as demo:
             seed_slider,
             steps_slider,
             guidance_scale_slider,
-            random_seed_checkbox
+            random_seed_checkbox,
+            gallery,
+            params_textbox
         ],
-        outputs=[output_image, params_textbox, seed_slider]
+        outputs=[gallery, params_textbox, seed_slider]
     )
     
     demo.queue(max_size=150, api_open=False)