Update app.py

app.py

@@ -115,37 +115,29 @@ def generate_message_gigachat_plus(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite+: {e}"
 
-# Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
+# Функция для обработки нажатия кнопки Submit и сохранения данных в State
 def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
         "prompt": standard_prompt,
-        "gpt4o": None,
-        "gigachat_pro": None,
-        "gigachat_lite": None,
-        "gigachat_plus": None
+        "gpt4o": generate_message_gpt4o(standard_prompt),
+        "gigachat_pro": generate_message_gigachat_pro(standard_prompt),
+        "gigachat_lite": generate_message_gigachat_lite(standard_prompt),
+        "gigachat_plus": generate_message_gigachat_plus(standard_prompt)
     }
 
-    yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
-
-    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
-    yield results["prompt"], results["gpt4o"], "", "", "", "Сообщение GPT-4o сгенерировано"
-
-    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", "", "Сообщение GigaChat-Pro сгенерировано"
-
-    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], "", "Сообщение GigaChat-Lite сгенерировано"
-
-    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
+    return (
+        results["prompt"],
+        results["gpt4o"],
+        results["gigachat_pro"],
+        results["gigachat_lite"],
+        results["gigachat_plus"],
+        results  # Возвращаем данные для сохранения в state
+    )
 
 # Функция для выполнения персонализации
-def personalize_messages(*selected_values):
+def personalize_messages(results, *selected_values):
     personalized_texts = []
     
     for i, feature in enumerate(features.keys()):
@@ -201,10 +193,12 @@ with gr.Blocks() as demo:
             output_text_gigachat_plus = gr.Textbox(label="Создание сообщения GigaChat-Lite+", lines=3)
             status_display = gr.Textbox(label="Статус выполнения", lines=4, interactive=False)  # Новый блок для отображения статуса
             
+            state = gr.State()  # Переменная состояния для сохранения данных между функциями
+            
             submit_btn.click(
                 generate_messages, 
                 inputs=[description_input, advantages_input] + selections, 
-                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, state]
             )
 
     with gr.Row():
@@ -213,7 +207,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         personalize_btn.click(
             personalize_messages,
-            inputs=selections,
+            inputs=selections + [state],
             outputs=[gr.Textbox(label=f"Персонализированный текст для {feature}", lines=3) for feature in features.keys()]
         )