Update app.py

app.py

@@ -171,15 +171,8 @@ def perform_personalization_gigachat(standard_message, personalization_prompt, m
     elif model == "gigachat_plus":
         return generate_message_gigachat_plus(full_prompt)
 
-# Глобальные переменные для персонализированных сообщений
-personalized_message_gpt4o = None
-personalized_message_gigachat_pro = None
-personalized_message_gigachat_lite = None
-personalized_message_gigachat_plus = None
-
+# Функция для выполнения персонализации с использованием yield
 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, "", "", "", "", "Промпт для персонализации сгенерирован"
 
@@ -195,6 +188,9 @@ def personalize_messages_with_yield(gpt4o_message, gigachat_pro_message, gigacha
     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, "Все персонализированные сообщения сгенерированы"
 
+    # Возвращаем результаты персонализации, чтобы их можно было использовать в check_errors_with_yield
+    return personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus
+
 # Функция для генерации промпта проверки текста
 def generate_error_check_prompt():
     prompt = (
@@ -239,7 +235,6 @@ def generate_error_check_prompt():
     )
     return prompt
 
-
 # Функция для выполнения проверки текста с использованием yield
 def check_errors_with_yield(*personalized_messages):
     if len(personalized_messages) < 4:
@@ -267,7 +262,6 @@ def check_errors_with_yield(*personalized_messages):
         
     yield error_check_prompt, error_message_gpt4o, error_message_gigachat_pro, error_message_gigachat_lite, error_message_gigachat_plus, "Все результаты проверки сгенерированы"
 
-
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -336,7 +330,6 @@ with gr.Blocks() as demo:
             ]
         )
 
-    
     with gr.Row():
         check_errors_btn = gr.Button("3. Проверить текст (нажимать только после кнопки 2)", elem_id="check_errors_button")