Update app.py

app.py

@@ -43,14 +43,18 @@ data_load_state.text("Loading SPY data...done!")
 
 # Calculate technical indicators: MACD and RSI
 macd_indicator = MACD(close=data['Close'])
+macd_val = macd_indicator.macd()
+macd_signal_val = macd_indicator.macd_signal()
 
-# Convert outputs to 1-dimensional lists
-macd_values = np.array(macd_indicator.macd()).flatten().tolist()
-macd_signal_values = np.array(macd_indicator.macd_signal()).flatten().tolist()
+# If the returned MACD is a DataFrame, extract the first column.
+if isinstance(macd_val, pd.DataFrame):
+    macd_val = macd_val.iloc[:, 0]
+if isinstance(macd_signal_val, pd.DataFrame):
+    macd_signal_val = macd_signal_val.iloc[:, 0]
 
-# Create the Series using the lists
-data['MACD'] = pd.Series(macd_values, index=data.index)
-data['MACD_signal'] = pd.Series(macd_signal_values, index=data.index)
+# Assign the MACD values to the DataFrame.
+data['MACD'] = macd_val
+data['MACD_signal'] = macd_signal_val
 
 rsi_indicator = RSIIndicator(close=data['Close'], window=14)
 data['RSI'] = rsi_indicator.rsi()
@@ -200,4 +204,4 @@ st.write(f"**Return:** {return_pct:.2f}%")
 
 st.markdown("""
 *This extended demo is for educational purposes only and does not constitute financial advice. Always test your strategies extensively before trading with real money.*
-""")
+""")