Update app.py

app.py

@@ -38,10 +38,19 @@ def fetch_stock_data(symbol, start_date, end_date):
 # Step 2: Add Technical Indicators
 def add_technical_indicators(data):
     print("Adding technical indicators...")
+    # Compute RSI
     data['RSI'] = ta.rsi(data['Close'], length=14)
-    data['MACD'] = ta.macd(data['Close'], fast=12, slow=26)['MACD_12_26_9']
-    bbands = ta.bbands(data['Close'])
-    data['BB_upper'], data['BB_middle'], data['BB_lower'] = bbands['BBU_20_2.0'], bbands['BBM_20_2.0'], bbands['BBL_20_2.0']
+    
+    # Compute MACD (usually generates three columns: MACD, MACD_Signal, MACD_Histogram)
+    macd = ta.macd(data['Close'], fast=12, slow=26, signal=9)
+    data['MACD'] = macd['MACD_12_26_9']
+    
+    # Compute Bollinger Bands
+    bbands = ta.bbands(data['Close'], length=20, std=2.0)
+    data['BB_upper'] = bbands[f'BBU_20_2.0'] if f'BBU_20_2.0' in bbands.columns else bbands.iloc[:, 0]
+    data['BB_middle'] = bbands[f'BBM_20_2.0'] if f'BBM_20_2.0' in bbands.columns else bbands.iloc[:, 1]
+    data['BB_lower'] = bbands[f'BBL_20_2.0'] if f'BBL_20_2.0' in bbands.columns else bbands.iloc[:, 2]
+    
     return data
 
 # Step 3: Analyze Sentiment using FinBERT