Update app.py

app.py

@@ -68,6 +68,8 @@ for sheet_name, df in data.items():
     try:
         if sheet_name == "Пол Поколение Психотип":
             features[sheet_name] = df.set_index(['Пол', 'Поколение', 'Психотип'])['Инструкция'].to_dict()
+        elif sheet_name == "Подход":
+            features[sheet_name] = df.set_index(['Пол', 'Поколение', 'Психотип'])['Подход'].to_dict()
         else:
             features[sheet_name] = df.set_index(df.columns[0]).to_dict()[df.columns[1]]
     except Exception as e:
@@ -657,34 +659,59 @@ def clean_prompt_for_display(prompt, prefixes, suffixes):
 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
-    prefixes = [
-        "Начни сообщение с призыва к действию с продуктом.",
-        "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
-        "Начни сообщение с вопроса, который указывает на пользу продукта для клиента."
-    ]
-    suffixes = [
-        "Убедись, что готовый текст начинается с призыва к действию с продуктом.",
-        "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении.",
-        "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
-    ]
+
+    # Map approach names to prefixes and suffixes
+    approach_mapping = {
+        "Указание на пользу": (
+            "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
+            "Убедись, что готовый текст начинается с указания на пользу продукта и использования глагола в побудительном наклонении."
+        ),
+        "Призыв к действию": (
+            "Начни сообщение с призыва к действию с продуктом.",
+            "Убедись, что готовый текст начинается с призыва к действию с продуктом."
+        ),
+        "Вопрос": (
+            "Начни сообщение с вопроса, который указывает на пользу продукта для клиента.",
+            "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента."
+        )
+    }
+
+    # Get the approaches from the "Подход" sheet
+    approaches_sheet = features.get("Подход", {})
+    key = (gender, generation, psychotype)
+
+    if not gender or not generation or not psychotype:
+        gr.Warning("Необходимо выбрать Пол, Поколение и Психотип для определения подхода.")
+        return
+
+    approach_list_str = approaches_sheet.get(key, "")
+    if not approach_list_str:
+        gr.Warning("Подход не найден для выбранных параметров.")
+        return
+
+    approach_list = [a.strip() for a in approach_list_str.split(",")]
+
+    # Now generate messages for the specified approaches
     non_personalized_messages = ''
     personalized_messages = ''
     flag = 1
-    for i in range(3):
-        current_prefix = prefixes[i]
+    for approach_name in approach_list:
+        if approach_name not in approach_mapping:
+            gr.Warning(f"Неизвестный подход: {approach_name}")
+            continue
+        current_prefix, current_suffix = approach_mapping[approach_name]
         personalization_prompt = generate_personalization_prompt(
             key_message, gender, generation, psychotype, business_stage, industry, opf,
-            prefix=prefixes[i], suffix=suffixes[i]
+            prefix=current_prefix, suffix=current_suffix
         )
-        display_personalization_prompt = clean_prompt_for_display(personalization_prompt, prefixes, suffixes)
+        display_personalization_prompt = clean_prompt_for_display(personalization_prompt, [current_prefix], [current_suffix])
         if flag == 1:
             yield standard_prompt, display_personalization_prompt, None, None
             flag += 1
-        prompt = add_prefix_suffix(standard_prompt, prefixes[i], suffixes[i])
+        prompt = add_prefix_suffix(standard_prompt, current_prefix, current_suffix)
         non_personalized_message = generate_message_mistral_with_retry(prompt, current_prefix, desc, key_message)
         non_personalized_length = len(non_personalized_message)
         non_personalized_display = f"{non_personalized_message}\n------\nКоличество знаков: {non_personalized_length}"
-        # Добавляем новое сообщение с двумя пустыми строками
         if non_personalized_messages:
             non_personalized_messages += '\n\n' + non_personalized_display
         else:
@@ -697,7 +724,6 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
         personalized_message = generate_message_mistral_with_retry(full_personalized_prompt, current_prefix, desc, key_message)
         personalized_length = len(personalized_message)
         personalized_display = f"{personalized_message}\n------\nКоличество знаков: {personalized_length}"
-        # Добавляем новое сообщение с двумя пустыми строками
         if personalized_messages:
             personalized_messages += '\n\n' + personalized_display
         else: