Update app.py

app.py

@@ -93,7 +93,7 @@ def generate_message_gigachat_pro_with_retry(prompt):
             return message
     return message
 
-# Измененная функция для генерации всех сообщений
+# Функция для постепенной генерации всех сообщений через yield
 def generate_all_messages(desc, benefits, key_message):
     # Варианты предложений для начала и конца
     prefixes = [
@@ -137,11 +137,11 @@ def generate_all_messages(desc, benefits, key_message):
         yield (
             non_personalized_prompt, personalized_prompt,  # Задания для копирайтера и редактора
             non_personalized_messages[0] if i >= 0 else None,  # Первое неперсонализированное сообщение
-            None,  # Первое персонализированное сообщение пока не готово
+            personalized_messages[0] if len(personalized_messages) > 0 else None,  # Первое персонализированное сообщение, если оно уже есть
             non_personalized_messages[1] if i >= 1 else None,  # Второе неперсонализированное сообщение
-            None,  # Второе персонализированное сообщение пока не готово
+            personalized_messages[1] if len(personalized_messages) > 1 else None,  # Второе персонализированное сообщение, если оно уже есть
             non_personalized_messages[2] if i >= 2 else None,  # Третье неперсонализированное сообщение
-            None  # Третье персонализированное сообщение пока не готово
+            personalized_messages[2] if len(personalized_messages) > 2 else None  # Третье персонализированное сообщение, если оно уже есть
         )
         
         # Генерация персонализированного сообщения
@@ -152,12 +152,12 @@ def generate_all_messages(desc, benefits, key_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  # Третье персонализированное сообщение
+            non_personalized_messages[0] if len(non_personalized_messages) > 0 else None,  # Первое неперсонализированное сообщение
+            personalized_messages[0] if len(personalized_messages) > 0 else None,  # Первое персонализированное сообщение
+            non_personalized_messages[1] if len(non_personalized_messages) > 1 else None,  # Второе неперсонализированное сообщение
+            personalized_messages[1] if len(personalized_messages) > 1 else None,  # Второе персонализированное сообщение
+            non_personalized_messages[2] if len(non_personalized_messages) > 2 else None,  # Третье неперсонализированное сообщение
+            personalized_messages[2] if len(personalized_messages) > 2 else None  # Третье персонализированное сообщение
         )
 
         # Небольшая пауза между выводом каждого сообщения