Update app.py

app.py

@@ -837,7 +837,7 @@ def update_button_text():
 
 def reset_button_text():
     time.sleep(3)  # Задержка в 3 секунды
-    return gr.update(value="👍 GPT-4o", visible=True)
+    return gr.update(value="Сохранить в базу", visible=True)
 
 
 # Создание интерфейса Gradio
@@ -891,7 +891,7 @@ with gr.Blocks() as demo:
         personalized_output_text_gpt4o = gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=3, interactive=False)
         comment_gpt4o = gr.Textbox(label="Комментарий к сообщению GPT-4o", lines=3)
         corrected_gpt4o = gr.Textbox(label="Откорректированное сообщение GPT-4o", lines=3)
-        save_gpt4o_btn = gr.Button("👍 GPT-4o")
+        save_gpt4o_btn = gr.Button("Сохранить в базу")
 
     # Второй ряд
     with gr.Row():
@@ -899,7 +899,7 @@ with gr.Blocks() as demo:
         personalized_output_text_gigachat_pro = gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=3, interactive=False)
         comment_gigachat_pro = gr.Textbox(label="Комментарий к сообщению GigaChat-Pro", lines=3)
         corrected_gigachat_pro = gr.Textbox(label="Откорректированное сообщение GigaChat-Pro", lines=3)
-        save_gigachat_pro_btn = gr.Button("👍 GigaChat-Pro")
+        save_gigachat_pro_btn = gr.Button("Сохранить в базу")
 
     # Третий ряд
     with gr.Row():
@@ -907,7 +907,7 @@ with gr.Blocks() as demo:
         personalized_output_text_gigachat_lite = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
         comment_gigachat_lite = gr.Textbox(label="Комментарий к сообщению GigaChat-Lite", lines=3)
         corrected_gigachat_lite = gr.Textbox(label="Откорректированное сообщение GigaChat-Lite", lines=3)
-        save_gigachat_lite_btn = gr.Button("👍 GigaChat-Lite")
+        save_gigachat_lite_btn = gr.Button("Сохранить в базу")
 
     # Четвертый ряд
     with gr.Row():
@@ -915,7 +915,7 @@ with gr.Blocks() as demo:
         personalized_output_text_gigachat_plus = gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=3, interactive=False)
         comment_gigachat_plus = gr.Textbox(label="Комментарий к сообщению GigaChat-Lite+", lines=3)
         corrected_gigachat_plus = gr.Textbox(label="Откорректированное сообщение GigaChat-Lite+", lines=3)
-        save_gigachat_plus_btn = gr.Button("👍 GigaChat-Lite+")
+        save_gigachat_plus_btn = gr.Button("Сохранить в базу")
         
     # Пятый ряд
     with gr.Row():
@@ -923,7 +923,7 @@ with gr.Blocks() as demo:
         personalized_output_text_meta_llama_405b = gr.Textbox(label="Персонализированное сообщение Meta-Llama-3.1-405B", lines=3, interactive=False)
         comment_meta_llama_405b = gr.Textbox(label="Комментарий к сообщению Meta-Llama-3.1-405B", lines=3)
         corrected_meta_llama_405b = gr.Textbox(label="Откорректированное сообщение Meta-Llama-3.1-405B", lines=3)
-        save_meta_llama_405b_btn = gr.Button("👍 Meta-Llama-3.1-405B")
+        save_meta_llama_405b_btn = gr.Button("Сохранить в базу")
     
     # Шестой ряд
     with gr.Row():
@@ -931,7 +931,7 @@ with gr.Blocks() as demo:
         personalized_output_text_meta_llama_70b = gr.Textbox(label="Персонализированное сообщение Meta-Llama-3.1-70B", lines=3, interactive=False)
         comment_meta_llama_70b = gr.Textbox(label="Комментарий к сообщению Meta-Llama-3.1-70B", lines=3)
         corrected_meta_llama_70b = gr.Textbox(label="Откорректированное сообщение Meta-Llama-3.1-70B", lines=3)
-        save_meta_llama_70b_btn = gr.Button("👍 Meta-Llama-3.1-70B")
+        save_meta_llama_70b_btn = gr.Button("Сохранить в базу")
     
     # Седьмой ряд
     with gr.Row():
@@ -939,7 +939,7 @@ with gr.Blocks() as demo:
         personalized_output_text_meta_llama_8b = gr.Textbox(label="Персонализированное сообщение Meta-Llama-3.1-8B", lines=3, interactive=False)
         comment_meta_llama_8b = gr.Textbox(label="Комментарий к сообщению Meta-Llama-3.1-8B", lines=3)
         corrected_meta_llama_8b = gr.Textbox(label="Откорректированное сообщение Meta-Llama-3.1-8B", lines=3)
