Update app.py

app.py

@@ -93,6 +93,13 @@ class StockPredictor:
                 future[column] = self.data[column].iloc[-1]  # Use last known value
             
             forecast = self.model.predict(future)
+            
+            # Calculate components
+            forecast['trend'] = forecast['trend']
+            forecast['yearly'] = forecast['yearly'] if 'yearly' in forecast.columns else 0
+            forecast['weekly'] = forecast['weekly'] if 'weekly' in forecast.columns else 0
+            forecast['daily'] = forecast['daily'] if 'daily' in forecast.columns else 0
+            
             return forecast
         except Exception as e:
             print(f"Error predicting with Prophet model: {str(e)}")
@@ -468,7 +475,8 @@ def predict_stock_prices():
                         st.plotly_chart(fig, use_container_width=True)
 
                         # Create forecast components plot using Plotly
-                        components = predictor.model.predict_components(predictions)
+                        # Instead of using predict_components, we'll extract the components from the predictions DataFrame
+                        components = predictions[['trend', 'yearly', 'weekly', 'daily']]
                         n_components = len(components.columns)
                         
                         fig_components = make_subplots(rows=n_components, cols=1,