Update app.py

app.py

@@ -3,6 +3,8 @@ import yfinance as yf
 import pandas as pd
 import plotly.graph_objects as go
 from datetime import datetime
+from ta.momentum import RSIIndicator
+from ta.volatility import BollingerBands
 
 def vsa_advanced_analysis(data):
     signals = []
@@ -11,7 +13,6 @@ def vsa_advanced_analysis(data):
             signals.append((data.index[i], 'Buy', data['Close'][i]))
         elif data['Close'][i] < data['Close'][i-1] and data['Volume'][i] > data['Volume'][i-1] and data['Close'][i-1] < data['Close'][i-2]:
             signals.append((data.index[i], 'Sell', data['Close'][i]))
-        # Example of adding more advanced patterns
         if data['Volume'][i] > data['Volume'][i-1] * 1.5 and data['Close'][i] < data['Open'][i]:
             signals.append((data.index[i], 'Selling Climax', data['Close'][i]))
     return signals
@@ -34,6 +35,14 @@ def plot_signals_advanced(data, signals):
     fig.update_layout(title='Stock Price with Buy and Sell Signals', xaxis_title='Date', yaxis_title='Price')
     st.plotly_chart(fig)
 
+def add_technical_indicators(data):
+    rsi = RSIIndicator(data['Close']).rsi()
+    bb = BollingerBands(data['Close'])
+    data['RSI'] = rsi
+    data['BB_High'] = bb.bollinger_hband()
+    data['BB_Low'] = bb.bollinger_lband()
+    return data
+
 def main():
     st.title('Advanced VSA Stock Analyzer')
     st.write('Input a stock ticker symbol to analyze and generate buy and sell signals using advanced Volume Spread Analysis (VSA).')
@@ -47,6 +56,7 @@ def main():
         if data.empty:
             st.write('No data found for the selected stock ticker and date range.')
         else:
+            data = add_technical_indicators(data)
             signals = vsa_advanced_analysis(data)
             st.write(f'Found {len(signals)} signals.')
             if signals: