Update app.py

app.py

@@ -153,8 +153,13 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
         formatted_line = format_instruction_string(instr_line)
         lines.append(f"{i}. {formatted_line}.")
     mandatory_terms = "\n".join(lines)
+    
+    extra_line = ""
+    if generation == "Z":
+        extra_line = "Обратись в SMS на ты. "
+
     prompt_1 = f"""Напиши три или четыре предложения суммарной длиной от 160 до 250 знаков с учетом пробелов. {approach_prefix}. 
-Напиши рекламное SMS для следующего продукта:
+{extra_line}Напиши рекламное SMS для следующего продукта:
 «{description}».
 Не изменяй название продукта: «{product_name}».
 Преимущества:
@@ -166,8 +171,9 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
 Убедись, что УМЕСТНО использовал КАЖДЫЙ необходимый термин.
 Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+
     prompt_2 = f"""Напиши три или четыре предложения суммарной длиной от 160 до 250 знаков с учетом пробелов. {approach_prefix}. 
-Напиши рекламное SMS для следующего продукта:
+{extra_line}Напиши рекламное SMS для следующего продукта:
 «{description}».
 Не изменяй название продукта: «{product_name}».
 Преимущества:
@@ -179,6 +185,7 @@ def generate_display_prompts(description, product_name, benefits, key_message, c
 Убедись, что УМЕСТНО использовал КАЖДЫЙ необходимый термин.
 Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
 Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+
     return prompt_1, prompt_2
 
 def call_model(model_prompt):