Update app.py

app.py

@@ -519,8 +519,8 @@ def 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, sms_1,
-    model_2_name, prompt_2, sms_2,
+    model_1_name, prompt_1, final_prompt_1, sms_1,
+    model_2_name, prompt_2, final_prompt_2, sms_2,
     chosen_sms
 ):
     """
@@ -545,9 +545,11 @@ def save_preferred_sms_to_github(
         "comment": presence_in_db,
         "model_1_name": model_1_name,
         "prompt_1": prompt_1,
+        "final_prompt_1": final_prompt_1,
         "sms_1": sms_1,
         "model_2_name": model_2_name,
         "prompt_2": prompt_2,
+        "final_prompt_2": final_prompt_2,
         "sms_2": sms_2,
         "preferred_sms": chosen_sms  # "sms_1" или "sms_2"
     }
@@ -573,8 +575,8 @@ def on_prefer_sms_1(
     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, sms_1,
-    model_2_name, prompt_2, sms_2
+    model_1_name, prompt_1, final_prompt_1, sms_1,
+    model_2_name, prompt_2, final_prompt_2, sms_2
 ):
     """
     Вызывается при нажатии кнопки «Я предпочитаю это SMS» (для SMS 1).
@@ -593,8 +595,8 @@ def on_prefer_sms_2(
     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, sms_1,
-    model_2_name, prompt_2, sms_2
+    model_1_name, prompt_1, final_prompt_1, sms_1,
+    model_2_name, prompt_2, final_prompt_2, sms_2
 ):
     """
     Вызывается при нажатии кнопки «Я предпочитаю это SMS» (для SMS 2).
@@ -686,8 +688,8 @@ with gr.Blocks(theme="default") as demo:
             sms_2 = gr.Textbox(label="SMS 2", lines=3, value="", interactive=False)
     
     with gr.Row():
-        prefer_sms_1_btn = gr.Button("Я предпочитаю это SMS (кнопка пока не работает)")
-        prefer_sms_2_btn = gr.Button("Я предпочитаю это SMS (кнопка пока не работает)")
+        prefer_sms_1_btn = gr.Button("Я предпочитаю это SMS")
+        prefer_sms_2_btn = gr.Button("Я предпочитаю это SMS")
 
     regen_btn = gr.Button("Перегенерировать SMS (не нравится ни одно из SMS)")
     
@@ -760,4 +762,61 @@ with gr.Blocks(theme="default") as demo:
                  sms_1, sms_2, comment_sms_1, comment_sms_2, corrected_sms_1, corrected_sms_2]
     )
 
+    prefer_sms_1_btn.click(
+        fn=on_prefer_sms_1,
+        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=[]
+    )
+
+    prefer_sms_2_btn.click(
+        fn=on_prefer_sms_2,
+        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=[]
+    )
+
+
     demo.queue().launch()