Update app.py

app.py

@@ -27,10 +27,10 @@ models = {
 # Очередь на генерации
 queue = []
 
-def generate_image(model_name, width, height, seed, steps, guidance_scale, random_seed):
+def generate_image(prompt, model_name, width, height, seed, steps, guidance_scale, random_seed):
     # Проверка очереди
     if len(queue) >= 150:
-        return "Очередь переполнена. Пожалуйста, подождите.", None, None
+        return "Очередь переполнена. Пожалуйста, подождите.", None, None, None
     
     # Добавляем задачу в очередь
     task_id = random.randint(100000, 999999)
@@ -56,7 +56,14 @@ def generate_image(model_name, width, height, seed, steps, guidance_scale, rando
     image_bytes = buffered.getvalue()
     
     # Формирование текста параметров
-    params_text = f"Модель: {model_name}\nРазмер: {width}x{height}\nСид: {seed}\nШаги: {steps}\nГуидансе скала: {guidance_scale}"
+    params_text = (
+        f"Промпт: {prompt}\n"
+        f"Модель: {model_name}\n"
+        f"Размер: {width}x{height}\n"
+        f"Сид: {seed}\n"
+        f"Шаги: {steps}\n"
+        f"Гуидансе скейл: {guidance_scale}"
+    )
     
     return image_bytes, params_text, seed
 
@@ -64,6 +71,9 @@ def generate_image(model_name, width, height, seed, steps, guidance_scale, rando
 with gr.Blocks() as demo:
     gr.Markdown("# Генератор изображений")
     
+    with gr.Row():
+        prompt_textbox = gr.Textbox(label="Промпт", placeholder="Введите ваш промпт здесь", lines=2)
+    
     with gr.Row():
         model_dropdown = gr.Dropdown(list(models.keys()), label="Модель", value="FLUX.1-dev")
     
@@ -79,7 +89,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         steps_slider = gr.Slider(1, 100, value=50, step=1, label="Шаги")
-        guidance_scale_slider = gr.Slider(0.1, 20.0, value=7.5, step=0.1, label="Гуидансе скала")
+        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="Сгенерированное изображение")
@@ -87,7 +97,16 @@ with gr.Blocks() as demo:
     
     generate_button.click(
         fn=generate_image,
-        inputs=[model_dropdown, width_slider, height_slider, seed_slider, steps_slider, guidance_scale_slider, random_seed_checkbox],
+        inputs=[
+            prompt_textbox,
+            model_dropdown,
+            width_slider,
+            height_slider,
+            seed_slider,
+            steps_slider,
+            guidance_scale_slider,
+            random_seed_checkbox
+        ],
         outputs=[output_image, params_textbox, seed_slider]
     )
     
@@ -107,4 +126,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()