Update app.py

app.py

@@ -93,7 +93,6 @@ def generate_message_gigachat_pro_with_retry(prompt):
             return message
     return message
 
-# Измененная функция для генерации всех сообщений
 # Измененная функция для генерации всех сообщений
 def generate_all_messages(desc, benefits, key_message):
     # Варианты предложений для начала и конца
@@ -115,6 +114,18 @@ def generate_all_messages(desc, benefits, key_message):
     non_personalized_prompt = f"{desc}\n\nПреимущества:\n{benefits}\n\nКлючевое сообщение: {key_message}"
     personalized_prompt = f"Персонализированное сообщение для клиента: {desc}\n\n{benefits}\n\nКлючевое сообщение: {key_message}"
 
+    # Выводим задание для копирайтера
+    yield non_personalized_prompt, None, None, None, None, None, None, None
+    
+    # Небольшая пауза для демонстрации постепенной генерации
+    time.sleep(1)
+    
+    # Выводим задание для редактора
+    yield non_personalized_prompt, personalized_prompt, None, None, None, None, None, None
+    
+    # Небольшая пауза для демонстрации постепенной генерации
+    time.sleep(1)
+    
     # Генерация трех неперсонализированных сообщений
     for i in range(3):
         prompt = add_prefix_suffix(non_personalized_prompt, prefixes[i], suffixes[i])
@@ -127,9 +138,9 @@ def generate_all_messages(desc, benefits, key_message):
         prompt = add_prefix_suffix(full_personalized_prompt, prefixes[i], suffixes[i])
         message = generate_message_gigachat_pro_with_retry(prompt)
         personalized_messages.append(message)
-
-    # Возвращаем значения для полей "Задание для копирайтера" и "Задание для редактора" и сообщения
-    return (
+    
+    # Выводим все сгенерированные сообщения
+    yield (
         non_personalized_prompt, personalized_prompt,  # Поля для задания копирайтера и редактора
         non_personalized_messages[0], personalized_messages[0],  # Первое неперсонализированное и персонализированное сообщение
         non_personalized_messages[1], personalized_messages[1],  # Второе неперсонализированное и персонализированное сообщение
@@ -222,7 +233,7 @@ with gr.Blocks() as demo:
             # Сначала переключаем вкладку, потом запускаем генерацию сообщений
             btn_to_prompts.click(
                 fn=change_tab, 
-                inputs=[gr.Number(value=2, visible=False)],  # Переключение на вкладку "Сообщения" (id=2)
+                inputs=[gr.Number(value=1, visible=False)],  # Переключение на вкладку "Сообщения" (id=1)
                 outputs=tabs  # Обновляем вкладку
             ).then(
                 fn=generate_all_messages,