Update app.py

app.py

@@ -153,8 +153,15 @@ def change_tab(id):
     return gr.Tabs(selected=id)
 
 # Вспомогательная функция для добавления префиксов и суффиксов
-def add_prefix_suffix(prompt, prefix, suffix):
-    return f"{prefix}\n{prompt}\n{suffix}"
+def add_prefix_suffix(prompt, prefix, suffix, product_name):
+    new_prompt = f"{prefix}\n"
+    # Добавляем инструкцию после prefix
+    new_prompt += f"Не изменяй название продукта: {product_name}.\n"
+    new_prompt += f"{prompt}\n"
+    # Добавляем инструкцию перед suffix
+    new_prompt += f"Убедись, что в готовом тексте без изменений, синонимов и перестановок слов используется наименование продукта: {product_name}.\n"
+    new_prompt += f"{suffix}"
+    return new_prompt
 
 # Функция для обрезки сообщения до последнего знака препинания
 def clean_message(message):
@@ -182,7 +189,7 @@ def generate_message_mistral_generate(prompt, max_retries=5):
                 model="mistral-large-latest",
                 temperature=1.0,
                 min_tokens=81,
-                max_tokens=109,
+                max_tokens=108,
                 messages=[
                     {
                         "role": "user",
@@ -592,8 +599,10 @@ def generate_standard_prompt(description, advantages, key_message, *selected_val
 
 
 # Функция для создания задания для редактора с добавлением prefix и suffix
-def generate_personalization_prompt(key_message, *selected_values, prefix, suffix):
-    prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами (но и не менее 190 знаков с пробелами), текст с учетом следующих особенностей:\n"
+def generate_personalization_prompt(key_message, *selected_values, prefix, suffix, product_name):
+    prompt = f"{prefix}\n"
+    prompt += f"Не изменяй название продукта: {product_name}.\n"
+    prompt += "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами (но и не менее 190 знаков с пробелами), текст с учетом следующих особенностей:\n"
     gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
     combined_instruction = ""
     additional_instructions = ""
@@ -634,12 +643,10 @@ def generate_personalization_prompt(key_message, *selected_values, prefix, suffi
     if additional_instructions:
         prompt += additional_instructions  # Добавляем остальные инструкции
 
-    # Добавляем префикс и суффикс для задания редактора
-    prompt = f"{prefix}\n{prompt}\n{suffix}"
-    prompt += "Убедись, что в готовом тексте не изменено название предлагаемого продукта.\n"
-    
-    # Добавляем ключевое сообщение
-    prompt += f"\nУбедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
+    prompt += f"{suffix}"
+    prompt += f"Убедись, что в готовом тексте без изменений, синонимов и перестановок слов используется наименование продукта: {product_name}.\n"
+    if key_message.strip():
+        prompt += f"\nУбедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
 
     return prompt.strip()
 
@@ -659,7 +666,7 @@ def clean_prompt_for_display(prompt, prefixes, suffixes):
     return cleaned_prompt.strip()
 
 # Функция для постепенной генерации всех сообщений через yield
-def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
+def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf, product_name):
     standard_prompt = generate_standard_prompt(desc, benefits, key_message)
     approach_mapping = {
         "Указание на пользу": (
@@ -729,7 +736,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         if flag == 1:
             yield selected_approaches_text_content, standard_prompt, display_personalization_prompt, None, None
             flag += 1
-        prompt = add_prefix_suffix(standard_prompt, current_prefix, current_suffix)
+        prompt = add_prefix_suffix(standard_prompt, current_prefix, current_suffix, product_name)
         non_personalized_message = generate_message_mistral_with_retry(prompt, approach_name, desc, key_message)
         non_personalized_length = len(non_personalized_message)
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"