Update app.py

app.py

@@ -740,7 +740,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             yield selected_approaches_text_content, standard_prompt_for_display, display_personalization_prompt, None, None
             flag += 1
         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_message = generate_message_mistral_with_retry(prompt, approach_name, desc, key_message, product_name)
         non_personalized_length = len(non_personalized_message)
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"
         if non_personalized_messages:
@@ -752,7 +752,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             non_personalized_messages, personalized_messages
         )
         full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
-        personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, approach_name, desc, key_message)
+        personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, approach_name, desc, key_message, product_name)
         personalized_length = len(personalized_message)
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
         if personalized_messages: