Update app.py

app.py

@@ -141,7 +141,6 @@ def generate_messages(description, advantages, *selected_values):
     results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
     yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"], "Все сообщения сгенерированы"
 
-    # Возвращаем результаты для дальнейшей персонализации
     return results
 
 # Функция для генерации персонализированного промпта
@@ -169,6 +168,23 @@ 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):
+    personalization_prompt = generate_personalization_prompt(*selected_values)
+    yield personalization_prompt, "", "", "", "", "Промпт для персонализации сгенерирован"
+
+    personalized_message_gpt4o = perform_personalization(standard_message, personalization_prompt)
+    yield personalization_prompt, personalized_message_gpt4o, "", "", "", "Персонализированное сообщение GPT-4o сгенерировано"
+
+    personalized_message_gigachat_pro = perform_personalization_gigachat(standard_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")
+    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")
+    yield personalization_prompt, personalized_message_gpt4o, personalized_message_gigachat_pro, personalized_message_gigachat_lite, personalized_message_gigachat_plus, "Все персонализированные сообщения сгенерированы"
+
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Генерация SMS-сообщений по заданным признакам")
@@ -224,20 +240,15 @@ with gr.Blocks() as demo:
         
     with gr.Row():
         personalize_btn.click(
-            lambda standard_message, *selected_values: (
-                generate_personalization_prompt(*selected_values),
-                perform_personalization(standard_message, generate_personalization_prompt(*selected_values)),
-                perform_personalization_gigachat(standard_message, generate_personalization_prompt(*selected_values), "gigachat_pro"),
-                perform_personalization_gigachat(standard_message, generate_personalization_prompt(*selected_values), "gigachat_lite"),
-                perform_personalization_gigachat(standard_message, generate_personalization_prompt(*selected_values), "gigachat_plus")
-            ),
+            personalize_messages_with_yield,
             inputs=[output_text_gpt4o] + selections,
             outputs=[
                 gr.Textbox(label="Промпт для персонализации", lines=6, interactive=False),
                 gr.Textbox(label="Персонализированное сообщение GPT-4o", lines=6),
                 gr.Textbox(label="Персонализированное сообщение GigaChat-Pro", lines=6),
                 gr.Textbox(label="Персонализированное сообщение GigaChat-Lite", lines=6),
-                gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=6)
+                gr.Textbox(label="Персонализированное сообщение GigaChat-Lite+", lines=6),
+                status_display
             ]
         )