Update app.py

app.py

@@ -21,6 +21,7 @@ st.title("Прогнозирование цены автомобиля")
 inputs = {}
 
 # Выбор бренда
+unique_brands = df['brand'].unique()
 selected_brand = st.selectbox("Выберите бренд авто", unique_brands)
 
 # Выбор модели в зависимости от бренда
@@ -34,15 +35,16 @@ categorical_columns = ['brand', 'model', 'поколение', 'тип прод
                         'модификация', 'тип двигателя', 'коробка передач', 'привод',
                         'комплектация', 'тип кузова', 'цвет', 'авито оценка']
 
+numerical_columns = ['год выпуска', 'пробег', 'объем двигателя']
+
 # Остальные окошки для ввода параметров
-for column in categorical_columns:
-    if column not in ['brand', 'model']:
-        if column == 'год выпуска' or column == 'пробег' or column == 'объем двигателя':
-            inputs[column] = st.number_input(f"Введите значение для {column}")
-        elif column in ['модификация', 'объем двигателя']:
-            inputs[column] = st.text_input(f"Введите значение для {column}")
-        else:
-            inputs[column] = st.text_input(f"Введите значение для {column}")
+for column in df.columns:
+    if column in categorical_columns:
+        inputs[column] = st.selectbox(f"Выберите значение для {column}", df[column].unique())
+    elif column in numerical_columns:
+        inputs[column] = st.number_input(f"Введите значение для {column}")
+    else:
+        inputs[column] = st.text_input(f"Введите значение для {column}")
 
 # Кнопка для запуска предсказания
 if st.button("Предсказать цену"):