budget_v31.05

'''UPD: Добавлена таблица aggrid с возможностью редактирования и флажками + поиск слова в таблице'''
import streamlit as st
import pandas as pd
from st_aggrid import AgGrid,GridOptionsBuilder, GridUpdateMode

st.set_page_config(page_title='Расчет бюджетной оценки',
                   page_icon=':pencil:',
                   layout='wide')

st.image('avaya.png',width=200) #вставка лого

#табличка с примером
df = pd.DataFrame({
  'first column': [1, 2, 3, 4],
  'second column': [10, 20, 30, 40]
})

#боковая панель
st.sidebar.header('Добавление новых записей в базу')
product_code = st.sidebar.number_input('Введите код товара (при наличии)',0,None) #param: min - 0, max - None
product_name = st.sidebar.text_input('Введите название позиции')
button_add_entry = st.sidebar.button("Добавить") #кнопка добавления новой записи в таблицу
if button_add_entry:
  new_row = {'first column':5}
  df.append(new_row, ignore_index=True)  #ошибка - не добавляется запись

st.title('Приложение для расчета и формирования бюджетной оценки') #заголовок

col1, col2 = st.columns(2) #разделение макета на 2 столбца

#поиск товара по коду (прецедент)
number_text = col1.number_input('Введите код товара',0,None,1)
st.write(f"Выбранный код товара: {int(number_text)}")
col2.caption('Нажмите на кпопку для поиска')
col2.button('Поиск')

#при использовании формы (form) страница не будет перегружаться каждый раз при изменении каких-то параметров,
#а только при нажатии кнопки (form_submit_button)
form1 = st.form(key='Table')
form1.table(df)
form1.form_submit_button('Click me')


find_expander = st.expander('Поиск по ключевым словам')
with find_expander:
  #поиск товара по ключевым словам (прецедент)
  find_expander.header('Поиск позиций по ключевым словам')
  options = find_expander.multiselect('Выберете ключевые слова',['Communication manager - CM',
'System manager - SMGR'])


@st.cache #кэширование данных
def file_upload(): #функция загрузки данных их excel-файла
  df2 = pd.read_excel(
        io='Landata.xlsx',
        engine='openpyxl',
        sheet_name='Landata',
        usecols='A:D',
        nrows=10)
  return df2

df2 = file_upload()

gb = GridOptionsBuilder.from_dataframe(df2)
gb.configure_pagination(paginationAutoPageSize=True) #есть навигация по страницам таблицы
gb.configure_default_column(editable=True) #возможность редактировать таблицу
gb.configure_selection('multiple', use_checkbox=True) #множетвенное выделение флажками
grid_options=gb.build()
grid_table = AgGrid(df2,gridOptions=grid_options,
                    update_mode = GridUpdateMode.SELECTION_CHANGED,
                    theme = 'light')

sel_rows = grid_table['selected_rows']
st.write(sel_rows)

st.markdown("""---""") #черта
#поиск по словам в столбце "Product Description"
df1 = df2[df2['Product Description'].str.contains('AAEP')] #выбор позиций, в которых есть слово 'AAEP'
st.write(df1)