Update app.py

app.py

@@ -115,39 +115,37 @@ def generate_message_gigachat_plus(prompt):
     except Exception as e:
         return f"Ошибка при обращении к GigaChat-Lite+: {e}"
 
-# Функция для обработки нажатия кнопки Submit и сохранения данных в State
+# Функция для обработки нажатия кнопки Submit и последовательного отображения результатов
 def generate_messages(description, advantages, *selected_values):
     standard_prompt = generate_standard_prompt(description, advantages, *selected_values)
 
     results = {
         "prompt": standard_prompt,
-        "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)
+        "gpt4o": None,
+        "gigachat_pro": None,
+        "gigachat_lite": None,
+        "gigachat_plus": None
     }
 
-    return (
-        results["prompt"],
-        results["gpt4o"],
-        results["gigachat_pro"],
-        results["gigachat_lite"],
-        results["gigachat_plus"],
-        results  # Возвращаем данные для сохранения в state
-    )
+    yield results["prompt"], "", "", "", "", selected_values[0]
+
+    results["gpt4o"] = generate_message_gpt4o(standard_prompt)
+    yield results["prompt"], results["gpt4o"], "", "", "", selected_values[0]
+
+    results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", "", selected_values[0]
+
+    results["gigachat_lite"] = generate_message_gigachat_lite(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], "", selected_values[0]
+
+    results["gigachat_plus"] = generate_message_gigachat_plus(standard_prompt)
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"], selected_values[0]
+
+    return results
 
 # Функция для выполнения персонализации
-def personalize_messages(results, *selected_values):
-    personalized_texts = []
-    
-    for i, feature in enumerate(features.keys()):
-        if selected_values[i]:
-            try:
-                personalized_texts.append(f"{features[feature][selected_values[i]]}")
-            except KeyError:
-                personalized_texts.append(f"Ошибка: выбранное значение {selected_values[i]} не найдено в данных.")
-    
-    return personalized_texts
+def personalize_gender(selected_gender):
+    return f"Выбранный пол: {selected_gender}"
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -191,14 +189,12 @@ with gr.Blocks() as demo:
             output_text_gigachat_pro = gr.Textbox(label="Создание сообщения GigaChat-Pro", lines=3)
             output_text_gigachat_lite = gr.Textbox(label="Создание сообщения GigaChat-Lite", lines=3)
             output_text_gigachat_plus = gr.Textbox(label="Создание сообщения GigaChat-Lite+", lines=3)
-            status_display = gr.Textbox(label="Статус выполнения", lines=4, interactive=False)  # Новый блок для отображения статуса
-            
-            state = gr.State()  # Переменная состояния для сохранения данных между функциями
+            selected_gender_display = gr.Textbox(label="Выбранный пол", lines=1, interactive=False)  # Блок для отображения выбранного пола
             
             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, state]
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, selected_gender_display]
             )
 
     with gr.Row():
@@ -206,9 +202,9 @@ with gr.Blocks() as demo:
         
     with gr.Row():
         personalize_btn.click(
-            personalize_messages,
-            inputs=selections + [state],
-            outputs=[gr.Textbox(label=f"Персонализированный текст для {feature}", lines=3) for feature in features.keys()]
+            personalize_gender,
+            inputs=[selected_gender_display],
+            outputs=[gr.Textbox(label="Персонализированный результат", lines=3)]
         )
 
     demo.launch()