Fixed error nonetype

modules/data_preparation.py

@@ -20,6 +20,12 @@ def prepare_df(df, categories, date_filter):
 
     #df['topic_verification'][(df.headline.str.contains('crude', case=False)) | df.body.str.contains('crude', case=False)] = 'Crude Oil'
 
+    # Check if categories is not empty before filtering
+    if categories:
+        news_data = df[df['topic_verification'].isin(categories)]
+    else:
+        news_data = df
+
     try:
         news_data = df[df['topic_verification'].isin(categories)]
 

page_2.py

@@ -60,4 +60,4 @@ img2 = Image.open(os.path.join('price_forecasting_ml',
                               'artifacts',
                               'crude_oil_df1ce299-117d-43c7-bcd5-7ecaeac0bc89',
                               'forecast_plot.jpg'))
-st.image(img2, caption="Crude Oil Futures Price Forecasting - Univariate with Exogenous Features (Horizon = 5)")
+st.image(img2, caption="Crude Oil Futures Price Forecasting - Multivariate (Horizon = 5)")

req.txt

@@ -0,0 +1,21 @@
+seaborn
+torch
+torchvision
+torchaudio
+pyYAML
+scipy
+scikit-learn
+packaging
+pyWavelets
+scipy
+tiffile
+pandas
+nltk
+ipykernel
+matplotlib
+streamlit
+plotly
+nbformat
+wordcloud
+openpyxl
+neuralforecast

requirements.txt

@@ -1,21 +1,4 @@
-seaborn
-torch
-torchvision
-torchaudio
-pyYAML
-scipy
-scikit-learn
-packaging
-pyWavelets
-scipy
-tiffile
+numpy
 pandas
-nltk
-ipykernel
-matplotlib
-streamlit
 plotly
-nbformat
-wordcloud
-openpyxl
-neuralforecast
+openpyxl