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,11 @@ models = {
 # Очередь на генерации
 queue = []
 
-def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed, output_image, params_textbox):
+def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed):
     # Проверка очереди
     if len(queue) >= 150:
-        return "Очередь переполнена. Пожалуйста, подождите.", None, None
+        yield "Очередь переполнена. Пожалуйста, подождите.", None, None
+        return
     
     # Добавляем задачу в очередь
     task_id = random.randint(100000, 999999)
@@ -51,8 +51,8 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         f"Гуидансе скейл: {guidance_scale}"
     )
     
-    # Обновление параметров
-    params_textbox.update(params_text)
+    # Отправляем текст параметров
+    yield None, params_text, seed
     
     for step in range(steps):
         # Имитация генерации изображения
@@ -63,12 +63,13 @@ def generate_image(prompt, model_name, width, height, seed, steps, guidance_scal
         image = Image.fromarray(image_array)
         
         # Обновление изображения
-        output_image.update(image)
+        yield image, None, None
     
     # Удаляем задачу из очереди
     queue.remove(task_id)
     
-    return output_image, params_text, seed
+    # Возвращаем финальное изображение и параметры
+    yield image, params_text, seed
 
 # Создание интерфейса
 with gr.Blocks() as demo:
@@ -108,9 +109,7 @@ with gr.Blocks() as demo:
             seed_slider,
             steps_slider,
             guidance_scale_slider,
-            random_seed_checkbox,
-            output_image,
-            params_textbox
+            random_seed_checkbox
         ],
         outputs=[output_image, params_textbox, seed_slider]
     )
@@ -131,4 +130,4 @@ demo = demo.add_css(hide_footer)
 # Установка таймаута
 demo.queue(api_name="/generate").set_config(max_size=150, api_open=False, max_threads=1, timeout=125)
 
-demo.launch()
+demo.launch()