Update app.py

app.py

@@ -4,12 +4,24 @@ import pandas as pd
 # Функция для загрузки данных из вкладок Excel
 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()
+    try:
+        data = pd.read_excel(file_path, sheet_name=sheet_name)
+        return data[column_name].dropna().unique().tolist()
+    except Exception as e:
+        print(f"Ошибка загрузки данных из листа '{sheet_name}': {e}")
+        return []
 
-# Загрузка данных из всех вкладок
+# Путь к файлу Excel
 file_path = "Исходные данные.xlsx"
-products, data_products = load_dropdown_data(file_path, "Продукты", "Наименование продукта"), pd.read_excel(file_path, sheet_name="Продукты")
+
+# Загрузка данных из всех вкладок
+products = load_dropdown_data(file_path, "Продукты", "Наименование продукта")
+try:
+    data_products = pd.read_excel(file_path, sheet_name="Продукты")
+except Exception as e:
+    print(f"Ошибка загрузки листа 'Продукты': {e}")
+    data_products = pd.DataFrame()
+
 genders = load_dropdown_data(file_path, "Пол", "Пол")
 generations = load_dropdown_data(file_path, "Поколение", "Поколение")
 psychotypes = load_dropdown_data(file_path, "Психотип", "Психотип")
@@ -17,18 +29,64 @@ business_stages = load_dropdown_data(file_path, "Стадия бизнеса", "
 industries = load_dropdown_data(file_path, "Отрасль", "Отрасль")
 opfs = load_dropdown_data(file_path, "ОПФ", "ОПФ")
 
+# Загрузка подходов
+try:
+    approaches_df = pd.read_excel(file_path, sheet_name="Подход")
+except Exception as e:
+    print(f"Ошибка загрузки листа 'Подход': {e}")
+    approaches_df = None
+
 # Функция для заполнения полей на основе выбранного продукта
 def fill_product_details(selected_product, data):
-    if selected_product:
-        product_row = data[data["Наименование продукта"] == selected_product].iloc[0]
-        return (
-            product_row["Описание предложения"],
-            product_row["Наименование продукта"],
-            product_row["Преимущества"],
-            product_row["Ключевое сообщение"]
-        )
+    if selected_product and not data.empty:
+        product_row = data[data["Наименование продукта"] == selected_product]
+        if not product_row.empty:
+            product_row = product_row.iloc[0]
+            return (
+                product_row.get("Описание предложения", ""),
+                product_row.get("Наименование продукта", ""),
+                product_row.get("Преимущества", ""),
+                product_row.get("Ключевое сообщение", "")
+            )
     return "", "", "", ""
 
+# Функция для определения подходов
+def determine_approaches(gender, generation, psychotype):
+    if approaches_df is None or approaches_df.empty:
+        return "Таблица 'Подход' не найдена или пуста."
+    
+    filters = []
+    for param_name, param_value in [('Пол', gender), ('Поколение', generation), ('Психотип', psychotype)]:
+        if param_value:
+            filters.append(approaches_df[param_name].fillna('') == param_value)
+        else:
+            # Если параметр не выбран, не фильтруем по нему
+            continue
+    
+    if filters:
+        combined_filter = filters[0]
+        for f in filters[1:]:
+            combined_filter &= f
+        matching_rows = approaches_df[combined_filter]
+    else:
+        # Если ни один параметр не выбран, все строки соответствуют
+        matching_rows = approaches_df.copy()
+    
+    if matching_rows.empty:
+        return "Подход не найден для выбранных параметров."
+    
+    approach_list = []
+    for approaches in matching_rows['Подход']:
+        if pd.isna(approaches):
+            continue
+        approach_names = [a.strip() for a in approaches.split(',')]
+        approach_list.extend(approach_names)
+    
+    approach_list = list(set(approach_list))  # Убираем дубликаты
+    selected_approaches_text_content = ', '.join(approach_list)
+    
+    return selected_approaches_text_content if approach_list else "Подход не найден для выбранных параметров."
+
 # Создание интерфейса
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -49,7 +107,7 @@ with gr.Blocks(theme="default") as demo:
             product_dropdown = gr.Dropdown(
                 label="Продукт", 
                 choices=products,
-                value=None
+                value=None  # Значение по умолчанию не устанавливается
             )
             description = gr.Textbox(label="Описание предложения", lines=5, value="")
             product_name = gr.Textbox(label="Наименование продукта", lines=1, value="")
@@ -65,15 +123,36 @@ 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)
-            chosen_approach = gr.Textbox(label="Выбранный подход", lines=1, value="")
+            gender_dropdown = gr.Dropdown(label="Пол", choices=["Не выбрано"] + genders, value="Не выбрано")
+            generation_dropdown = gr.Dropdown(label="Поколение", choices=["Не выбрано"] + generations, value="Не выбрано")
+            psychotype_dropdown = gr.Dropdown(label="Психотип", choices=["Не выбрано"] + psychotypes, value="Не выбрано")
+            business_stage_dropdown = gr.Dropdown(label="Стадия бизнеса", choices=["Не выбрано"] + business_stages, value="Не выбрано")
+            industry_dropdown = gr.Dropdown(label="Отрасль", choices=["Не выбрано"] + industries, value="Не выбрано")
+            opf_dropdown = gr.Dropdown(label="ОПФ", choices=["Не выбрано"] + opfs, value="Не выбрано")
+            chosen_approach = gr.Textbox(label="Выбранный подход", lines=2, value="", interactive=False)
             presence_in_db = gr.Textbox(label="Наличие в базе", lines=1, value="", interactive=False)
 
+    # Обработчики для определения подходов
+    # Когда изменяется любое из трех полей, вызывается функция determine_approaches
+    inputs_for_approach = [gender_dropdown, generation_dropdown, psychotype_dropdown]
+    approach_output = chosen_approach
+
+    def update_approach(gender, generation, psychotype):
+        # Преобразуем "Не выбрано" в None для фильтрации
+        gender = None if gender == "Не выбрано" else gender
+        generation = None if generation == "Не выбрано" else generation
+        psychotype = None if psychotype == "Не выбрано" else psychotype
+        result = determine_approaches(gender, generation, psychotype)
+        return result
+
+    # Привязываем обработчик к каждому из дропдаунов
+    for input_component in inputs_for_approach:
+        input_component.change(
+            fn=update_approach,
+            inputs=inputs_for_approach,
+            outputs=approach_output
+        )
+
     # Кнопки: Вернуть параметры, Создать персонализированное SMS
     with gr.Row():
         return_params_btn = gr.Button("Вернуть параметры предыдущего запроса")