Update app.py

app.py

@@ -577,7 +577,7 @@ def generate_personalized_sms_wrapper(selected_product, description, product_nam
     final_prompt_2 = generate_final_prompt_from_display(prompt_2, chosen_single_approach_2, is_prompt_1=False)
     print("Final Prompt 1:", final_prompt_1)
     print("Final Prompt 2:", final_prompt_2)
-    sms_1 = generate_sms_with_timer(final_prompt_1, product_name)
+    sms_1 = generate_sms_with_timer(final_prompt_1, product_name, key_message)
 
     # Выполняем проверки sms_1
     cut_sms_1 = cut_message(sms_1)
@@ -586,7 +586,7 @@ def generate_personalized_sms_wrapper(selected_product, description, product_nam
 
     yield(sms_1, "", final_prompt_1, "", checks_formatted_1, "")
 
-    sms_2 = generate_sms_with_timer(final_prompt_2, product_name)
+    sms_2 = generate_sms_with_timer(final_prompt_2, product_name, key_message)
 
     # Выполняем проверки sms_2
     cut_sms_2 = cut_message(sms_2)
@@ -619,8 +619,8 @@ def on_regenerate(
     print("Regen Final Prompt 1:", final_prompt_1)
     print("Regen Final Prompt 2:", final_prompt_2)
 
-    sms_1 = generate_sms_with_timer(final_prompt_1, product_name)
-    sms_2 = generate_sms_with_timer(final_prompt_2, product_name)
+    sms_1 = generate_sms_with_timer(final_prompt_1, product_name, key_message)
+    sms_2 = generate_sms_with_timer(final_prompt_2, product_name, key_message)
 
     # Проверяем заново:
     cut_sms_1 = cut_message(sms_1)