Update app.py

app.py

@@ -94,6 +94,94 @@ categorical_defaults = data['categorical_defaults']
 # Основной заголовок
 st.title('Классификатор пациентов')
 
+## Примеры:
+# Примеры типичных случаев
+st.header('Примеры типичных случаев')
+
+# Создаем примеры для каждой группы (замените значения на реальные типичные случаи)
+example_cases = {
+    'контр': {
+        'title': 'Пример контрольной группы',
+        'description': 'Типичный случай для контрольной группы',
+        'values': {
+            # Заполните типичными значениями для контрольной группы
+            'age': 45,
+            'Шкала_ПВА 0': 15,
+            'HADS_Тревога 0': 8,
+            'HADS_Депрессия 0': 7,
+            # ... добавьте остальные значения
+        }
+    },
+    'топирамат': {
+        'title': 'Пример группы топирамата',
+        'description': 'Типичный случай для группы топирамата',
+        'values': {
+            # Заполните типичными значениями для группы топирамата
+            'age': 42,
+            'Шкала_ПВА 0': 18,
+            'HADS_Тревога 0': 10,
+            'HADS_Депрессия 0': 9,
+            # ... добавьте остальные значения
+        }
+    },
+    'леветирацетам': {
+        'title': 'Пример группы леветирацетама',
+        'description': 'Типичный случай для группы леветирацетама',
+        'values': {
+            # Заполните типичными значениями для группы леветирацетама
+            'age': 38,
+            'Шкала_ПВА 0': 20,
+            'HADS_Тревога 0': 12,
+            'HADS_Депрессия 0': 11,
+            # ... добавьте остальные значения
+        }
+    }
+}
+
+# Отображение примеров в виде карточек
+col1, col2, col3 = st.columns(3)
+
+def create_example_card(title, description, values, key):
+    with st.container():
+        st.markdown(f"""
+            <div style="
+                padding: 1rem;
+                border-radius: 0.5rem;
+                border: 1px solid #e0e0e0;
+                margin: 0.5rem 0;
+                background-color: white;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1);">
+                <h3>{title}</h3>
+                <p>{description}</p>
+            </div>
+        """, unsafe_allow_html=True)
+        if st.button('Применить этот пример', key=key):
+            st.session_state['current_example'] = values
+
+with col1:
+    create_example_card(
+        example_cases['контр']['title'],
+        example_cases['контр']['description'],
+        example_cases['контр']['values'],
+        'example_control'
+    )
+
+with col2:
+    create_example_card(
+        example_cases['топирамат']['title'],
+        example_cases['топирамат']['description'],
+        example_cases['топирамат']['values'],
+        'example_topirama'
+    )
+
+with col3:
+    create_example_card(
+        example_cases['леветирацетам']['title'],
+        example_cases['леветирацетам']['description'],
+        example_cases['леветирацетам']['values'],
+        'example_levetiracetam'
+    )
+
 # Описание в основной части
 with st.container():
     st.markdown("""
@@ -120,6 +208,9 @@ with st.sidebar:
     
     input_data = {}
     
+    # Получаем значения из примера, если он выбран
+    current_example = st.session_state.get('current_example', {})
+    
     with tab1:
         # Числовые признаки
         for i in range(0, len(numeric_features), 2):
@@ -127,22 +218,22 @@ with st.sidebar:
             with col1:
                 if i < len(numeric_features):
                     feature = numeric_features[i]
-                    default_value = float(numeric_defaults[feature])
+                    default_value = current_example.get(feature, float(numeric_defaults[feature]))
                     input_data[feature] = st.number_input(
                         f'{feature}',
                         value=default_value,
                         format="%.2f",
-                        help=f"Среднее: {default_value:.2f}"
+                        help=f"Среднее: {numeric_defaults[feature]:.2f}"
                     )
             with col2:
                 if i + 1 < len(numeric_features):
                     feature = numeric_features[i + 1]
-                    default_value = float(numeric_defaults[feature])
+                    default_value = current_example.get(feature, float(numeric_defaults[feature]))
                     input_data[feature] = st.number_input(
                         f'{feature}',
                         value=default_value,
                         format="%.2f",
-                        help=f"Среднее: {default_value:.2f}"
+                        help=f"Среднее: {numeric_defaults[feature]:.2f}"
                     )
     
     with tab2:
@@ -153,7 +244,8 @@ with st.sidebar:
                 if i < len(categorical_features):
                     feature = categorical_features[i]
                     options = categorical_options[feature]
-                    default_idx = options.index(categorical_defaults[feature]) if categorical_defaults[feature] in options else 0
+                    default_value = current_example.get(feature, categorical_defaults[feature])
+                    default_idx = options.index(default_value) if default_value in options else 0
                     input_data[feature] = st.selectbox(
                         f'{feature}',
                         options,
@@ -164,7 +256,8 @@ with st.sidebar:
                 if i + 1 < len(categorical_features):
                     feature = categorical_features[i + 1]
                     options = categorical_options[feature]
-                    default_idx = options.index(categorical_defaults[feature]) if categorical_defaults[feature] in options else 0
+                    default_value = current_example.get(feature, categorical_defaults[feature])
+                    default_idx = options.index(default_value) if default_value in options else 0
                     input_data[feature] = st.selectbox(
                         f'{feature}',
                         options,
@@ -289,7 +382,11 @@ if 'prediction' in st.session_state:
                 - При уверенности ниже 50% рекомендуется дополнительное обследование
                 - Результаты модели носят рекомендательный характер
             """)
-
+# Добавьте кнопку сброса примера
+    if st.session_state.get('current_example'):
+        if st.button('Сбросить пример'):
+            del st.session_state['current_example']
+            st.experimental_rerun()
 # Добавляем footer
 st.markdown("""
     ---