Update app.py

app.py

@@ -127,31 +127,33 @@ def generate_messages(description, advantages, *selected_values):
         "gigachat_plus": None
     }
 
-    yield results["prompt"], "", "", "", "", selected_values[0]
+    yield results["prompt"], "", "", "", "", "Генерация стандартного промпта завершена"
 
     results["gpt4o"] = generate_message_gpt4o(standard_prompt)
-    yield results["prompt"], results["gpt4o"], "", "", "", selected_values[0]
+    yield results["prompt"], results["gpt4o"], "", "", "", "Сообщение GPT-4o сгенерировано"
 
     results["gigachat_pro"] = generate_message_gigachat_pro(standard_prompt)
-    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], "", "", selected_values[0]
+    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"], "", selected_values[0]
+    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"], selected_values[0]
+    yield results["prompt"], results["gpt4o"], results["gigachat_pro"], results["gigachat_lite"], results["gigachat_plus"], "Все сообщения сгенерированы"
 
+    # Возвращаем результаты для дальнейшей персонализации
     return results
 
 # Функция для выполнения персонализации
-def personalize_gender(selected_gender):
-    try:
-        if selected_gender in features["Пол"]:
-            return f"Текст для пола '{selected_gender}': {features['Пол'][selected_gender]}"
+def personalize_all_texts(*selected_values):
+    personalized_texts = []
+    for i, feature in enumerate(features.keys()):
+        if selected_values[i]:
+            text = features[feature].get(selected_values[i], "Текст не найден")
+            personalized_texts.append(f"Персонализированный текст для {feature}: {text}")
         else:
-            return "Не найден текст для выбранного пола."
-    except Exception as e:
-        return f"Ошибка при обработке данных: {e}"
+            personalized_texts.append(f"Персонализированный текст для {feature}: не выбран")
+    return personalized_texts
 
 # Создание интерфейса Gradio
 with gr.Blocks() as demo:
@@ -195,12 +197,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)
-            selected_gender_display = gr.Textbox(label="Выбранный пол", lines=1, interactive=False)  # Блок для отображения выбранного пола
+            status_display = gr.Textbox(label="Статус выполнения", lines=4, 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, selected_gender_display]
+                outputs=[prompt_display, output_text_gpt4o, output_text_gigachat_pro, output_text_gigachat_lite, output_text_gigachat_plus, status_display]
             )
 
     with gr.Row():
@@ -208,9 +210,9 @@ with gr.Blocks() as demo:
         
     with gr.Row():
         personalize_btn.click(
-            personalize_gender,
-            inputs=[selected_gender_display],
-            outputs=[gr.Textbox(label="Персонализированный текст для Пола", lines=3)]
+            personalize_all_texts,
+            inputs=selections,
+            outputs=[gr.Textbox(label=f"Персонализированный текст для {feature}", lines=3) for feature in features.keys()]
         )
 
     demo.launch()