Update app.py

app.py

@@ -7,22 +7,43 @@ def load_dropdown_data(file_path, sheet_name, column_name):
     data = pd.read_excel(file_path, sheet_name=sheet_name)
     return data[column_name].dropna().unique().tolist()
 
-# Путь к файлу
 file_path = "Исходные данные.xlsx"
 
 # Загрузка данных
 products, data_products = load_dropdown_data(file_path, "Продукты", "Наименование продукта"), pd.read_excel(file_path, sheet_name="Продукты")
-genders = load_dropdown_data(file_path, "Пол", "Пол")
-generations = load_dropdown_data(file_path, "Поколение", "Поколение")
-psychotypes = load_dropdown_data(file_path, "Психотип", "Психотип")
-business_stages = load_dropdown_data(file_path, "Стадия бизнеса", "Стадия бизнеса")
-industries = load_dropdown_data(file_path, "Отрасль", "Отрасль")
-opfs = load_dropdown_data(file_path, "ОПФ", "ОПФ")
-
-# Загрузка данных для "Подход"
+genders_data = pd.read_excel(file_path, sheet_name="Пол")
+generations_data = pd.read_excel(file_path, sheet_name="Поколение")
+psychotypes_data = pd.read_excel(file_path, sheet_name="Психотип")
+business_stages_data = pd.read_excel(file_path, sheet_name="Стадия бизнеса")
+industries_data = pd.read_excel(file_path, sheet_name="Отрасль")
+opfs_data = pd.read_excel(file_path, sheet_name="ОПФ")
+
+genders = genders_data["Пол"].dropna().unique().tolist()
+generations = generations_data["Поколение"].dropna().unique().tolist()
+psychotypes = psychotypes_data["Психотип"].dropna().unique().tolist()
+business_stages = business_stages_data["Стадия бизнеса"].dropna().unique().tolist()
+industries = industries_data["Отрасль"].dropna().unique().tolist()
+opfs = opfs_data["ОПФ"].dropna().unique().tolist()
+
 approaches_data = pd.read_excel(file_path, sheet_name="Подход")
 
-# Функция для заполнения полей на основе выбранного продукта
+# Словарь подходов
+approach_dict = {
+    "Указание на пользу": {
+        "prefix": "Начни SMS с указания на пользу продукта. Используй глагол в побудительном наклонении. Не начинай с вопроса",
+        "suffix": "Убедись, что SMS начинается с указания на пользу продукта и использования глагола в побудительном наклонении и не начинается с вопроса"
+    },
+    "Вопрос": {
+        "prefix": "Начни сообщение с вопроса, который указывает на пользу продукта для клиента",
+        "suffix": "Убедись, что готовый текст начинается с вопроса, который указывает на пользу продукта для клиента"
+    },
+    "Призыв к действию": {
+        "prefix": "Начни SMS с призыва к действию с продуктом. Не начинай с вопроса",
+        "suffix": "Убедись, что готовый текст начинается с призыва к действию с продуктом и не начинается с вопроса"
+    }
+}
+
+# Функция для заполнения полей продукта
 def fill_product_details(selected_product, data):
     if selected_product:
         product_row = data[data["Наименование продукта"] == selected_product].iloc[0]
@@ -34,7 +55,6 @@ def fill_product_details(selected_product, data):
         )
     return "", "", "", ""
 
-# Функция для получения списка подходов
 def get_approaches(gender, generation, psychotype, approaches_df):
     if approaches_df is None or approaches_df.empty:
         return "Таблица 'Подход' не найдена."
@@ -65,15 +85,122 @@ def get_approaches(gender, generation, psychotype, approaches_df):
     selected_approaches_text_content = ', '.join(approach_list)
     return selected_approaches_text_content
 
-# Функция-обёртка для обновления подхода
 def update_approach(gender, generation, psychotype):
     return get_approaches(gender, generation, psychotype, approaches_data)
 
