Update app.py

app.py

@@ -56,11 +56,18 @@ def add_technical_indicators(data):
 
     # Add Bollinger Bands and handle potential missing columns
     bbands = ta.bbands(data['Close'])
-    if 'BBU_20_2.0' in bbands.columns:
-        data['BB_upper'], data['BB_middle'], data['BB_lower'] = bbands['BBU_20_2.0'], bbands['BBM_20_2.0'], bbands['BBL_20_2.0']
+    if bbands is not None:
+        data['BB_upper'] = bbands.get('BBU_20_2.0')
+        data['BB_middle'] = bbands.get('BBM_20_2.0')
+        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.")
+            data.drop(columns=['BB_upper', 'BB_middle', 'BB_lower'], inplace=True, errors='ignore')
     else:
         print("Bollinger Bands data not available.")
-    
+
     return data
 
 # Step 3: Analyze Sentiment using FinBERT