Update app.py

app.py

@@ -141,6 +141,7 @@ def generate_standard_prompt(description, advantages, key_message, *selected_val
 
 
 # Функция для создания задания для редактора с добавлением prefix и suffix
+# Функция для создания задания для редактора
 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]
@@ -192,6 +193,14 @@ def generate_personalization_prompt(key_message, *selected_values, prefix, suffi
     return prompt.strip()
 
 
+# Функция для удаления префиксов и суффиксов перед выводом на экран
+def remove_prefix_suffix(prompt, prefixes, suffixes):
+    for prefix in prefixes:
+        prompt = prompt.replace(prefix, "")
+    for suffix in suffixes:
+        prompt = prompt.replace(suffix, "")
+    return prompt.strip()
+
 # Функция для постепенной генерации всех сообщений через yield
 def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
     # Генерация задания для копирайтера
@@ -231,9 +240,12 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             prefix=prefixes[i], suffix=suffixes[i]
         )
 
+        # Удаляем префиксы и суффиксы перед выводом на экран
+        display_personalization_prompt = remove_prefix_suffix(personalization_prompt, prefixes, suffixes)
+
         # Выводим неперсонализированное сообщение и задание для редактора
         yield (
-            standard_prompt, personalization_prompt,
+            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,
@@ -251,7 +263,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
 
         # Выводим персонализированное сообщение
         yield (
-            standard_prompt, personalization_prompt,
+            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,