Update app.py

app.py

@@ -38,6 +38,8 @@ finbert_model = AutoModelForSequenceClassification.from_pretrained(FINBERT_MODEL
 def fetch_stock_data(symbol, start_date, end_date):
     print("Fetching stock data...")
     data = yf.download(symbol, start=start_date, end=end_date)
+    if data.empty:
+        raise ValueError("Stock data could not be fetched. Please check the symbol and date range.")
     data = add_technical_indicators(data)
     return data
 
@@ -48,7 +50,7 @@ def add_technical_indicators(data):
 
     # Calculate MACD  and select only the MACD line
     macd = ta.macd(data['Close'], fast=12, slow=26)
-    if 'MACD_12_26_9' in macd.columns:
+    if macd is not None and 'MACD_12_26_9' in macd.columns:
         data['MACD'] = macd['MACD_12_26_9']
     else:
         print("MACD data not available.")
@@ -62,8 +64,8 @@ def add_technical_indicators(data):
         data['BB_lower'] = bbands.get('BBL_20_2.0')
 
         # Ensure the columns are added only if they exist
-        if data[['BB_upper', 'BB_middle', 'BB_lower']].isnull().all().all():
-            print("Bollinger Bands data is incomplete or not available.")
+        if data[['BB_upper', 'BB_middle', 'BB_lower']].isnull().any().any():
+            print("Bollinger Bands data is incomplete or has NaNs. Dropping these columns.")
             data.drop(columns=['BB_upper', 'BB_middle', 'BB_lower'], inplace=True, errors='ignore')
     else:
         print("Bollinger Bands data not available.")
@@ -103,9 +105,10 @@ def execute_trade(signal, symbol='Genting Bhd', qty=1):
 def plot_klci(data):
     plt.figure(figsize=(14, 7))
     plt.plot(data['Close'], label='KLCI Close Price', color='blue')
-    plt.plot(data['BB_upper'], label='Bollinger Upper Band', color='red')
-    plt.plot(data['BB_middle'], label='Bollinger Middle Band', color='green')
-    plt.plot(data['BB_lower'], label='Bollinger Lower Band', color='red')
+    if 'BB_upper' in data.columns and 'BB_middle' in data.columns and 'BB_lower' in data.columns:
+        plt.plot(data['BB_upper'], label='Bollinger Upper Band', color='red')
+        plt.plot(data['BB_middle'], label='Bollinger Middle Band', color='green')
+        plt.plot(data['BB_lower'], label='Bollinger Lower Band', color='red')
     plt.title('KLCI with Technical Indicators')
     plt.legend()
     plt.show()