Update app.py

app.py

@@ -888,6 +888,63 @@ def update_best_example_prompt(*selected_values):
         )
     return best_prompt
 
+def adapt_messages_and_perform_checks(
+    personalized_gigachat_pro,
+    personalized_gigachat_lite,
+    personalized_gigachat_plus,
+    personalized_gpt4o,
+    personalized_meta_llama_405b,
+    key_message,
+    approach,
+    *selected_values
+):
+    adaptation_generator = adapt_messages_to_best_example(
+        personalized_gigachat_pro,
+        personalized_gigachat_lite,
+        personalized_gigachat_plus,
+        personalized_gpt4o,
+        personalized_meta_llama_405b,
+        key_message,
+        approach,
+        *selected_values
+    )
+    last_adaptation_result = None
+    for adaptation_result in adaptation_generator:
+        last_adaptation_result = adaptation_result
+        yield (
+            adaptation_result[0],
+            adaptation_result[1],
+            adaptation_result[2],
+            adaptation_result[3],
+            adaptation_result[4],
+            "", "", "", "", ""  # Пустые строки для проверок
+        )
+    if last_adaptation_result:
+        # Выполнение проверок на адаптированных сообщениях
+        checks_gigachat_pro = perform_checks(last_adaptation_result[0])
+        checks_gigachat_lite = perform_checks(last_adaptation_result[1])
+        checks_gigachat_plus = perform_checks(last_adaptation_result[2])
+        checks_gpt4o = perform_checks(last_adaptation_result[3])
+        checks_meta_llama_405b = perform_checks(last_adaptation_result[4])
+        formatted_checks = [
+            format_checks(checks_gigachat_pro),
+            format_checks(checks_gigachat_lite),
+            format_checks(checks_gigachat_plus),
+            format_checks(checks_gpt4o),
+            format_checks(checks_meta_llama_405b)
+        ]
+        yield (
+            last_adaptation_result[0],
+            last_adaptation_result[1],
+            last_adaptation_result[2],
+            last_adaptation_result[3],
+            last_adaptation_result[4],
+            formatted_checks[0],
+            formatted_checks[1],
+            formatted_checks[2],
+            formatted_checks[3],
+            formatted_checks[4]
+        )
 
 
 # Функция для генерации стандартного промпта
@@ -2550,7 +2607,7 @@ with gr.Blocks() as demo:
     )
 
     best_example_btn.click(
-        fn=adapt_messages_to_best_example,
+        fn=adapt_messages_and_perform_checks,
         inputs=[
             personalized_output_text_gigachat_pro,
             personalized_output_text_gigachat_lite,
@@ -2566,6 +2623,11 @@ with gr.Blocks() as demo:
             adapted_output_text_gigachat_plus,
             adapted_output_text_gpt4o,
             adapted_output_text_meta_llama_405b,
+            validation_display_1,
+            validation_display_2,
+            validation_display_3,
+            validation_display_4,
+            validation_display_5
         ]
     )