Update app.py

app.py

@@ -187,22 +187,6 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)  # Увеличенная высота
                     
-        # Сначала переключаем вкладку, потом запускаем генерацию сообщений
-        btn_to_prompts.click(
-            fn=change_tab, 
-            inputs=[gr.Number(value=2, visible=False)],  # Переключение на вкладку "Сообщения" (id=2)
-            outputs=tabs  # Обновляем вкладку
-        ).then(
-            fn=generate_all_messages, 
-            inputs=[desc, benefits, key_message],  # Входные текстовые поля
-            outputs=[
-                non_personalized_prompt, personalized_prompt,  # Поля для задания копирайтера и редактора (на вкладке "Ассистент")
-                non_personalized_1, personalized_1,  # Сообщения на вкладке "Сообщения"
-                non_personalized_2, personalized_2,
-                non_personalized_3, personalized_3
-            ]
-        )
-   
         # Вкладка 3: Сообщения
         with gr.TabItem("Сообщения", id=2):
             with gr.Row():
@@ -221,8 +205,26 @@ with gr.Blocks() as demo:
                 non_personalized_3 = gr.Textbox(label="Стандартное сообщение 3", lines=4, interactive=False)
                 personalized_3 = gr.Textbox(label="Персонализированное сообщение 3", lines=4, interactive=False)
 
-            btn_generate_messages = gr.Button("Генерировать сообщения")
-            btn_generate_messages.click(fn=generate_all_messages, inputs=[non_personalized_prompt, personalized_prompt], outputs=[non_personalized_1, personalized_1, non_personalized_2, personalized_2, non_personalized_3, personalized_3])
+            # Четвертый ряд
+            with gr.Row():
+                btn_check = gr.Button("Проверить", elem_id="check3")
+                btn_check.click(fn=change_tab, inputs=[gr.Number(value=3, visible=False)], outputs=tabs)
+
+        # Сначала переключаем вкладку, потом запускаем генерацию сообщений
+        btn_to_prompts.click(
+            fn=change_tab, 
+            inputs=[gr.Number(value=2, visible=False)],  # Переключение на вкладку "Сообщения" (id=2)
+            outputs=tabs  # Обновляем вкладку
+        ).then(
+            fn=generate_all_messages, 
+            inputs=[desc, benefits, key_message],  # Входные текстовые поля
+            outputs=[
+                non_personalized_prompt, personalized_prompt,  # Поля для задания копирайтера и редактора (на вкладке "Ассистент")
+                non_personalized_1, personalized_1,  # Сообщения на вкладке "Сообщения"
+                non_personalized_2, personalized_2,
+                non_personalized_3, personalized_3
+            ]
+        )
 
         # Вкладка 4: Проверка
         with gr.TabItem("Проверка", id=3):