Update app.py

app.py

@@ -206,25 +206,6 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
     # Небольшая пауза для демонстрации постепенной генерации
     time.sleep(1)
 
-    # Варианты предложений для начала и конца
-    prefixes = [
-        "Начни сообщение с призыва к действию с продуктом.",
-        "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
-        "Начни сообщение с вопроса, который указывает на пользу продукта для клиента."
-    ]
-    suffixes = [
-        "Убедись, что готовый текст начинается с призыва к действию с продуктом.",
-        "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.",
-        "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
-    ]
-
-    # Генерация задания для редактора
-    personalization_prompt = generate_personalization_prompt(key_message, prefixes[i], suffixes[i], gender, generation, psychotype, business_stage, industry, opf)
-    yield standard_prompt, personalization_prompt, None, None, None, None, None, None
-    
-    # Небольшая пауза для демонстрации постепенной генерации
-    time.sleep(1)
-
     # Варианты предложений для начала и конца
     prefixes = [
         "Начни сообщение с призыва к действию с продуктом.",
@@ -239,19 +220,23 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
     
     non_personalized_messages = []
     personalized_messages = []
-    
+
     # Генерация и постепенная подача каждого сообщения
     for i in range(3):
+        # Генерация задания для редактора с использованием 'prefixes' и 'suffixes'
+        personalization_prompt = generate_personalization_prompt(key_message, gender, generation, psychotype, business_stage, industry, opf)
+        prompt_with_prefix_suffix = add_prefix_suffix(personalization_prompt, prefixes[i], suffixes[i])
+        
         # Генерация неперсонализированного сообщения
         prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
         non_personalized_message = generate_message_gigachat_pro_with_retry(prompt)
         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)
-        
+
         # Выводим неперсонализированное сообщение
         yield (
-            standard_prompt, personalization_prompt,  # Задания для копирайтера и редактора
+            standard_prompt, prompt_with_prefix_suffix,  # Задания для копирайтера и редактора
             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,  # Второе неперсонализированное сообщение
@@ -259,9 +244,9 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             non_personalized_messages[2] if i >= 2 else None,  # Третье неперсонализированное сообщение
             personalized_messages[2] if len(personalized_messages) > 2 else None  # Третье персонализированное сообщение, если оно уже есть
         )
-        
+
         # Генерация персонализированного сообщения
-        full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
+        full_personalized_prompt = f"{prompt_with_prefix_suffix}\n\nТекст для адаптации: {non_personalized_message}"
         personalized_message = generate_message_gigachat_pro_with_retry(full_personalized_prompt)
         personalized_length = len(personalized_message)  # Подсчитываем количество знаков
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
@@ -269,7 +254,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
 
         # Выводим персонализированное сообщение
         yield (
-            standard_prompt, personalization_prompt,  # Задания для копирайтера и редактора
+            standard_prompt, prompt_with_prefix_suffix,  # Задания для копирайтера и редактора
             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,  # Второе неперсонализированное сообщение
@@ -281,6 +266,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         # Небольшая пауза между выводом каждого сообщения
         time.sleep(1)
 
+
 # ФУНКЦИИ ПРОВЕРОК (НАЧАЛО)
 
 # 1. Запрещенные слова