Update app.py

app.py

@@ -743,7 +743,6 @@ def on_save_sms_1(
     )
     return "SMS 1 сохранено в базу"
 
-
 def on_save_sms_2(
     selected_product,
     description,
@@ -788,6 +787,15 @@ def on_save_sms_2(
     )
     return "SMS 2 сохранено в базу"
 
+def prepare_button_text():
+    return gr.update(value="Сохраняется...", visible=True)
+
+def update_button_text():
+    return gr.update(value="Сохранено!", visible=True)
+
+def reset_button_text():
+    time.sleep(2)
+    return gr.update(value="Сохранить в базу", visible=True)
 
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -962,6 +970,10 @@ with gr.Blocks(theme="default") as demo:
     )
 
     prefer_sms_1_btn.click(
+        fn=prepare_button_text,
+        inputs=[],
+        outputs=[prefer_sms_1_btn]
+    ).then(
         fn=on_prefer_sms_1,
         inputs=[
             product_dropdown,
@@ -986,10 +998,22 @@ with gr.Blocks(theme="default") as demo:
             final_prompt_2_state,
             sms_2
         ],
-        outputs=[]
+        outputs=[]  # или выводим что-то в текстбокс
+    ).then(
+        fn=update_button_text,
+        inputs=[],
+        outputs=[prefer_sms_1_btn]
+    ).then(
+        fn=reset_button_text,
+        inputs=[],
+        outputs=[prefer_sms_1_btn]
     )
 
     prefer_sms_2_btn.click(
+        fn=prepare_button_text,
+        inputs=[],
+        outputs=[prefer_sms_2_btn]
+    ).then(
         fn=on_prefer_sms_2,
         inputs=[
             product_dropdown,
@@ -1014,10 +1038,22 @@ with gr.Blocks(theme="default") as demo:
             final_prompt_2_state,
             sms_2
         ],
-        outputs=[]
+        outputs=[]  # или выводим что-то в текстбокс
+    ).then(
+        fn=update_button_text,
+        inputs=[],
+        outputs=[prefer_sms_2_btn]
+    ).then(
+        fn=reset_button_text,
+        inputs=[],
+        outputs=[prefer_sms_2_btn]
     )
 
     save_sms_1_btn.click(
+        fn=prepare_button_text,
+        inputs=[],
+        outputs=[save_sms_1_btn]
+    ).then(
         fn=on_save_sms_1,
         inputs=[
             product_dropdown,
@@ -1041,9 +1077,21 @@ with gr.Blocks(theme="default") as demo:
             corrected_sms_1
         ],
         outputs=[]
+    ).then(
+        fn=update_button_text,
+        inputs=[],
+        outputs=[save_sms_1_btn]
+    ).then(
+        fn=reset_button_text,
+        inputs=[],
+        outputs=[save_sms_1_btn]
     )
-    
+
     save_sms_2_btn.click(
+        fn=prepare_button_text,
+        inputs=[],
+        outputs=[save_sms_2_btn]
+    ).then(
         fn=on_save_sms_2,
         inputs=[
             product_dropdown,
@@ -1067,7 +1115,15 @@ with gr.Blocks(theme="default") as demo:
             corrected_sms_2
         ],
         outputs=[]
+    ).then(
+        fn=update_button_text,
+        inputs=[],
+        outputs=[save_sms_2_btn]
+    ).then(
+        fn=reset_button_text,
+        inputs=[],
+        outputs=[save_sms_2_btn]
     )
-
+    
 
     demo.queue().launch()