Update app.py

app.py

@@ -664,6 +664,66 @@ def clear_unnecessary_fields():
         "", ""  # Дополнительное пустое значение
     )
 
+def regen_message_gpt4o(description, advantages, key_message, approach, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, key_message, approach, *selected_values)
+    gpt4o_message = generate_message_gpt4o_with_retry(standard_prompt)
+    gpt4o_length = len(gpt4o_message)
+    return f"{gpt4o_message}\n\n------\nКоличество знаков: {gpt4o_length}"
+
+def regen_message_gigachat_pro(description, advantages, key_message, approach, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, key_message, approach, *selected_values)
+    gigachat_pro_message = generate_message_gigachat_pro_with_retry(standard_prompt)
+    gigachat_pro_length = len(gigachat_pro_message)
+    return f"{gigachat_pro_message}\n\n------\nКоличество знаков: {gigachat_pro_length}"
+
+def regen_message_gigachat_lite(description, advantages, key_message, approach, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, key_message, approach, *selected_values)
+    gigachat_lite_message = generate_message_gigachat_lite_with_retry(standard_prompt)
+    gigachat_lite_length = len(gigachat_lite_message)
+    return f"{gigachat_lite_message}\n\n------\nКоличество знаков: {gigachat_lite_length}"
+
+def regen_message_gigachat_plus(description, advantages, key_message, approach, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, key_message, approach, *selected_values)
+    gigachat_plus_message = generate_message_gigachat_plus_with_retry(standard_prompt)
+    gigachat_plus_length = len(gigachat_plus_message)
+    return f"{gigachat_plus_message}\n\n------\nКоличество знаков: {gigachat_plus_length}"
+
+def regen_message_meta_llama_405b(description, advantages, key_message, approach, *selected_values):
+    standard_prompt = generate_standard_prompt(description, advantages, key_message, approach, *selected_values)
+    meta_llama_405b_message = generate_message_meta_llama_3_1_405b_with_retry(standard_prompt)
+    meta_llama_405b_length = len(meta_llama_405b_message)
+    return f"{meta_llama_405b_message}\n\n------\nКоличество знаков: {meta_llama_405b_length}"
+
+def personalize_message_gpt4o(gpt4o_message, key_message, approach, *selected_values):
+    personalization_prompt = generate_personalization_prompt(key_message, approach, *selected_values)
+    personalized_message = perform_personalization(gpt4o_message, personalization_prompt)
+    gpt4o_length = len(personalized_message)
+    return f"{personalized_message}\n\n------\nКоличество знаков: {gpt4o_length}"
+
+def personalize_message_gigachat_pro(gigachat_pro_message, key_message, approach, *selected_values):
+    personalization_prompt = generate_personalization_prompt(key_message, approach, *selected_values)
+    personalized_message = perform_personalization_gigachat(gigachat_pro_message, personalization_prompt, "gigachat_pro")
+    gigachat_pro_length = len(personalized_message)
+    return f"{personalized_message}\n\n------\nКоличество знаков: {gigachat_pro_length}"
+
+def personalize_message_gigachat_lite(gigachat_lite_message, key_message, approach, *selected_values):
+    personalization_prompt = generate_personalization_prompt(key_message, approach, *selected_values)
+    personalized_message = perform_personalization_gigachat(gigachat_lite_message, personalization_prompt, "gigachat_lite")
+    gigachat_lite_length = len(personalized_message)
+    return f"{personalized_message}\n\n------\nКоличество знаков: {gigachat_lite_length}"
+
+def personalize_message_gigachat_plus(gigachat_plus_message, key_message, approach, *selected_values):
+    personalization_prompt = generate_personalization_prompt(key_message, approach, *selected_values)
+    personalized_message = perform_personalization_gigachat(gigachat_plus_message, personalization_prompt, "gigachat_plus")
+    gigachat_plus_length = len(personalized_message)
+    return f"{personalized_message}\n\n------\nКоличество знаков: {gigachat_plus_length}"
+
+def personalize_message_meta_llama_405b(meta_llama_405b_message, key_message, approach, *selected_values):
+    personalization_prompt = generate_personalization_prompt(key_message, approach, *selected_values)
+    personalized_message = perform_personalization_meta_llama_405b(meta_llama_405b_message, personalization_prompt)
+    meta_llama_405b_length = len(personalized_message)
+    return f"{personalized_message}\n\n------\nКоличество знаков: {meta_llama_405b_length}"
+
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -723,7 +783,14 @@ with gr.Blocks() as demo:
         personalize_btn = gr.Button("2. Выполнить персонализацию (нажимать только после кнопки 1)", elem_id="personalize_button")
         load_btn = gr.Button("Вернуть параметры предыдущего запроса")
         
