Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import pandas as pd
 import time
 from langchain.schema import SystemMessage
-from langchain_community.chat_models.gigachat import GigaChat
+#from langchain_community.chat_models.gigachat import GigaChat
 from openpyxl import load_workbook
 import plotly.graph_objects as go
 import random
@@ -661,7 +661,6 @@ 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)
-    # Определение подходов
     approach_mapping = {
         "Указание на пользу": (
             "Начни сообщение с указания на пользу продукта. Используй глагол в побудительном наклонении.",
@@ -693,19 +692,31 @@ def generate_all_messages(desc, benefits, key_message, gender, generation, psych
     if matching_rows.empty:
         gr.Warning("Подход не найден для выбранных параметров.")
         return
-    approach_set = set()
+    approach_list = []
     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)
-    
-    # Генерация стандартного задания
+        approach_list.extend(approach_names)
+    approach_list = list(set(approach_list))  # Убираем дубликаты
+    selected_approaches_text_content = ', '.join(approach_list)
+
     yield selected_approaches_text_content, standard_prompt, None, None, None
-    
+
     non_personalized_messages = ''
     personalized_messages = ''
     flag = 1
-    for approach_name in approach_set:
+
+    # Подготовим список подходов для генерации 3 сообщений
+    if len(approach_list) == 0:
+        gr.Warning("Не выбрано ни одного подхода.")
+        return
+    elif len(approach_list) == 1:
+        approaches_for_messages = [approach_list[0]] * 3
+    elif len(approach_list) >= 2:
+        approaches_for_messages = [approach_list[0], approach_list[1], approach_list[0]]
+    else:
+        approaches_for_messages = approach_list[:3]
+
+    for approach_name in approaches_for_messages:
         if approach_name not in approach_mapping:
             gr.Warning(f"Неизвестный подход: {approach_name}")
             continue