Update app.py

app.py

@@ -668,7 +668,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, product_name):
+def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf, product_name, advantages):
     standard_prompt = generate_standard_prompt(desc, benefits, key_message)
     standard_prompt_for_display = f"Не изменяй название продукта: {product_name}.\n{standard_prompt}\nУбедись, что в готовом тексте без изменений, синонимов и перестановок слов используется наименование продукта: {product_name}.\n"
     approach_mapping = {
@@ -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, product_name)
+        non_personalized_message = generate_message_mistral_with_retry(prompt, approach_name, desc, key_message, product_name, advantages)
         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, product_name)
+        personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, approach_name, desc, key_message, product_name, advantages)
         personalized_length = len(personalized_message)
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
         if personalized_messages:
@@ -1752,7 +1752,7 @@ with gr.Blocks() as demo:
                 outputs=tabs
             ).then(
                 fn=generate_all_messages,
-                inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf, product_name],
+                inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf, product_name, advantages],
                 outputs=[
                     selected_approaches_text,
                     non_personalized_prompt, personalized_prompt,