Update app.py

app.py

@@ -656,7 +656,7 @@ def clean_prompt_for_display(prompt, prefixes, suffixes):
 # Функция для постепенной генерации всех сообщений через yield
 def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
     standard_prompt = generate_standard_prompt(desc, benefits, key_message)
-    yield standard_prompt, None, None, None, None, None, None, None
+    yield standard_prompt, None, None, None
     prefixes = [
         "Начни сообщение с призыва к действию с продуктом.",
         "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
@@ -667,8 +667,8 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.",
         "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
     ]
-    non_personalized_messages = []
-    personalized_messages = []
+    non_personalized_messages = ''
+    personalized_messages = ''
     flag = 1
     for i in range(3):
         current_prefix = prefixes[i]
@@ -677,46 +677,36 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             prefix=prefixes[i], suffix=suffixes[i]
         )
         display_personalization_prompt = clean_prompt_for_display(personalization_prompt, prefixes, suffixes)
-        while flag == 1:
-            yield standard_prompt, display_personalization_prompt, None, None, None, None, None, None
+        if flag == 1:
+            yield standard_prompt, display_personalization_prompt, None, None
             flag += 1
         prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
         non_personalized_message = generate_message_mistral_with_retry(prompt, current_prefix, desc, key_message)
         non_personalized_length = len(non_personalized_message)
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"
-        non_personalized_messages.append(non_personalized_display)
-
-        # Выводим неперсонализированное сообщение и задание для редактора
+        # Добавляем новое сообщение с двумя пустыми строками
+        if non_personalized_messages:
+            non_personalized_messages += '\n\n' + non_personalized_display
+        else:
+            non_personalized_messages = non_personalized_display
         yield (
-            standard_prompt, display_personalization_prompt,  # Задание для редактора без префиксов, суффиксов и пустых строк
-            non_personalized_messages[0] if i >= 0 else None,
-            personalized_messages[0] if len(personalized_messages) > 0 else None,
-            non_personalized_messages[1] if i >= 1 else None,
-            personalized_messages[1] if len(personalized_messages) > 1 else None,
-            non_personalized_messages[2] if i >= 2 else None,
-            personalized_messages[2] if len(personalized_messages) > 2 else None
+            standard_prompt, display_personalization_prompt,
+            non_personalized_messages, personalized_messages
         )
-
-        # Генерация персонализированного сообщения
         full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
         personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, current_prefix, desc, key_message)
         personalized_length = len(personalized_message)
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
-        personalized_messages.append(personalized_display)
-
-        # Выводим персонализированное сообщение
+        # Добавляем новое сообщение с двумя пустыми строками
+        if personalized_messages:
+            personalized_messages += '\n\n' + personalized_display
+        else:
+            personalized_messages = personalized_display
         yield (
-            standard_prompt, display_personalization_prompt,  # Задание для редактора без префиксов, суффиксов и пустых строк
-            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
+            standard_prompt, display_personalization_prompt,
+            non_personalized_messages, personalized_messages
         )
-        
         time.sleep(1)
-
     save_statistics_to_github(approach_stats)