Spaces:
Runtime error
Runtime error
Ken Sang Tang commited on
Update app.py
Browse files
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 |
-
|
| 49 |
-
|
| 50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|