Update app.py

app.py

@@ -8,35 +8,97 @@ from datetime import datetime, timedelta
 from PIL import Image
 from plotly import graph_objs as go
 from datetime import date
+from model.lstm_model import BiLSTM
+
+# Загрузка весов модели (выполняется только при первом запуске)
+@st.cache(allow_output_mutation=True)
+def load_model_weights():
+    model = BiLSTM(input_size, hidden_size, num_layers, output_size)
+    model.load_state_dict(torch.load('model/model_weights.pth'))
+    model.eval()
+    return model
+
 
 st.set_page_config(layout='wide', initial_sidebar_state='expanded')
 st.set_option('deprecation.showPyplotGlobalUse', False)
 st.title('ML Wall Street')
 st.image('images/img.png')
 
-# @st.cache_data
 # Функция для получения данных о ценах акций
+@st.cache(allow_output_mutation=True)
 def get_stock_data():
     dow_tickers = ['UNH', 'MSFT', 'GS', 'HD', 'AMGN', 'MCD', 'CAT', 'CRM', 'V', 'BA', 'HON', 'TRV', 'AAPL', 'AXP', 'JPM', 'IBM', 'JNJ', 'WMT', 'PG', 'CVX', 'MRK', 'MMM', 'NKE', 'DIS', 'KO', 'DOW', 'CSCO', 'INTC', 'VZ', 'WBA']
     start_date = (datetime.now() - timedelta(days=365)).strftime('%Y-%m-%d')
     end_date = datetime.now().strftime('%Y-%m-%d')
-    dow_data = yf.download(dow_tickers, start=start_date, end=end_date)
+
+    # Проверка, прошло ли более 24 часов с последнего обновления данных
+    if 'last_stock_update' not in st.session_state or (datetime.now() - st.session_state.last_stock_update).total_seconds() > 43200:
+        dow_data = yf.download(dow_tickers, start=start_date, end=end_date)
+
+        # Сохранение данных в кэш и в сессионном состоянии
+        st.cache(f"{start_date}_{end_date}")(dow_data)
+        st.session_state.stock_data = dow_data
+        st.session_state.last_stock_update = datetime.now()
+    else:
+        # Если данные уже в сессионном состоянии, возвращаем их
+        dow_data = st.session_state.stock_data
+
     return dow_data
 
+# Функция для получения данных по индексу
+@st.cache(allow_output_mutation=True)
+def load_data(index_symbol):
+    start_date = "2021-01-01"
+    end_date = date.today().strftime("%Y-%m-%d")
+
+    # Проверка, прошло ли более 24 часов с последнего обновления данных
+    last_update_key = f'last_{index_symbol.lower()}_update'
+    data_key = f'{index_symbol.lower()}_data'
+
+    if last_update_key not in st.session_state or (datetime.now() - st.session_state[last_update_key]).total_seconds() > 86400:
+        df = yf.download(index_symbol, start=start_date, end=end_date)
+        df.reset_index(inplace=True)
+
+        # Сохранение данных в кэш и в сессионном состоянии
+        st.cache(f"{index_symbol.lower()}_{start_date}_{end_date}")(df)
+        st.session_state[data_key] = df
+        st.session_state[last_update_key] = datetime.now()
+    else:
+        # Если данные уже в сессионном состоянии, возвращаем их
+        df = st.session_state[data_key]
+
+    return df
+
+# Проверка, прошло ли более 12 часов с последнего обновления данных для индекса
+if last_update_key not in st.session_state or (datetime.now() - st.session_state[last_update_key]).total_seconds() > 43200:
+    df = yf.download(index_symbol, start=start_date, end=end_date)
+    df.reset_index(inplace=True)
+
+    # Сохранение данных в кэш и в сессионном состоянии
+    st.cache(f"{index_symbol.lower()}_{start_date}_{end_date}")(df)
+    st.session_state[data_key] = df
+    st.session_state[last_update_key] = datetime.now()
+else:
+    # Если данные уже в сессионном состоянии, возвращаем их
+    df = st.session_state[data_key]
+
+# Пример использования для разных индексов
+bitcoin_data = load_data('BTC-USD')
+sse_data = load_data('000001.SS')
+moex_data = load_data('IMOEX.ME')
+dji_data = load_data('^DJI')
+sp500_data = load_data('^GSPC')
+
+# Получение данных о ценах акций
 data = get_stock_data() 
 latest_date = data.index[-1].strftime('%Y-%m-%d')
-data = data.loc[latest_date, 'Close'].reset_index()
-data.columns = ['Ticker', 'Close']
-data['Close'] = data['Close'].round(2)
-
-# Добавляем кнопку обновления данных
-# if st.button("Обновить данные", type="primary"):
-#     data = get_stock_data()
-#     latest_date = data.index[-1].strftime('%Y-%m-%d')
-#     data = data.loc[latest_date, 'Close'].reset_index()
-#     data.columns = ['Ticker', 'Close']
-#     data['Close'] = data['Close'].round(2)
-#     st.success("Данные успешно обновлены!")
+
+model_weights = load_model_weights()
+
+# Сохранение весов модели в ��ессионном состоянии
+if 'model_weights' not in st.session_state:
+    st.session_state.model_weights = model_weights
+
 
 st.markdown(f"<h3 style='text-align: center;'>Цены актуальны на последнюю дату закрытия торгов {latest_date}</h3>", unsafe_allow_html=True)
 
@@ -71,14 +133,6 @@ with col2:
     st.dataframe(data, height=1088, column_config={"Logo": image_column, "Ticker":ticker_column, 'Close':price_column})
 
 with col1:
-    START = "1920-01-01"
-    TODAY = date.today().strftime("%Y-%m-%d")
-    # @st.cache_data
-    def load_data(ticker):
-        data = yf.download(ticker, START, TODAY)
-        data.reset_index(inplace=True)
-        return data
-
     def plot_raw_data(data, text):
         fig = go.Figure()
         fig.add_trace(go.Scatter(x=data['Date'], y=data['Close'], name="Цена закрытия"))