-        save_meta_llama_8b_btn = gr.Button("👍 Meta-Llama-3.1-8B")
+        save_meta_llama_8b_btn = gr.Button("Сохранить в базу")
     
     # Восьмой ряд
     with gr.Row():
@@ -947,7 +947,7 @@ with gr.Blocks() as demo:
         personalized_output_text_gemma_27b = gr.Textbox(label="Персонализированное сообщение Gemma-2-27B-IT", lines=3, interactive=False)
         comment_gemma_27b = gr.Textbox(label="Комментарий к сообщению Gemma-2-27B-IT", lines=3)
         corrected_gemma_27b = gr.Textbox(label="Откорректированное сообщение Gemma-2-27B-IT", lines=3)
-        save_gemma_27b_btn = gr.Button("👍 Gemma-2-27B-IT")
+        save_gemma_27b_btn = gr.Button("Сохранить в базу")
     
     # Девятый ряд
     with gr.Row():
@@ -955,7 +955,7 @@ with gr.Blocks() as demo:
         personalized_output_text_gemma_9b = gr.Textbox(label="Персонализированное сообщение Gemma-2-9B-IT", lines=3, interactive=False)
         comment_gemma_9b = gr.Textbox(label="Комментарий к сообщению Gemma-2-9B-IT", lines=3)
         corrected_gemma_9b = gr.Textbox(label="Откорректированное сообщение Gemma-2-9B-IT", lines=3)
-        save_gemma_9b_btn = gr.Button("👍 Gemma-2-9B-IT")
+        save_gemma_9b_btn = gr.Button("Сохранить в базу")
 
     # Десятый ряд
     with gr.Row():
@@ -963,7 +963,7 @@ with gr.Blocks() as demo:
         personalized_output_text_mistral = gr.Textbox(label="Персонализированное сообщение Mistral-Large-2407", lines=3, interactive=False)
         comment_mistral = gr.Textbox(label="Комментарий к сообщению Mistral-Large-2407", lines=3)
         corrected_mistral = gr.Textbox(label="Откорректированное сообщение Mistral-Large-2407", lines=3)
-        save_mistral_btn = gr.Button("👍 Mistral-Large-2407")
+        save_mistral_btn = gr.Button("Сохранить в базу")
 
    
     # Добавление функционала для кнопок
@@ -1135,6 +1135,10 @@ with gr.Blocks() as demo:
 
 
     save_gigachat_pro_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_gigachat_pro_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "GigaChat-Pro", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1153,9 +1157,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_gigachat_pro_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_gigachat_pro_btn]
     )
 
     save_gigachat_lite_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_gigachat_lite_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "GigaChat-Lite", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1174,9 +1188,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_gigachat_lite_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_gigachat_lite_btn]
     )
 
     save_gigachat_plus_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_gigachat_plus_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "GigaChat-Lite+", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1195,9 +1219,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_gigachat_plus_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_gigachat_plus_btn]
     )
 
     save_meta_llama_405b_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_meta_llama_405b_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "Meta-Llama-3.1-405B", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1216,9 +1250,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_meta_llama_405b_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_meta_llama_405b_btn]
     )
     
     save_meta_llama_70b_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_meta_llama_70b_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "Meta-Llama-3.1-70B", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1237,9 +1281,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_meta_llama_70b_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_meta_llama_70b_btn]
     )
     
     save_meta_llama_8b_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_meta_llama_8b_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "Meta-Llama-3.1-8B", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1258,9 +1312,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_meta_llama_8b_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_meta_llama_8b_btn]
     )
     
     save_gemma_27b_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_gemma_27b_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "Gemma-2-27B-IT", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1279,9 +1343,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_gemma_27b_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_gemma_27b_btn]
     )
     
     save_gemma_9b_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_gemma_9b_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "Gemma-2-9B-IT", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1300,9 +1374,19 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_gemma_9b_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_gemma_9b_btn]
     )
 
     save_mistral_btn.click(
+        fn=prepare_button_text,  # Сначала меняем текст на "Сохраняется..."
+        inputs=[],
+        outputs=[save_mistral_btn]
+    ).then(
         fn=lambda personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form: 
             save_to_github(personalized_message, "Mistral", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form),
         inputs=[
@@ -1321,6 +1405,12 @@ with gr.Blocks() as demo:
             selections[5]   # ОПФ
         ],
         outputs=None
+    ).then(
+        fn=update_button_text,  # Обновляем текст на "Сохранено!" после сохранения
+        outputs=[save_mistral_btn]
+    ).then(
+        fn=reset_button_text,  # Возвращаем текст на кнопке обратно через 3 секунды
+        outputs=[save_mistral_btn]
     )