Update app.py

app.py

@@ -552,7 +552,7 @@ def generate_error_check_prompt():
     return prompt
 
 
-def save_to_github(personalized_message, model_name, comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach):
+def save_to_github(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):
     # Собираем все данные в один словарь
     data_to_save = {
         "Модель": model_name,
@@ -565,6 +565,7 @@ def save_to_github(personalized_message, model_name, comment, corrected_message,
         "Подход": approach,  # Save approach
         "Неперсонализированный промпт": non_personalized_prompt,
         "Неперсонализированное сообщение": non_personalized_message,
+        "Персонализированный промпт": personalization_prompt,  # Добавляем персонализированный промпт
         "Пол": gender,
         "Поколение": generation,
         "Психотип": psychotype,
@@ -782,8 +783,6 @@ with gr.Blocks() as demo:
         submit_btn = gr.Button("1. Создать неперсонализированное сообщение")
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
         load_btn = gr.Button("Вернуть параметры предыдущего запроса")
-
-    gr.Markdown("---")  # Добавляет горизонтальную линию
         
     # Ряд кнопок "Перегенерировать"
     with gr.Row():
@@ -1031,8 +1030,8 @@ with gr.Blocks() as demo:
         inputs=[],
         outputs=[save_gpt4o_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, key_message, approach: 
-            save_to_github(personalized_message, "GPT-4o", comment, corrected_message, description, advantages, non_personalized_prompt, non_personalized_message, gender, generation, psychotype, business_stage, industry, legal_form, key_message, approach),
+        fn=lambda 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(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=[
             personalized_output_text_gpt4o,
             comment_gpt4o,
@@ -1041,6 +1040,7 @@ with gr.Blocks() as demo:
             advantages_input,
             prompt_display,
             output_text_gpt4o,
+            personalization_prompt,
             selections[0],  # Пол
             selections[1],  # Поколение
             selections[2],  # Психотип
@@ -1066,8 +1066,8 @@ with gr.Blocks() as demo:
         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, key_message, approach:
-            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, key_message, approach),
+        fn=lambda 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(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=[
             personalized_output_text_gigachat_pro,
             comment_gigachat_pro,
@@ -1076,6 +1076,7 @@ with gr.Blocks() as demo:
             advantages_input,
             prompt_display,
             output_text_gigachat_pro,
+            personalization_prompt,
             selections[0],  # Пол
             selections[1],  # Поколение
             selections[2],  # Психотип
@@ -1100,8 +1101,8 @@ with gr.Blocks() as demo:
         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, key_message, approach:
-            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, key_message, approach),
+        fn=lambda 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(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=[
             personalized_output_text_gigachat_lite,
             comment_gigachat_lite,
@@ -1110,6 +1111,7 @@ with gr.Blocks() as demo:
             advantages_input,
             prompt_display,
             output_text_gigachat_lite,
+            personalization_prompt,
             selections[0],  # Пол
             selections[1],  # Поколение
             selections[2],  # Психотип
@@ -1134,8 +1136,8 @@ with gr.Blocks() as demo:
         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, key_message, approach:
-            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, key_message, approach),
+        fn=lambda 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(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=[
             personalized_output_text_gigachat_plus,
             comment_gigachat_plus,
@@ -1144,6 +1146,7 @@ with gr.Blocks() as demo:
             advantages_input,
             prompt_display,
             output_text_gigachat_plus,
+            personalization_prompt,
             selections[0],  # Пол
             selections[1],  # Поколение
             selections[2],  # Психотип
@@ -1167,8 +1170,8 @@ with gr.Blocks() as demo:
         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, key_message, approach:
-            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, key_message, approach),
+        fn=lambda 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(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=[
             personalized_output_text_meta_llama_405b,
             comment_meta_llama_405b,
@@ -1177,6 +1180,7 @@ with gr.Blocks() as demo:
             advantages_input,
             prompt_display,
             output_text_meta_llama_405b,
+            personalization_prompt,
             selections[0],  # Пол
             selections[1],  # Поколение
             selections[2],  # Психотип