Update app.py

app.py

@@ -59,7 +59,36 @@ def upscale_image(image, upscale_factor):
         # В случае ошибки возвращаем сообщение об ошибке
         return f"Error: {response.text}"
 
-# Определение интерфейса с использованием блоков Gradio (код интерфейса остается тем же)
+with gr.Blocks() as demo:
+    gr.Markdown("<style>footer {display: none;}</style>")  # CSS для скрытия футера
+    with gr.Tabs():
+        with gr.TabItem("Базовые настройки"):
+            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Введите текст для генерации изображения")
+        with gr.TabItem("Расширенные настройки"):
+            seed_input = gr.Number(label="Seed", value=0)
+            sampling_method_input = gr.Radio(label="Sampling Method", choices=["random", "VQ-VAE", "DDIM"], value="random")
+            cfg_input = gr.Slider(label="CFG", minimum=0.0, maximum=1.0, value=0.7)
+            width_input = gr.Number(label="Width", value=256)
+            height_input = gr.Number(label="Height", value=256)
+        with gr.TabItem("Настройки апскейлинга"):
+            upscale_factor_input = gr.Slider(label="Увеличить в", minimum=2, maximum=6, value=2, step=1)
 
+    generate_button = gr.Button("Генерация")
+    generated_image_output = gr.Image(label="Сгенерированное изображение")
+    upscale_button = gr.Button("Улучшить качество")
+    upscaled_image_output = gr.Image(label="Изображение с улучшенным качеством")
+
+    # Обработчики кнопок
+    generate_button.click(
+        generate_image,
+        inputs=[prompt_input, seed_input, sampling_method_input, cfg_input, width_input, height_input],
+        outputs=generated_image_output
+    )
+    upscale_button.click(
+        upscale_image,
+        inputs=[generated_image_output, upscale_factor_input],
+        outputs=upscaled_image_output
+    )
+        
 # Запуск интерфейса
 demo.launch()