Update app.py

app.py

@@ -4,24 +4,14 @@ import pandas as pd
 # Функция для загрузки данных из вкладок Excel
 def load_dropdown_data(file_path, sheet_name, column_name):
     """Загружает уникальные значения из столбца на заданной вкладке."""
-    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 []
+    data = pd.read_excel(file_path, sheet_name=sheet_name)
+    return data[column_name].dropna().unique().tolist()
 
-# Путь к файлу Excel
+# Путь к файлу
 file_path = "Исходные данные.xlsx"
 
-# Загрузка данных из всех вкладок
-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()
-
+# Загрузка данных
+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, "Психотип", "Психотип")
@@ -29,63 +19,55 @@ 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
+# Загрузка данных для "Подход"
+approaches_data = pd.read_excel(file_path, sheet_name="Подход")
 
 # Функция для заполнения полей на основе выбранного продукта
 def fill_product_details(selected_product, data):
-    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("Ключевое сообщение", "")
-            )
+    if selected_product:
+        product_row = data[data["Наименование продукта"] == selected_product].iloc[0]
+        return (
+            product_row.get("Описание предложения", ""),
+            product_row.get("Наименование продукта", ""),
+            product_row.get("Преимущества", ""),
+            product_row.get("Ключевое сообщение", "")
+        )
     return "", "", "", ""
 
-# Функция для определения подходов
-def determine_approaches(gender, generation, psychotype):
+# Функция для получения списка подходов
+def get_approaches(gender, generation, psychotype, approaches_df):
     if approaches_df is None or approaches_df.empty:
-        return "Таблица 'Подход' не найдена или пуста."
-    
+        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
+            filters.append(approaches_df[param_name].isnull() | (approaches_df[param_name].fillna('') == ''))
     
-    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()
+    combined_filter = filters[0]
+    for f in filters[1:]:
+        combined_filter &= f
     
+    matching_rows = approaches_df[combined_filter]
     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_names = [a.strip() for a in str(approaches).split(',')]
         approach_list.extend(approach_names)
     
-    approach_list = list(set(approach_list))  # Убираем дубликаты
-    selected_approaches_text_content = ', '.join(approach_list)
+    # Убираем дубликаты
+    approach_list = list(set(approach_list))
     
-    return selected_approaches_text_content if approach_list else "Подход не найден для выбранных параметров."
+    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)
 
 # Создание интерфейса
 with gr.Blocks(theme="default") as demo:
@@ -107,7 +89,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="")
@@ -123,35 +105,19 @@ with gr.Blocks(theme="default") as demo:
         
         with gr.Column(scale=1):
             gr.Markdown("**Клиент**")
-            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)
+            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)
-
-    # Обработчики для определения подходов
-    # Когда изменяется любое из трех полей, вызывается функция 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
-        )
+            
+            # При изменении Пол, Поколение, Психотип вызываем 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():