Update app.py

app.py

@@ -118,6 +118,11 @@ def main():
     try:
         # 1. Fetch and Prepare Data
         klci_data = fetch_stock_data(SYMBOL, START_DATE, END_DATE)
+
+        # Ensure columns for Bollinger Bands exist before referencing them
+        bb_columns = ['BB_upper', 'BB_middle', 'BB_lower']
+        available_columns = [col for col in bb_columns if col in klci_data.columns]
+
     
         # 2. Run FinBERT Sentiment Analysis
         sample_text = "The market sentiment is bullish for KLCI."  # Example text
@@ -129,7 +134,11 @@ def main():
         available_columns = [col for col in indicators if col in klci_data.columns]
         indicator_data = klci_data[available_columns].iloc[-1].to_dict()
 
-        prompt = f"The market trend for KLCI with sentiment {sentiment} and indicators {klci_data[['RSI', 'MACD', 'BB_upper', 'BB_middle', 'BB_lower']].iloc[-1].to_dict()}"
+        if available_columns:
+            prompt = f"The market trend for KLCI with sentiment {sentiment} and indicators {klci_data[['RSI', 'MACD'] + available_columns].iloc[-1].to_dict()}"
+        else:
+            prompt = f"The market trend for KLCI with sentiment {sentiment} and indicators {klci_data[['RSI', 'MACD']].iloc[-1].to_dict()}"
+         
         prediction = generate_prediction(prompt)
         print(f"Dolly Prediction: {prediction}")
     
@@ -142,7 +151,8 @@ def main():
             print("No clear trade signal from prediction.")
     
         # 5. Plot KLCI Data with Indicators
-        plot_klci(klci_data)
+        if available_columns:
+            plot_klci(klci_data)
 
     except Exception as e:
         print (f"An error occured: {e}")