Update app.py

app.py

@@ -187,9 +187,16 @@ def generate_standard_prompt(description, advantages, key_message, approach, *se
         "Убедись, что в готовом тексте до 250 знаков с пробелами.\n"
     )
 
-    if key_message.strip():
-        prompt += f"Убедись, что в сообщении есть следующая ключевая информация: {key_message.strip()}"
+    if approach == "Призыв к действию":
+        prompt+ = "Убедись, что готовый текст начинается с призыва к действию с продуктом.\n"
+    elif approach == "Указание на пользу":
+        prompt = "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.\n"
+    elif approach == "Вопрос":
+        prompt = "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента.\n"
         
+    if key_message.strip():
+        prompt += f"Убедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
+     
     return prompt
 
 # Функции для генерации сообщений
@@ -499,7 +506,7 @@ def generate_messages(description, advantages, key_message, approach, *selected_
 
 
 # Функция для генерации персонализированного промпта
-def generate_personalization_prompt(key_message, *selected_values):
+def generate_personalization_prompt(key_message, approach, *selected_values):
     prompt = "Адаптируй, не превышая длину сообщения в 250 знаков с пробелами, текст с учетом следующих особенностей:\n"
     gender, generation, psychotype = selected_values[0], selected_values[1], selected_values[2]
     combined_instruction = ""
@@ -551,7 +558,14 @@ def generate_personalization_prompt(key_message, *selected_values):
 
     prompt += "Убедись, что в готовом тексте до 250 знаков с пробелами."
 
-    prompt += f"\nУбедись, что в сообщении есть следующая ключевая информация: {key_message.strip()}"
+    if approach == "Призыв к действию":
+        prompt+ = "Убедись, что готовый текст начинается с призыва к действию с продуктом.\n"
+    elif approach == "Указание на пользу":
+        prompt = "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.\n"
+    elif approach == "Вопрос":
+        prompt = "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента.\n"
+
+    prompt += f"Убедись, что в готовом тексте есть следующая ключевая информация: {key_message.strip()}"
 
     return prompt.strip()
 
@@ -612,7 +626,7 @@ def personalize_messages_with_yield(
     *selected_values
 ):
     
-    personalization_prompt = generate_personalization_prompt(key_message, *selected_values)
+    personalization_prompt = generate_personalization_prompt(key_message, approach, *selected_values)
     yield personalization_prompt, "", "", "", "", "", "", "", "", "", ""
 
     personalized_message_gpt4o = perform_personalization(gpt4o_message, personalization_prompt)