Update app.py

app.py

@@ -1,6 +1,27 @@
 import gradio as gr
 import plotly.graph_objects as go
 import random
+import pandas as pd
+
+# Загрузка данных из Excel-файла
+try:
+    data = pd.read_excel('Признаки.xlsx', sheet_name=None)
+except Exception as e:
+    print(f"Ошибка при загрузке Excel-файла: {e}")
+    data = {}
+
+# Создание списка признаков и их значений
+features = {}
+for sheet_name, df in data.items():
+    try:
+        if sheet_name == "Пол Поколение Психотип":
+            # Создаем словарь, где ключи — это кортежи (Пол, Поколение, Психотип), а значения — инструкции
+            features[sheet_name] = df.set_index(['Пол', 'Поколение', 'Психотип'])['Инструкция'].to_dict()
+        else:
+            features[sheet_name] = df.set_index(df.columns[0]).to_dict()[df.columns[1]]
+    except Exception as e:
+        print(f"Ошибка при обработке данных листа {sheet_name}: {e}")
+        features[sheet_name] = {}
 
 # Функция для создания спидометра
 def create_gauge(value):
@@ -49,12 +70,13 @@ with gr.Blocks() as demo:
                     benefits = gr.Textbox(label="Преимущества", lines=5)  # Увеличенная высота
                     key_message = gr.Textbox(label="Ключевое сообщение", lines=5)  # Увеличенная высота
                 with gr.Column():
-                    gender = gr.Dropdown(label="Пол", choices=["Мужчина", "Женщина", "Не указан"])
-                    generation = gr.Dropdown(label="Поколение", choices=["Поколение Z", "Миллениалы", "Поколение X", "Бэби-бумеры"])
-                    psychotype = gr.Textbox(label="Психотип")
-                    business_stage = gr.Textbox(label="Стадия бизнеса")
-                    industry = gr.Textbox(label="Отрасль")
-                    opf = gr.Textbox(label="ОПФ")
+                    # Используем данные из Excel для выбора
+                    gender = gr.Dropdown(label="Пол", choices=[None] + list(features.get('Пол', {}).keys()))
+                    generation = gr.Dropdown(label="Поколение", choices=[None] + list(features.get('Поколение', {}).keys()))
+                    psychotype = gr.Dropdown(label="Психотип", choices=[None] + list(features.get('Психотип', {}).keys()))
+                    business_stage = gr.Dropdown(label="Стадия бизнеса", choices=[None] + list(features.get('Стадия бизнеса', {}).keys()))
+                    industry = gr.Dropdown(label="Отрасль", choices=[None] + list(features.get('Отрасль', {}).keys()))
+                    opf = gr.Dropdown(label="ОПФ", choices=[None] + list(features.get('ОПФ', {}).keys()))
 
             btn_to_prompts = gr.Button("Создать")
             btn_to_prompts.click(fn=change_tab, inputs=[gr.Number(value=1, visible=False)], outputs=tabs)