Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
-import time
+import numpy as np
+import PIL.Image as Image
 import random
+import io
 
 # Список моделей с отображаемыми названиями и внутренними именами
 models = {
@@ -25,31 +27,45 @@ models = {
 # Очередь на генерации
 queue = []
 
-def generate_image(model_name, width, height, seed, steps, guidance_scale):
+def generate_image(model_name, width, height, seed, steps, guidance_scale, random_seed):
     # Проверка очереди
     if len(queue) >= 150:
-        return "Очередь переполнена. Пожалуйста, подождите."
+        return "Очередь переполнена. Пожалуйста, подождите.", None, None
     
     # Добавляем задачу в очередь
     task_id = random.randint(100000, 999999)
     queue.append(task_id)
     
+    # Генерация сида, если выбрана галочка "Случайный сид"
+    if random_seed:
+        seed = random.randint(0, 1000000)
+    
     # Имитация генерации изображения
     time.sleep(random.uniform(0.1, 1.0))  # Таймаут до 1 секунды
     
     # Удаляем задачу из очереди
     queue.remove(task_id)
     
-    # Имитация изображения
-    image = f"generated_images/{task_id}.png"
-    return image
+    # Имитация изображения (шум)
+    image_array = np.random.randint(0, 256, (height, width, 3), dtype=np.uint8)
+    image = Image.fromarray(image_array)
+    
+    # Сохранение изображения в буфер
+    buffered = io.BytesIO()
+    image.save(buffered, format="PNG")
+    image_bytes = buffered.getvalue()
+    
+    # Формирование текста параметров
+    params_text = f"Модель: {model_name}\nРазмер: {width}x{height}\nСид: {seed}\nШаги: {steps}\nГуидансе скала: {guidance_scale}"
+    
+    return image_bytes, params_text, seed
 
 # Создание интерфейса
 with gr.Blocks() as demo:
     gr.Markdown("# Генератор изображений")
     
     with gr.Row():
-        model_dropdown = gr.Dropdown(list(models.keys()), label="Модель")
+        model_dropdown = gr.Dropdown(list(models.keys()), label="Модель", value="FLUX.1-dev")
     
     with gr.Row():
         with gr.Column():
@@ -59,6 +75,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         seed_slider = gr.Slider(0, 1000000, value=random.randint(0, 1000000), step=1, label="Сид")
+        random_seed_checkbox = gr.Checkbox(label="Случайный сид")
     
     with gr.Row():
         steps_slider = gr.Slider(1, 100, value=50, step=1, label="Шаги")
@@ -66,12 +83,12 @@ with gr.Blocks() as demo:
     
     generate_button = gr.Button("Сгенерировать изображение")
     output_image = gr.Image(label="Сгенерированное изображение")
+    params_textbox = gr.Textbox(label="Параметры генерации", interactive=False)
     
     generate_button.click(
         fn=generate_image,
-        inputs=[model_dropdown, width_slider, height_slider, seed_slider, steps_slider, guidance_scale_slider],
-        outputs=output_image,
-        api_name="/generate"
+        inputs=[model_dropdown, width_slider, height_slider, seed_slider, steps_slider, guidance_scale_slider, random_seed_checkbox],
+        outputs=[output_image, params_textbox, seed_slider]
     )
     
     demo.queue(max_size=150, api_open=False)
@@ -90,4 +107,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()