Update app.py

app.py

@@ -3,7 +3,6 @@ import numpy as np
 import PIL.Image as Image
 import random
 import time
-import threading
 
 # Список моделей с отображаемыми названиями и внутренними именами
 models = {
@@ -28,10 +27,10 @@ models = {
 # Очередь на генерации
 queue = []
 
-def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, gallery, params_textbox):
+def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, gallery, params_text):
     # Проверка очереди
     if len(queue) >= 150:
-        return "Очередь переполнена. Пожалуйста, подождите.", None, None
+        return "Очередь переполнена. Пожалуйста, подождите.", gallery, params_text, seed
     
     # Добавляем задачу в очередь
     task_id = random.randint(100000, 999999)
@@ -51,9 +50,6 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         f"Гуидансе скейл: {guidance_scale}"
     )
     
-    # Обновление параметров
-    params_textbox.update(params_text)
-    
     images = []
     for step in range(steps):
         # Имитация генерации изображения
@@ -65,12 +61,12 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         
         # Добавление изображения в галерею
         images.append(image)
-        gallery.update(images)
+        gallery = images
     
     # Удаляем задачу из очереди
     queue.remove(task_id)
     
-    return images, params_text, seed
+    return gallery, params_text, seed
 
 # Создание интерфейса
 with gr.Blocks() as demo: