Update app.py

app.py

@@ -302,24 +302,39 @@ def generate_final_prompt_from_display(prompt_text, single_approach, is_prompt_1
                          f"\n{suffix}.\nУбедись, что УМЕСТНО использовал КАЖДЫЙ необходимый термин.\nУбедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта:\nУбедись, что в SMS есть следующая ключевая информация:", prompt_text, flags=re.DOTALL)
     return prompt_text
 
+final_prompt_1_state = gr.State("")
+final_prompt_2_state = gr.State("")
+
 def generate_personalized_sms_wrapper(selected_product, description, product_name, benefits, key_message,
                                       gender, generation, psychotype, business_stage, industry, opf,
                                       chosen_approach, prompt_1, prompt_2):
     if "Для формирования промпта выберите хотя бы один личный персональный параметр" in prompt_1 or chosen_approach == "Подход не найден для выбранных параметров.":
         gr.Warning("Задайте хотя бы один личный персональный параметр для определения подхода, чтобы был сформирован промпт")
-        return "", ""
+        return "", "", "", ""
     approach_list = [a.strip() for a in chosen_approach.split(',') if a.strip()]
     if not approach_list:
         gr.Warning("Задайте хотя бы один личный персональный параметр для определения подхода, чтобы был сформирован промпт")
-        return "", ""
+        return "", "", "", ""
     chosen_single_approach_1 = random.choice(approach_list) if len(approach_list) > 1 else approach_list[0]
     chosen_single_approach_2 = random.choice(approach_list) if len(approach_list) > 1 else approach_list[0]
     final_prompt_1 = generate_final_prompt_from_display(prompt_1, chosen_single_approach_1, is_prompt_1=True)
     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_message_with_retry(final_prompt_1)
     sms_2 = generate_message_with_retry(final_prompt_2)
     personalization_params = [gender, generation, psychotype, business_stage, industry, opf]
     save_user_request_to_github(selected_product, description, product_name, benefits, key_message, chosen_approach, personalization_params)
+    return sms_1, sms_2, final_prompt_1, final_prompt_2
+
+def regen_sms(final_prompt_1, final_prompt_2):
+    if not final_prompt_1.strip() or not final_prompt_2.strip():
+        gr.Warning("Нечего перегенерировать, сначала создайте SMS.")
+        return "", ""
+    print("Regen Final Prompt 1:", final_prompt_1)
+    print("Regen Final Prompt 2:", final_prompt_2)
+    sms_1 = generate_message_with_retry(final_prompt_1)
+    sms_2 = generate_message_with_retry(final_prompt_2)
     return sms_1, sms_2
 
 def on_load_previous():
@@ -414,7 +429,7 @@ with gr.Blocks(theme="default") as demo:
         prefer_sms_1_btn = gr.Button("Я предпочитаю это SMS (кнопка пока не работает)")
         prefer_sms_2_btn = gr.Button("Я предпочитаю это SMS (кнопка пока не работает)")
 
-    regen_btn = gr.Button("Перегенерировать SMS (не нравится ни одно из SMS) (кнопка пока не работает)")
+    regen_btn = gr.Button("Перегенерировать SMS (не нравится ни одно из SMS)")
     
     with gr.Row():
         comment_sms_1 = gr.Textbox(label="Комментарий к SMS 1", lines=2, value="")
@@ -428,6 +443,9 @@ with gr.Blocks(theme="default") as demo:
         save_sms_1_btn = gr.Button("Сохранить в базу (кнопка пока не работает)")
         save_sms_2_btn = gr.Button("Сохранить в базу (кнопка пока не работает)")
 
+    final_prompt_1_state = gr.State("")
+    final_prompt_2_state = gr.State("")
+
     product_dropdown.change(
         fn=on_product_change,
         inputs=[product_dropdown, description, product_name, benefits, key_message,
@@ -463,6 +481,12 @@ with gr.Blocks(theme="default") as demo:
                 gender_dropdown, generation_dropdown, psychotype_dropdown,
                 business_stage_dropdown, industry_dropdown, opf_dropdown,
                 chosen_approach, prompt_1, prompt_2],
+        outputs=[sms_1, sms_2, final_prompt_1_state, final_prompt_2_state]
+    )
+
+    regen_btn.click(
+        fn=regen_sms,
+        inputs=[final_prompt_1_state, final_prompt_2_state],
         outputs=[sms_1, sms_2]
     )