Update app.py

app.py

@@ -141,14 +141,12 @@ def generate_standard_prompt(description, advantages, key_message, *selected_val
 
 
 # Функция для создания задания для редактора с добавлением prefix и suffix
-def generate_personalization_prompt(key_message, prefix, suffix, *selected_values):
-    prompt = f"{prefix}\nАдаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
+def generate_personalization_prompt(key_message, *selected_values, prefix, suffix):
+    prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
     gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
     combined_instruction = ""
     additional_instructions = ""
 
-    print(f"Выбранные значения: Пол={gender}, Поколение={generation}, Психотип={psychotype}")
-
     # Проверяем, выбраны ли все три параметра: Пол, Поколение, Психотип
     if gender and generation and psychotype:
         # Получаем данные с листа "Пол Поколение Психотип"
@@ -158,19 +156,14 @@ def generate_personalization_prompt(key_message, prefix, suffix, *selected_value
         key = (gender, generation, psychotype)
         if key in sheet:
             combined_instruction = sheet[key]
-            print(f"Найдена комбинированная инструкция: {combined_instruction}")
-        else:
-            print(f"Комбинированная инструкция для ключа {key} не найдена.")
 
     # Если не найдена комбинированная инструкция, добавляем индивидуальные инструкции
     if not combined_instruction:
-        print("Добавляем индивидуальные инструкции для Пол, Поколение, Психотип.")
         for i, feature in enumerate(["Пол", "Поколение", "Психотип"]):
             if selected_values[i]:
                 try:
                     instruction = features[feature][selected_values[i]]
                     additional_instructions += f"{instruction}\n"
-                    print(f"Добавлена инструкция из {feature}: {instruction}")
                 except KeyError:
                     return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
 
@@ -181,7 +174,6 @@ def generate_personalization_prompt(key_message, prefix, suffix, *selected_value
                 try:
                     instruction = features[feature][selected_values[i]]
                     additional_instructions += f"{instruction}\n"
-                    print(f"Добавлена инструкция из {feature}: {instruction}")
                 except KeyError:
                     return f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных."
 
@@ -191,9 +183,12 @@ def generate_personalization_prompt(key_message, prefix, suffix, *selected_value
     if additional_instructions:
         prompt += additional_instructions  # Добавляем остальные инструкции
 
-    prompt += f"\n{suffix}"
+    # Добавляем префикс и суффикс для задания редактора
+    prompt = f"{prefix}\n{prompt}\n{suffix}"
+
+    # Добавляем ключевое сообщение
     prompt += f"\nУбедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
-    
+
     return prompt.strip()
 
 
@@ -206,13 +201,6 @@ 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 = [
         "Начни сообщение с призыва к действию с продуктом.",
@@ -224,50 +212,54 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.",
         "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
     ]
-    
+
     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)
-        
-        # Выводим неперсонализированное сообщение
+
+        # Генерация задания для редактора
+        personalization_prompt = generate_personalization_prompt(
+            key_message, gender, generation, psychotype, business_stage, industry, opf,
+            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  # Третье персонализированное сообщение, если оно уже есть
+            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
         )
-        
+
         # Генерация персонализированного сообщения
-        personalization_prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
         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 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, 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)