Update app.py

app.py

@@ -213,52 +213,53 @@ 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 = []
-    
+
     # Генерация и постепенная подача каждого сообщения
     for i in range(3):
-        # Генерация неперсонализированного сообщения с добавлением префиксов и суффиксов
+        # Генерация неперсонализированного сообщения
         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 = generate_personalization_prompt(
+        # Генерация задания для редактора
+        personalization_prompt = generate_personalization_prompt(
             key_message, gender, generation, psychotype, business_stage, industry, opf,
-            prefix=prefixes[i],  # Добавляем префиксы для генерации
-            suffix=suffixes[i]    # Добавляем суффиксы для генерации
+            prefix=prefixes[i], suffix=suffixes[i]
         )
+
+        # Выводим неперсонализированное сообщение и задание для редактора
+        yield (
+            standard_prompt, 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
+        )
+
+        # Генерация персонализированного сообщения
+        full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
         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 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, 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
         )
 
-        # Небольшая пауза между выводом каждого сообщения
         time.sleep(1)