Update app.py

app.py

@@ -1,11 +1,21 @@
 import gradio as gr
 import pandas as pd
 
-# Функция для загрузки данных из файла Excel
-def load_products_data(file_path):
-    data = pd.read_excel(file_path, sheet_name="Продукты")
-    products = data["Наименование продукта"].unique().tolist()
-    return data, products
+# Функция для загрузки данных из вкладок 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()
+
+# Загрузка данных из всех вкладок
+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, "ОПФ", "ОПФ")
 
 # Функция для заполнения полей на основе выбранного продукта
 def fill_product_details(selected_product, data):
@@ -19,10 +29,6 @@ def fill_product_details(selected_product, data):
         )
     return "", "", "", ""
 
-# Загрузка данных
-file_path = "Исходные данные.xlsx"
-data, products = load_products_data(file_path)
-
 # Создание интерфейса
 with gr.Blocks(theme="default") as demo:
     gr.Markdown("**Процент созданных SMS по выбранному продукту**")
@@ -50,23 +56,23 @@ with gr.Blocks(theme="default") as demo:
             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),
-            inputs=[product_dropdown],
-            outputs=[description, product_name, benefits, key_message]
-        )
+            # Обработчик выбора продукта
+            product_dropdown.change(
+                fn=lambda selected: fill_product_details(selected, data_products),
+                inputs=[product_dropdown],
+                outputs=[description, product_name, benefits, key_message]
+            )
         
         with gr.Column(scale=1):
             gr.Markdown("**Клиент**")
-            gender = gr.Dropdown(label="Пол", choices=["Мужской", "Женский", "Не определено"], value=None)
-            generation = gr.Dropdown(label="Поколение", choices=["X", "Y", "Z"], value=None)
-            psychotype = gr.Dropdown(label="Психотип", choices=["Аналитик", "Эмоциональный", "Практик"], value=None)
-            business_stage = gr.Dropdown(label="Стадия бизнеса", choices=["Стартап", "Рост", "Зрелость"], value=None)
-            industry = gr.Dropdown(label="Отрасль", choices=["Ритейл", "IT", "Производство"], value=None)
-            opf = gr.Dropdown(label="ОПФ", choices=["ООО", "ИП", "АО"], 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="В базе есть SMS под данные параметры персонализации.", interactive=False)
+            presence_in_db = gr.Textbox(label="Наличие в базе", lines=1, value="В базе есть SMS под данные параметры персонализации.", interactive=False)
 
     # Кнопки: Вернуть параметры, Создать персонализированное SMS
     with gr.Row():
@@ -106,7 +112,7 @@ with gr.Blocks(theme="default") as demo:
     
     # Кнопки сохранения
     with gr.Row():
-        save_sms_1_btn = gr.Button("Сохранить в БД")
-        save_sms_2_btn = gr.Button("Сохранить в БД")
+        save_sms_1_btn = gr.Button("Сохранить в базу")
+        save_sms_2_btn = gr.Button("Сохранить в базу")
     
     demo.launch()