Ken Sang Tang commited on
Commit
19d8247
·
verified ·
1 Parent(s): e94c28f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -45,13 +45,22 @@ def fetch_stock_data(symbol, start_date, end_date):
45
  def add_technical_indicators(data):
46
  print("Adding technical indicators...")
47
  data['RSI'] = ta.rsi(data['Close'], length=14)
48
- data['MACD'], = ta.macd(data['Close'], fast=12, slow=26)['MACD_12_26_9']
49
- bbands = ta.bbands(data['Close'], length=20, std=2.0)
50
- # Handle potential missing columns
 
 
 
 
 
 
 
 
51
  if 'BBU_20_2.0' in bbands.columns:
52
  data['BB_upper'], data['BB_middle'], data['BB_lower'] = bbands['BBU_20_2.0'], bbands['BBM_20_2.0'], bbands['BBL_20_2.0']
53
  else:
54
  print("Bollinger Bands data not available.")
 
55
  return data
56
 
57
  # Step 3: Analyze Sentiment using FinBERT
 
45
  def add_technical_indicators(data):
46
  print("Adding technical indicators...")
47
  data['RSI'] = ta.rsi(data['Close'], length=14)
48
+
49
+ # Calculate MACD and select only the MACD line
50
+ macd = ta.macd(data['Close'], fast=12, slow=26)
51
+ if MACD_12_26_9' in macd.columns:
52
+ data['MACD'] = macd['MACD_12_26_9']
53
+ else:
54
+ print("MACD data not available.")
55
+
56
+
57
+ # Add Bollinger Bands and handle potential missing columns
58
+ bbands = ta.bbands(data['Close'])
59
  if 'BBU_20_2.0' in bbands.columns:
60
  data['BB_upper'], data['BB_middle'], data['BB_lower'] = bbands['BBU_20_2.0'], bbands['BBM_20_2.0'], bbands['BBL_20_2.0']
61
  else:
62
  print("Bollinger Bands data not available.")
63
+
64
  return data
65
 
66
  # Step 3: Analyze Sentiment using FinBERT