Update app.py

app.py

@@ -174,6 +174,38 @@ def save_message_to_table(data):
         print(f"Error updating messages file: {response.status_code}")
 
 
+def save_both(adapted_message, personalized_message, model_name, comment, corrected_message, description, advantages,
+              non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
+              business_stage, industry, legal_form, key_message, approach):
+    # Вызываем функцию для сохранения в JSON-файл
+    save_to_github(adapted_message, personalized_message, model_name, comment, corrected_message, description,
+                   advantages, non_personalized_prompt, non_personalized_message, personalization_prompt, gender,
+                   generation, psychotype, business_stage, industry, legal_form, key_message, approach)
+    
+    # Подготавливаем данные для сохранения в CSV/XLSX
+    data = {
+        "Timestamp": time.time(),
+        "Модель": model_name,
+        "Адаптированное сообщение": adapted_message,
+        "Персонализированное сообщение": personalized_message,
+        "Комментарий": comment,
+        "Откорректированное сообщение": corrected_message,
+        "Описание предложения": description,
+        "Преимущества": advantages,
+        "Ключевое сообщение": key_message,
+        "Подход": approach,
+        "Пол": gender,
+        "Поколение": generation,
+        "Психотип": psychotype,
+        "Стадия бизнеса": business_stage,
+        "Отрасль": industry,
+        "ОПФ": legal_form
+    }
+    
+    # Вызываем функцию для сохранения в CSV/XLSX
+    save_message_to_table(data)
+
+
 def load_previous_user_request_from_github():
     global current_request_index  # Используем глобальную переменную
 