-# Создание интерфейса
+# Функция для извлечения инструкций из таблиц параметров
+def get_instructions_for_param(param_name, param_value, df, param_col_name):
+    # Если param_value пустое или "Не выбрано" – игнорируем
+    if not param_value or param_value == "Не выбрано":
+        return None, None
+    row = df[df[param_col_name] == param_value]
+    if row.empty:
+        return None, None
+    instr1 = row.iloc[0].get("Инструкция 1", "")
+    instr2 = row.iloc[0].get("Ин��трукция 2", "")
+    return instr1, instr2
+
+def format_instruction_string(instr):
+    # Превращаем "слово1, слово2, слово3" в "слово1 или слово2 или слово3"
+    terms = [t.strip() for t in instr.split(',') if t.strip()]
+    return " или ".join(terms) if terms else ""
+
+def generate_prompts(description, product_name, benefits, key_message, chosen_approach, 
+                     gender, generation, psychotype, business_stage, industry, opf):
+    # Если подход не определён
+    if chosen_approach == "Подход не найден для выбранных параметров.":
+        return ("Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода",
+                "Для формирования промпта выберите хотя бы один личный персональный параметр для определения подхода")
+
+    # Собираем approach_prefix и approach_suffix
+    approach_list = [a.strip() for a in chosen_approach.split(',')]
+    prefix_parts = []
+    suffix_parts = []
+    for a in approach_list:
+        if a in approach_dict:
+            prefix_parts.append(approach_dict[a]["prefix"])
+            suffix_parts.append(approach_dict[a]["suffix"])
+    approach_prefix = " ".join(prefix_parts) if prefix_parts else ""
+    approach_suffix = " ".join(suffix_parts) if suffix_parts else ""
+
+    # Собираем инструкции из персонализационных параметров
+    # Порядок: Пол, Поколение, Психотип, Стадия бизнеса, Отрасль, ОПФ
+    instructions_data = [
+        ("Пол", gender, genders_data, "Пол"),
+        ("Поколение", generation, generations_data, "Поколение"),
+        ("Психотип", psychotype, psychotypes_data, "Психотип"),
+        ("Стадия бизнеса", business_stage, business_stages_data, "Стадия бизнеса"),
+        ("Отрасль", industry, industries_data, "Отрасль"),
+        ("ОПФ", opf, opfs_data, "ОПФ")
+    ]
+
+    used_instructions = []
+    for (param_name, param_value, df, col) in instructions_data:
+        instr1, instr2 = get_instructions_for_param(param_name, param_value, df, col)
+        # Добавляем инструкцию, только если есть значение
+        if instr1 and instr1.strip():
+            instr_str = format_instruction_string(instr1)
+            used_instructions.append(instr_str)
+        if instr2 and instr2.strip():
+            instr_str = format_instruction_string(instr2)
+            used_instructions.append(instr_str)
+
+    # Формируем блок обязательных терминов
+    # Каждый элемент used_instructions – это уже подготовленная строка с "или".
+    # Нужно их пронумеровать:
+    mandatory_terms = ""
+    if used_instructions:
+        lines = []
+        for i, instr_line in enumerate(used_instructions, start=1):
+            lines.append(f"{i}. {instr_line}")
+        mandatory_terms = "\n".join(lines)
+    else:
+        # Если нет выбранных параметров или инструкции пусты,
+        # можно по умолчанию не выводить пункты или оставить пустым
+        # Но по условию задачи, если нет подхода, уже обработали выше.
+        # Если есть подход, но нет инструкций, значит никаких персональных параметров не выбрано
+        # Тогда будет без обязательных терминов. Можно просто не выводить пункты.
+        mandatory_terms = "Нет обязательных терминов, так как не выбрано ни одного параметра"
+
+    # Формируем итоговый промпт
+    if used_instructions:
+        prompt = f"""Напиши три или четыре предложения. {approach_prefix}. Напиши рекламное SMS для следующего продукта:
+«{description}».
+Не изменяй название продукта: «{product_name}».
+Преимущества:
+«{benefits}».
+ОБЯЗАТЕЛЬНО используй в SMS КАЖДЫЙ из следующих терминов:
+{mandatory_terms}
+Убедись, что написал не меньше трех и не больше четырех предложений.
+{approach_suffix}.
+Убедись, что УМЕСТНО использовал КАЖДЫЙ необходимый термин.
+Убедись, что в SMS без изменений, синонимов и перестановок слов используется ��аименование продукта: «{product_name}».
+Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+    else:
+        # Если нет выбранных параметров с инструкциями, можно вывести вариант без инструкций
+        prompt = f"""Напиши три или четыре предложения. {approach_prefix}. Напиши рекламное SMS для следующего продукта:
+«{description}».
+Не изменяй название продукта: «{product_name}».
+Преимущества:
+«{benefits}».
+Убедись, что написал не меньше трех и не больше четырех предложений.
+{approach_suffix}.
+Убедись, что в SMS без изменений, синонимов и перестановок слов используется наименование продукта: «{product_name}».
+Убедись, что в SMS есть следующая ключевая информация: «{key_message}»."""
+
+    return prompt, prompt
+
+def create_personalized_sms(description, product_name, benefits, key_message, chosen_approach,
+                            gender, generation, psychotype, business_stage, industry, opf):
+    # Генерируем промпты 1 и 2 по новой логике
+    prompt_1, prompt_2 = generate_prompts(description, product_name, benefits, key_message,
+                                           chosen_approach, gender, generation, psychotype,
+                                           business_stage, industry, opf)
+    # Возвращаем промпты для отображения
+    return prompt_1, prompt_2
+
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
-    
-    # Прогресс-бар
     progress_bar_html = """
     <div style="width: 100%; background-color: #e0e0e0; border-radius: 10px; overflow: hidden;">
         <div style="width: 0%; background-color: #4caf50; height: 20px; text-align: center; color: white;">
