Update app.py

app.py

@@ -171,21 +171,28 @@ def perform_personalization_gigachat(standard_message, personalization_prompt, m
     elif model == "gigachat_plus":
         return generate_message_gigachat_plus(full_prompt)
 
-# Функция для выполнения персонализации с использованием yield
-def personalize_messages_with_yield(standard_message, *selected_values):
+# Глобальные переменные для персонализированных сообщений
+personalized_message_gpt4o = None
+personalized_message_gigachat_pro = None
+personalized_message_gigachat_lite = None
+personalized_message_gigachat_plus = None
+
+def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigachat_lite_message, gigachat_plus_message, *selected_values):
+    global personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus
+    
     personalization_prompt = generate_personalization_prompt(*selected_values)
     yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
 
-    personalized_message_gpt4o = perform_personalization(standard_message, personalization_prompt)
+    personalized_message_gpt4o = perform_personalization(gpt4o_message, personalization_prompt)
     yield personalization_prompt, personalized_message_gpt4o, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
 
-    personalized_message_gigachat_pro = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_pro")
+    personalized_message_gigachat_pro = perform_personalization_gigachat(gigachat_pro_message, personalization_prompt, "gigachat_pro")
     yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, "", "", "Персонализированное сообщение GigaChat-Pro сгенерировано"
 
-    personalized_message_gigachat_lite = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_lite")
+    personalized_message_gigachat_lite = perform_personalization_gigachat(gigachat_lite_message, personalization_prompt, "gigachat_lite")
     yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, "", "Персонализированное сообщение GigaChat-Lite сгенерировано"
 
-    personalized_message_gigachat_plus = perform_personalization_gigachat(standard_message, personalization_prompt, "gigachat_plus")
+    personalized_message_gigachat_plus = perform_personalization_gigachat(gigachat_plus_message, personalization_prompt, "gigachat_plus")
     yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
 
 # Функция для генерации промпта проверки текста
@@ -318,7 +325,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         personalize_btn.click(
             personalize_messages_with_yield,
-            inputs=[output_text_gpt4o] + selections,
+            inputs=[output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus] + selections,
             outputs=[
                 gr.Textbox(label="Промпт для персонализации", lines=6, interactive=False),
                 gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6),
@@ -328,6 +335,7 @@ with gr.Blocks() as demo:
                 status_display
             ]
         )
+
     
     with gr.Row():
         check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2)", elem_id="check_errors_button")
@@ -335,7 +343,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         check_errors_btn.click(
             check_errors_with_yield,
-            inputs=[output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus],
+            inputs=[personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus],
             outputs=[
                 gr.Textbox(label="Промпт для проверки текста", lines=6, interactive=False),
                 gr.Textbox(label="Результат проверки GPT-4o", lines=6),
@@ -346,4 +354,5 @@ with gr.Blocks() as demo:
             ]
         )
 
+
     demo.launch()