Update app.py

app.py

@@ -660,10 +660,7 @@ def clean_prompt_for_display(prompt, prefixes, suffixes):
 
 # Функция для постепенной генерации всех сообщений через yield
 def generate_all_messages(desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf):
-    standard_prompt = generate_standard_prompt(desc, benefits, key_message)
-    yield standard_prompt, None, None, None
-
-    # Сопоставление названий подходов с префиксами и суффиксами
+    # Определение подходов
     approach_mapping = {
         "Указание на пользу": (
             "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
@@ -678,40 +675,34 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
             "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
         )
     }
-
-    # Получаем DataFrame подходов
     approaches_df = features.get("Подход", None)
-
     if approaches_df is None:
         gr.Warning("Таблица 'Подход' не найдена.")
         return
-
-    # Создаем фильтры на основе выбранных параметров
     filters = []
     for param_name, param_value in [('Пол', gender), ('Поколение', generation), ('Психотип', psychotype)]:
         if param_value:
             filters.append(approaches_df[param_name].fillna('') == param_value)
         else:
             filters.append(approaches_df[param_name].isnull() | (approaches_df[param_name] == ''))
-
-    # Применяем комбинированный фильтр
     combined_filter = filters[0]
     for f in filters[1:]:
         combined_filter &= f
-
     matching_rows = approaches_df[combined_filter]
-
     if matching_rows.empty:
         gr.Warning("Подход не найден для выбранных параметров.")
         return
-
-    # Извлекаем уникальные подходы из отфильтрованных строк
     approach_set = set()
     for approaches in matching_rows['Подход']:
         approach_names = [a.strip() for a in approaches.split(',')]
         approach_set.update(approach_names)
-
-    # Генерируем сообщения для каждого подхода
+    selected_approaches_text_content = "Выбранные подходы согласно рекомендациям Нейролаборатории: " + ', '.join(approach_set)
+    
+    # Генерация стандартного задания
+    standard_prompt = generate_standard_prompt(desc, benefits, key_message)
+    yield selected_approaches_text_content, standard_prompt, None, None, None
+    
+    # Инициализация переменных для сообщений
     non_personalized_messages = ''
     personalized_messages = ''
     flag = 1
@@ -726,7 +717,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         )
         display_personalization_prompt = clean_prompt_for_display(personalization_prompt, [current_prefix], [current_suffix])
         if flag == 1:
-            yield standard_prompt, display_personalization_prompt, None, None
+            yield selected_approaches_text_content, standard_prompt, display_personalization_prompt, None, None
             flag += 1
         prompt = add_prefix_suffix(standard_prompt, current_prefix, current_suffix)
         non_personalized_message = generate_message_mistral_with_retry(prompt, approach_name, desc, key_message)
@@ -737,7 +728,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         else:
             non_personalized_messages = non_personalized_display
         yield (
-            standard_prompt, display_personalization_prompt,
+            selected_approaches_text_content, standard_prompt, display_personalization_prompt,
             non_personalized_messages, personalized_messages
         )
         full_personalized_prompt = f"{personalization_prompt}\n\nТекст для адаптации: {non_personalized_message}"
@@ -749,7 +740,7 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         else:
             personalized_messages = personalized_display
         yield (
-            standard_prompt, display_personalization_prompt,
+            selected_approaches_text_content, standard_prompt, display_personalization_prompt,
             non_personalized_messages, personalized_messages
         )
         time.sleep(1)
@@ -1641,14 +1632,16 @@ with gr.Blocks() as demo:
             
         # Вкладка 2: Промпты
         with gr.TabItem("Ассистент", id=1):
+            selected_approaches_text = gr.Markdown()
             with gr.Row():
                 with gr.Column():
                     non_personalized_prompt = gr.Textbox(
-                        label="Задание для копирайтера", 
-                        lines=25,
+                        label="Задание для копирайтера",
+                        lines=23,
                         interactive=False)
                 with gr.Column():
-                    personalized_prompt = gr.Textbox(label="Задание для редактора", lines=25)  # Увеличенная высота
+                    personalized_prompt = gr.Textbox(label="Задание для редактора", lines=23)
+
                     
         # Вкладка 3: Сообщения
         with gr.TabItem("Сообщения", id=2):
@@ -1662,15 +1655,16 @@ with gr.Blocks() as demo:
 
             # Сначала переключаем вкладку, потом запускаем генерацию сообщений
             btn_to_prompts.click(
-                fn=change_tab, 
-                inputs=[gr.Number(value=1, visible=False)],  # Переключение на вкладку "Ассистент" (id=1)
-                outputs=tabs  # Обновляем вкладку
+                fn=change_tab,
+                inputs=[gr.Number(value=1, visible=False)],
+                outputs=tabs
             ).then(
-                fn=generate_all_messages, 
-                inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf],  # Входные текстовые поля
+                fn=generate_all_messages,
+                inputs=[desc, benefits, key_message, gender, generation, psychotype, business_stage, industry, opf],
                 outputs=[
-                    non_personalized_prompt, personalized_prompt,  # Поля для задания копирайтера и редактора (на вкладке "Ассистент")
-                    non_personalized_messages, personalized_messages  # Сообщения на вкладке "Сообщения"
+                    selected_approaches_text,
+                    non_personalized_prompt, personalized_prompt,
+                    non_personalized_messages, personalized_messages
                 ]
             )