Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
-import pandas as pd
 from datetime import datetime, timedelta
+import pandas as pd
 
-# Import our custom modules
+# Import custom modules
 from yfinance_data.fetcher import fetch_stock_data
 from indicators.sma import calculate_21_50_sma
 from indicators.bollinger_bands import calculate_bollinger_bands
@@ -12,33 +12,39 @@ from utils.plotting import plot_stock_data_with_signals
 # Streamlit app title
 st.title('StockSwingApp: Stock Trading Signal Generator')
 
-# User inputs
+# Sidebar for user inputs
 st.sidebar.header('User Input Parameters')
 default_symbol = 'AAPL'
 symbol = st.sidebar.text_input('Stock Symbol', value=default_symbol, max_chars=5).upper()
-start_date = st.sidebar.date_input('Start Date', value=datetime.now() - timedelta(days=365))
+start_date = st.sidebar.date_input('Start Date', value=datetime.now() - timedelta(days=90))
 end_date = st.sidebar.date_input('End Date', value=datetime.now())
 
-# Fetch stock data
-st.write(f"Fetching data for {symbol} from {start_date} to {end_date}...")
-stock_data = fetch_stock_data(symbol, start_date.strftime('%Y-%m-%d'), end_date.strftime('%Y-%m-%d'), interval='1d')
+# Data fetching and processing
+if st.sidebar.button('Generate Signals'):
+    st.write(f"Fetching data for {symbol} from {start_date} to {end_date}...")
 
-if not stock_data.empty:
-    # Calculate indicators
-    stock_data_with_indicators = calculate_21_50_sma(stock_data)
-    stock_data_with_indicators = calculate_bollinger_bands(stock_data_with_indicators)
+    try:
+        stock_data = fetch_stock_data(symbol, start_date.strftime('%Y-%m-%d'), end_date.strftime('%Y-%m-%d'), interval='1d')
 
-    # Generate signals
-    signals_data = generate_signals(stock_data_with_indicators)
+        if not stock_data.empty:
+            # Calculate indicators and signals
+            stock_data = calculate_21_50_sma(stock_data)
+            stock_data = calculate_bollinger_bands(stock_data)
+            signals_data = generate_signals(stock_data)
 
-    # Display the chart
-    st.write(f"Displaying data for {symbol}")
-    plot_stock_data_with_signals(signals_data)
+            # Plotting
+            fig = plot_stock_data_with_signals(signals_data)
+            st.pyplot(fig)
 
-    # Optionally, display the data table
-    show_data = st.checkbox('Show Data Table')
-    if show_data:
-        st.write(signals_data)
-else:
-    st.write("No data available for the given inputs. Please try different parameters.")
+            # Option to display raw data
+            if st.checkbox('Show Raw Data'):
+                st.write(signals_data)
 
+        else:
+            st.error("No data available for the given symbol. Please try another one.")
+
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
+
+else:
+    st.info('Enter parameters and click "Generate Signals" to view analysis.')