@@ -86,17 +213,12 @@ with gr.Blocks(theme="default") as demo:
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("**Продукт**")
-            product_dropdown = gr.Dropdown(
-                label="Продукт", 
-                choices=products,
-                value=None
-            )
+            product_dropdown = gr.Dropdown(label="Продукт", choices=products, value=None)
             description = gr.Textbox(label="Описание предложения", lines=5, value="")
             product_name = gr.Textbox(label="Наименование продукта", lines=1, value="")
             benefits = gr.Textbox(label="Преимущества", lines=9, value="")
             key_message = gr.Textbox(label="Ключевое сообщение", lines=2, value="")
         
-            # Обработчик выбора продукта
             product_dropdown.change(
                 fn=lambda selected: fill_product_details(selected, data_products),
                 inputs=[product_dropdown],
@@ -105,59 +227,59 @@ with gr.Blocks(theme="default") as demo:
         
         with gr.Column(scale=1):
             gr.Markdown("**Клиент**")
-            gender_dropdown = gr.Dropdown(label="Пол", choices=genders, value=None)
-            generation_dropdown = gr.Dropdown(label="Поколение", choices=generations, value=None)
-            psychotype_dropdown = gr.Dropdown(label="Психотип", choices=psychotypes, value=None)
-            business_stage_dropdown = gr.Dropdown(label="Стадия бизнеса", choices=business_stages, value=None)
-            industry_dropdown = gr.Dropdown(label="Отрасль", choices=industries, value=None)
-            opf_dropdown = gr.Dropdown(label="ОПФ", choices=opfs, value=None)
+            gender_dropdown = gr.Dropdown(label="Пол", choices=["Не выбрано"]+genders, value=None)
+            generation_dropdown = gr.Dropdown(label="Поколение", choices=["Не выбрано"]+generations, value=None)
+            psychotype_dropdown = gr.Dropdown(label="Психотип", choices=["Не выбрано"]+psychotypes, value=None)
+            business_stage_dropdown = gr.Dropdown(label="Стадия бизнеса", choices=["Не выбрано"]+business_stages, value=None)
+            industry_dropdown = gr.Dropdown(label="Отрасль", choices=["Не выбрано"]+industries, value=None)
+            opf_dropdown = gr.Dropdown(label="ОПФ", choices=["Не выбрано"]+opfs, value=None)
             chosen_approach = gr.Textbox(label="Выбранный подход", lines=1, value="")
             presence_in_db = gr.Textbox(label="Наличие в базе", lines=1, value="", interactive=False)
             
-            # При изменении Пол, Поколение, Психотип вызываем update_approach
             gender_dropdown.change(fn=update_approach, inputs=[gender_dropdown, generation_dropdown, psychotype_dropdown], outputs=[chosen_approach])
             generation_dropdown.change(fn=update_approach, inputs=[gender_dropdown, generation_dropdown, psychotype_dropdown], outputs=[chosen_approach])
             psychotype_dropdown.change(fn=update_approach, inputs=[gender_dropdown, generation_dropdown, psychotype_dropdown], outputs=[chosen_approach])
 
-    # Кнопки: Вернуть параметры, Создать персонализированное SMS
     with gr.Row():
         return_params_btn = gr.Button("Вернуть параметры предыдущего запроса")
         create_personal_sms_btn = gr.Button("Создать персонализированное SMS")
-    
-    # Блок для Модель 1, Промпт 1, SMS 1
+
     with gr.Row():
         with gr.Column():
-            model_1_name = gr.Textbox(label="Модель 1", value="Скрыто для слепого тестирования", interactive=False)
-            prompt_1 = gr.Textbox(label="Промпт 1", value="Скрыто для слепого тестирования", interactive=False)
+            model_1_name = gr.Textbox(label="Модель 1", value="Модель 1", interactive=False)
+            prompt_1 = gr.Textbox(label="Промпт 1", value="", interactive=False, lines=10)
             sms_1 = gr.Textbox(label="SMS 1", lines=3, value="")
         
-        # Блок для Модель 2, Промпт 2, SMS 2
         with gr.Column():
-            model_2_name = gr.Textbox(label="Модель 2", value="Скрыто для слепого тестирования", interactive=False)
-            prompt_2 = gr.Textbox(label="Промпт 2", value="Скрыто для слепого тестирования", interactive=False)
+            model_2_name = gr.Textbox(label="Модель 2", value="Модель 2", interactive=False)
+            prompt_2 = gr.Textbox(label="Промпт 2", value="", interactive=False, lines=10)
             sms_2 = gr.Textbox(label="SMS 2", lines=3, value="")
     
-    # Кнопки выбора SMS
     with gr.Row():
         prefer_sms_1_btn = gr.Button("Я предпочитаю это SMS")
         prefer_sms_2_btn = gr.Button("Я предпочитаю это SMS")
 
-    # Кнопка "Перегенерировать SMS"
     regen_btn = gr.Button("Перегенерировать SMS (не нравится ни одно из SMS)")
     
-    # Комментарии к SMS 1 и 2
     with gr.Row():
         comment_sms_1 = gr.Textbox(label="Комментарий к SMS 1", lines=2, value="")
         comment_sms_2 = gr.Textbox(label="Комментарий к SMS 2", lines=2, value="")
     
-    # Откорректированные SMS
     with gr.Row():
         corrected_sms_1 = gr.Textbox(label="Откорректированное SMS 1", lines=3, value="")
         corrected_sms_2 = gr.Textbox(label="Откорректированное SMS 2", lines=3, value="")
     
-    # Кнопки сохранения
     with gr.Row():
         save_sms_1_btn = gr.Button("Сохранить в базу")
         save_sms_2_btn = gr.Button("Сохранить в базу")
-    
+
+    # При нажатии на "Создать персонализированное SMS" формируем промпты
+    create_personal_sms_btn.click(
+        fn=create_personalized_sms,
+        inputs=[description, product_name, benefits, key_message, chosen_approach, 
+                gender_dropdown, generation_dropdown, psychotype_dropdown, 
+                business_stage_dropdown, industry_dropdown, opf_dropdown],
+        outputs=[prompt_1, prompt_2]
+    )
+
     demo.launch()