-
+    # Ряд кнопок "Перегенерировать"
+    with gr.Row():
+        regen_gpt4o_btn = gr.Button("Перегенерировать GPT-4o")
+        regen_gigachat_pro_btn = gr.Button("Перегенерировать GigaChat-Pro")
+        regen_gigachat_lite_btn = gr.Button("Перегенерировать GigaChat-Lite")
+        regen_gigachat_plus_btn = gr.Button("Перегенерировать GigaChat-Plus")
+        regen_meta_llama_405b_btn = gr.Button("Перегенерировать Meta-Llama-3.1-405B") 
+    
     # Первый ряд: неперсонализированные сообщения
     with gr.Row():
         output_text_gpt4o = gr.Textbox(label="Неперсонализированное сообщение GPT-4o", lines=3, interactive=False)
@@ -731,6 +798,14 @@ with gr.Blocks() as demo:
         output_text_gigachat_lite = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Lite", lines=3, interactive=False)
         output_text_gigachat_plus = gr.Textbox(label="Неперсонализированное сообщение GigaChat-Plus", lines=3, interactive=False)
         output_text_meta_llama_405b = gr.Textbox(label="Неперсонализированное сообщение Meta-Llama-3.1-405B", lines=3, interactive=False)
+
+    # Ряд кнопок "Персонализировать"
+    with gr.Row():
+        personalize_gpt4o_btn = gr.Button("Персонализировать")
+        personalize_gigachat_pro_btn = gr.Button("Персонализировать")
+        personalize_gigachat_lite_btn = gr.Button("Персонализировать")
+        personalize_gigachat_plus_btn = gr.Button("Персонализировать")
+        personalize_meta_llama_405b_btn = gr.Button("Персонализировать")
     
     # Второй ряд: персонализированные сообщения
     with gr.Row():
@@ -888,6 +963,65 @@ with gr.Blocks() as demo:
         ]
     )
 
+    regen_gpt4o_btn.click(
+        regen_message_gpt4o,
+        inputs=[description_input, advantages_input, key_message_input, approach_input] + selections,
+        outputs=output_text_gpt4o
+    )
+    
+    regen_gigachat_pro_btn.click(
+        regen_message_gigachat_pro,
+        inputs=[description_input, advantages_input, key_message_input, approach_input] + selections,
+        outputs=output_text_gigachat_pro
+    )
+    
+    regen_gigachat_lite_btn.click(
+        regen_message_gigachat_lite,
+        inputs=[description_input, advantages_input, key_message_input, approach_input] + selections,
+        outputs=output_text_gigachat_lite
+    )
+    
+    regen_gigachat_plus_btn.click(
+        regen_message_gigachat_plus,
+        inputs=[description_input, advantages_input, key_message_input, approach_input] + selections,
+        outputs=output_text_gigachat_plus
+    )
+    
+    regen_meta_llama_405b_btn.click(
+        regen_message_meta_llama_405b,
+        inputs=[description_input, advantages_input, key_message_input, approach_input] + selections,
+        outputs=output_text_meta_llama_405b
+    )
+    
+    personalize_gpt4o_btn.click(
+        personalize_message_gpt4o,
+        inputs=[output_text_gpt4o, key_message_input, approach_input] + selections,
+        outputs=personalized_output_text_gpt4o
+    )
+    
+    personalize_gigachat_pro_btn.click(
+        personalize_message_gigachat_pro,
+        inputs=[output_text_gigachat_pro, key_message_input, approach_input] + selections,
+        outputs=personalized_output_text_gigachat_pro
+    )
+    
+    personalize_gigachat_lite_btn.click(
+        personalize_message_gigachat_lite,
+        inputs=[output_text_gigachat_lite, key_message_input, approach_input] + selections,
+        outputs=personalized_output_text_gigachat_lite
+    )
+    
+    personalize_gigachat_plus_btn.click(
+        personalize_message_gigachat_plus,
+        inputs=[output_text_gigachat_plus, key_message_input, approach_input] + selections,
+        outputs=personalized_output_text_gigachat_plus
+    )
+    
+    personalize_meta_llama_405b_btn.click(
+        personalize_message_meta_llama_405b,
+        inputs=[output_text_meta_llama_405b, key_message_input, approach_input] + selections,
+        outputs=personalized_output_text_meta_llama_405b
+    )
 
     # Привязка кнопок к функциям сохранения
     save_gpt4o_btn.click(