@@ -2152,12 +2184,12 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[save_gigachat_pro_btn]
     ).then(
-        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt,
-            non_personalized_message, personalization_prompt, gender, generation, psychotype, business_stage, industry, legal_form,
-            key_message, approach:
-            save_to_github(adapted_message, personalized_message, "GigaChat-Pro", comment, corrected_message, description, advantages,
-                           non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
-                           business_stage, industry, legal_form, key_message, approach),
+        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages,
+               non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
+               business_stage, industry, legal_form, key_message, approach:
+               save_both(adapted_message, personalized_message, "GigaChat-Pro", comment, corrected_message, description,
+                         advantages, non_personalized_prompt, non_personalized_message, personalization_prompt, gender,
+                         generation, psychotype, business_stage, industry, legal_form, key_message, approach),
         inputs=[
             adapted_output_text_gigachat_pro,
             personalized_output_text_gigachat_pro,
@@ -2186,19 +2218,19 @@ with gr.Blocks() as demo:
         outputs=[save_gigachat_pro_btn]
     )
 
-    
+
     # Повторяем аналогично для других кнопок:
     save_gigachat_lite_btn.click(
         fn=prepare_button_text,
         inputs=[],
         outputs=[save_gigachat_lite_btn]
     ).then(
-        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt,
-            non_personalized_message, personalization_prompt, gender, generation, psychotype, business_stage, industry, legal_form,
-            key_message, approach:
-            save_to_github(adapted_message, personalized_message, "GigaChat-Lite", comment, corrected_message, description, advantages,
-                           non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
-                           business_stage, industry, legal_form, key_message, approach),
+        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages,
+               non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
+               business_stage, industry, legal_form, key_message, approach:
+               save_both(adapted_message, personalized_message, "GigaChat-Lite", comment, corrected_message, description,
+                         advantages, non_personalized_prompt, non_personalized_message, personalization_prompt, gender,
+                         generation, psychotype, business_stage, industry, legal_form, key_message, approach),
         inputs=[
             adapted_output_text_gigachat_lite,
             personalized_output_text_gigachat_lite,
@@ -2209,14 +2241,14 @@ with gr.Blocks() as demo:
             prompt_display,
             output_text_gigachat_lite,
             personalization_prompt,
-            selections[0],  # Пол
-            selections[1],  # Поколение
-            selections[2],  # Психотип
-            selections[3],  # Стадия бизнеса
-            selections[4],  # Отрасль
-            selections[5],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
+            selections[0],
+            selections[1],
+            selections[2],
+            selections[3],
+            selections[4],
+            selections[5],
+            key_message_input,
+            approach_input
         ],
         outputs=None
     ).then(
@@ -2233,12 +2265,12 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[save_gigachat_plus_btn]
     ).then(
-        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt,
-            non_personalized_message, personalization_prompt, gender, generation, psychotype, business_stage, industry, legal_form,
-            key_message, approach:
-            save_to_github(adapted_message, personalized_message, "GigaChat-Lite+", comment, corrected_message, description, advantages,
-                           non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
-                           business_stage, industry, legal_form, key_message, approach),
+        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages,
+               non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
+               business_stage, industry, legal_form, key_message, approach:
+               save_both(adapted_message, personalized_message, "GigaChat-Lite+", comment, corrected_message, description,
+                         advantages, non_personalized_prompt, non_personalized_message, personalization_prompt, gender,
+                         generation, psychotype, business_stage, industry, legal_form, key_message, approach),
         inputs=[
             adapted_output_text_gigachat_plus,
             personalized_output_text_gigachat_plus,
@@ -2249,14 +2281,14 @@ with gr.Blocks() as demo:
             prompt_display,
             output_text_gigachat_plus,
             personalization_prompt,
-            selections[0],  # Пол
-            selections[1],  # Поколение
-            selections[2],  # Психотип
-            selections[3],  # Стадия бизнеса
-            selections[4],  # Отрасль
-            selections[5],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
+            selections[0],
+            selections[1],
+            selections[2],
+            selections[3],
+            selections[4],
+            selections[5],
+            key_message_input,
+            approach_input
         ],
         outputs=None
     ).then(
@@ -2273,12 +2305,12 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[save_gpt4o_btn]
     ).then(
-        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt,
-            non_personalized_message, personalization_prompt, gender, generation, psychotype, business_stage, industry, legal_form,
-            key_message, approach:
-            save_to_github(adapted_message, personalized_message, "GPT-4o", comment, corrected_message, description, advantages,
-                           non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
-                           business_stage, industry, legal_form, key_message, approach),
+        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages,
+               non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
+               business_stage, industry, legal_form, key_message, approach:
+               save_both(adapted_message, personalized_message, "GPT-4o", comment, corrected_message, description,
+                         advantages, non_personalized_prompt, non_personalized_message, personalization_prompt, gender,
+                         generation, psychotype, business_stage, industry, legal_form, key_message, approach),
         inputs=[
             adapted_output_text_gpt4o,
             personalized_output_text_gpt4o,
@@ -2289,14 +2321,14 @@ with gr.Blocks() as demo:
             prompt_display,
             output_text_gpt4o,
             personalization_prompt,
-            selections[0],  # Пол
-            selections[1],  # Поколение
-            selections[2],  # Психотип
-            selections[3],  # Стадия бизнеса
-            selections[4],  # Отрасль
-            selections[5],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
+            selections[0],
+            selections[1],
+            selections[2],
+            selections[3],
+            selections[4],
+            selections[5],
+            key_message_input,
+            approach_input
         ],
         outputs=None
     ).then(
@@ -2307,18 +2339,18 @@ with gr.Blocks() as demo:
         outputs=[save_gpt4o_btn]
     )
 
-    
+
     save_meta_llama_405b_btn.click(
         fn=prepare_button_text,
         inputs=[],
         outputs=[save_meta_llama_405b_btn]
     ).then(
-        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages, non_personalized_prompt,
-            non_personalized_message, personalization_prompt, gender, generation, psychotype, business_stage, industry, legal_form,
-            key_message, approach:
-            save_to_github(adapted_message, personalized_message, "Meta-Llama-3.1-405B", comment, corrected_message, description, advantages,
-                           non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
-                           business_stage, industry, legal_form, key_message, approach),
+        fn=lambda adapted_message, personalized_message, comment, corrected_message, description, advantages,
+               non_personalized_prompt, non_personalized_message, personalization_prompt, gender, generation, psychotype,
+               business_stage, industry, legal_form, key_message, approach:
+               save_both(adapted_message, personalized_message, "Meta-Llama-3.1-405B", comment, corrected_message, description,
+                         advantages, non_personalized_prompt, non_personalized_message, personalization_prompt, gender,
+                         generation, psychotype, business_stage, industry, legal_form, key_message, approach),
         inputs=[
             adapted_output_text_meta_llama_405b,
             personalized_output_text_meta_llama_405b,
@@ -2329,14 +2361,14 @@ with gr.Blocks() as demo:
             prompt_display,
             output_text_meta_llama_405b,
             personalization_prompt,
-            selections[0],  # Пол
-            selections[1],  # Поколение
-            selections[2],  # Психотип
-            selections[3],  # Стадия бизнеса
-            selections[4],  # Отрасль
-            selections[5],  # ОПФ
-            key_message_input,  # Ключевое сообщение
-            approach_input      # Подход
+            selections[0],
+            selections[1],
+            selections[2],
+            selections[3],
+            selections[4],
+            selections[5],
+            key_message_input,
+            approach_input
         ],
         outputs=None
     ).then(