Update app.py

app.py

@@ -201,14 +201,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
     # Небольшая пауза для демонстрации постепенной генерации
     time.sleep(1)
 
-    # Генерация задания для редактора без префиксов и суффиксов
-    personalization_prompt = generate_personalization_prompt(key_message, gender, generation, psychotype, business_stage, industry, opf)
-    yield standard_prompt, personalization_prompt, None, None, None, None, None, None
-
-    # Небольшая пауза для демонстрации постепенной генерации
-    time.sleep(1)
-
-    # Варианты предложений для начала и конца, которые будут добавляться, но не отображаться в поле редактора
+    # Варианты предложений для начала и конца
     prefixes = [
         "Начни сообщение с призыва к действию с продуктом.",
         "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
@@ -219,7 +212,18 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.",
         "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
     ]
-    
+
+    # Генерация задания для редактора без отображения префиксов и суффиксов
+    personalization_prompt = generate_personalization_prompt(
+        key_message, gender, generation, psychotype, business_stage, industry, opf,
+        prefix="",  # Пустые префиксы для отображения в поле задания
+        suffix=""   # Пустые суффиксы для отображения в поле задания
+    )
+    yield standard_prompt, personalization_prompt, None, None, None, None, None, None
+
+    # Небольшая пауза для демонстрации постепенной генерации
+    time.sleep(1)
+
     non_personalized_messages = []
     personalized_messages = []
     
@@ -228,26 +232,30 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         # Генерация неперсонализированного сообщения с добавлением префиксов и суффиксов
         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_length = len(non_personalized_message)  # Подсчитываем количество знаков
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"
         non_personalized_messages.append(non_personalized_display)
 
-        # Генерация персонализированного сообщения (с добавлением префиксов и суффиксов за кулисами)
-        full_personalized_prompt = add_prefix_suffix(personalization_prompt, prefixes[i], suffixes[i])
+        # Генерация персонализированного сообщения с добавлением префиксов и суффиксов только для генерации
+        full_personalized_prompt = generate_personalization_prompt(
+            key_message, gender, generation, psychotype, business_stage, industry, opf,
+            prefix=prefixes[i],  # Добавляем префиксы для генерации
+            suffix=suffixes[i]    # Добавляем суффиксы для генерации
+        )
         personalized_message = generate_message_gigachat_pro_with_retry(full_personalized_prompt)
-        personalized_length = len(personalized_message)
+        personalized_length = len(personalized_message)  # Подсчитываем количество знаков
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
         personalized_messages.append(personalized_display)
 
         # Выводим неперсонализированное и персонализированное сообщение
         yield (
             standard_prompt, 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
+            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  # Третье персонализированное сообщение
         )
 
         # Небольшая пауза между выводом каждого сообщения