Update app.py

app.py

@@ -492,14 +492,42 @@ def generate_personalized_sms_wrapper(selected_product, description, product_nam
     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, product_name):
+def on_regenerate(
+    selected_product, description, product_name, benefits, key_message,
+    gender, generation, psychotype, business_stage, industry, opf,
+    chosen_approach, presence_in_db,
+    model_1_name, prompt_1, final_prompt_1, sms_1,
+    model_2_name, prompt_2, final_prompt_2, sms_2
+):
+    """
+    Функция для кнопки «Перегенерировать SMS (не нравится ни одно из SMS)».
+    1) Перегенерирует sms_1, sms_2 с помощью final_prompt_1, final_prompt_2 и product_name.
+    2) Сохраняет все те же данные, что и on_prefer_sms_1/2, но с chosen_sms="none".
+    3) Возвращает новые sms_1, sms_2 для обновления интерфейса.
+    """
+
+    # Перегенерируем SMS (аналогично regen_sms)
     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, product_name)
     sms_2 = generate_message_with_retry(final_prompt_2, product_name)
+
+    # Теперь сохраняем всё, как при «Я предпочитаю это SMS»,
+    # только chosen_sms="none"
+    save_preferred_sms_to_github(
+        selected_product, description, product_name, benefits, key_message,
+        gender, generation, psychotype, business_stage, industry, opf,
+        chosen_approach, presence_in_db,
+        model_1_name, prompt_1, final_prompt_1, new_sms_1,
+        model_2_name, prompt_2, final_prompt_2, new_sms_2,
+        chosen_sms="none"  # <-- признак, что ни одно SMS не выбрано
+    )
+
     return sms_1, sms_2
 
 def on_load_previous():
@@ -747,8 +775,30 @@ with gr.Blocks(theme="default") as demo:
     )
 
     regen_btn.click(
-        fn=regen_sms,
-        inputs=[final_prompt_1_state, final_prompt_2_state, product_name],
+        fn=on_regenerate,
+        inputs=[
+            product_dropdown,
+            description,
+            product_name,
+            benefits,
+            key_message,
+            gender_dropdown,
+            generation_dropdown,
+            psychotype_dropdown,
+            business_stage_dropdown,
+            industry_dropdown,
+            opf_dropdown,
+            chosen_approach,
+            presence_in_db,
+            model_1_name,
+            prompt_1,
+            final_prompt_1_state,
+            sms_1,
+            model_2_name,
+            prompt_2,
+            final_prompt_2_state,
+            sms_2
+        ],
         outputs=[sms_1, sms_2]
     )