Update app.py

app.py

@@ -126,26 +126,42 @@ def generate_all_messages(desc, benefits, key_message):
     # Небольшая пауза для демонстрации постепенной генерации
     time.sleep(1)
     
-    # Генерация трех неперсонализированных сообщений
+    # Генерация и постепенная подача каждого сообщения
     for i in range(3):
+        # Генерация неперсонализированного сообщения
         prompt = add_prefix_suffix(non_personalized_prompt, prefixes[i], suffixes[i])
-        message = generate_message_gigachat_pro_with_retry(prompt)
-        non_personalized_messages.append(message)
-    
-    # Генерация трех персонализированных сообщений
-    for i in range(3):
-        full_personalized_prompt = f"{personalized_prompt}\n\nТекст для адаптации: {non_personalized_messages[i]}"
-        prompt = add_prefix_suffix(full_personalized_prompt, prefixes[i], suffixes[i])
-        message = generate_message_gigachat_pro_with_retry(prompt)
-        personalized_messages.append(message)
-    
-    # Выводим все сгенерированные сообщения
-    yield (
-        non_personalized_prompt, personalized_prompt,  # Поля для задания копирайтера и редактора
-        non_personalized_messages[0], personalized_messages[0],  # Первое неперсонализированное и персонализированное сообщение
-        non_personalized_messages[1], personalized_messages[1],  # Второе неперсонализированное и персонализированное сообщение
-        non_personalized_messages[2], personalized_messages[2]   # Третье неперсонализированное и персонализированное сообщение
-    )
+        non_personalized_message = generate_message_gigachat_pro_with_retry(prompt)
+        non_personalized_messages.append(non_personalized_message)
+        
+        # Выводим неперсонализированное сообщение
+        yield (
+            non_personalized_prompt, personalized_prompt,  # Задания для копирайтера и редактора
+            non_personalized_messages[0] if i >= 0 else None,  # Первое неперсонализированное сообщение
+            None,  # Первое персонализированное сообщение пока не готово
+            non_personalized_messages[1] if i >= 1 else None,  # Второе неперсонализированное сообщение
+            None,  # Второе персонализированное сообщение пока не готово
+            non_personalized_messages[2] if i >= 2 else None,  # Третье неперсонализированное сообщение
+            None  # Третье персонализированное сообщение пока не готово
+        )
+        
+        # Генерация персонализированного сообщения
+        full_personalized_prompt = f"{personalized_prompt}\n\nТекст для адаптации: {non_personalized_message}"
+        personalized_message = generate_message_gigachat_pro_with_retry(full_personalized_prompt)
+        personalized_messages.append(personalized_message)
+
+        # Выводим персонализированное сообщение
+        yield (
+            non_personalized_prompt, personalized_prompt,  # Задания для копирайтера и редактора
+            non_personalized_messages[0] if i >= 0 else None,  # Первое неперсонализированное сообщение
+            personalized_messages[0] if i >= 0 else None,  # Первое персонализированное сообщение
+            non_personalized_messages[1] if i >= 1 else None,  # Второе неперсонализированное сообщение
+            personalized_messages[1] if i >= 1 else None,  # Второе персонализированное сообщение
+            non_personalized_messages[2] if i >= 2 else None,  # Третье неперсонализированное сообщение
+            personalized_messages[2] if i >= 2 else None  # Третье персонализированное сообщение
+        )
+
+        # Небольшая пауза между выводом каждого сообщения
+        time.sleep(1)
 
 
 # Интерфейс Gradio