Update app.py

app.py

@@ -9,7 +9,6 @@ from numpy import average as npAverage, nan as npNaN, log as npLog, power as npP
 from pandas_ta.utils import get_offset, verify_series
 from datetime import datetime
 from matplotlib.dates import date2num
-from datetime import datetime
 
 st.set_page_config(layout="wide")
 
@@ -36,7 +35,13 @@ with st.sidebar.expander("How to Use:", expanded=False):
 # Function to fetch data
 @st.cache_data
 def get_data(ticker, start_date, end_date):
-    data = yf.download(ticker, start=start_date, end=end_date)
+    data = yf.download(ticker, start=start_date, end=end_date, auto_adjust=False)
+    if isinstance(data.columns, pd.MultiIndex):
+        data.columns = data.columns.get_level_values(0)
+    if data.empty:
+        raise ValueError(f"No data retrieved for {ticker}")
+    if len(data) < 512:  # Ensure enough data for largest possible Rainbow MA period
+        raise ValueError(f"Insufficient data points for {ticker}. Need at least 512 days.")
     return data
 
 # Function to create the base plot with the stock price
@@ -459,29 +464,28 @@ end_date = st.sidebar.date_input(
 
 # Fetch Data button
 if st.sidebar.button('Fetch Data'):
-    # Only fetch if the ticker or date range has changed
-    if (
-        ticker_symbol != st.session_state.get('ticker_symbol') or 
-        start_date != st.session_state.get('start_date') or 
-        end_date != st.session_state.get('end_date')
-    ):
-        data = get_data(ticker_symbol, start_date, end_date)
-        st.session_state['data'] = data
-        st.session_state['ticker_symbol'] = ticker_symbol
-        st.session_state['start_date'] = start_date
-        st.session_state['end_date'] = end_date
-        st.session_state['price_plot'] = create_price_plot(data, ticker_symbol)
-        st.session_state['current_fig'] = st.session_state['price_plot']
+    try:
+        # Only fetch if the ticker or date range has changed
+        if (
+            ticker_symbol != st.session_state.get('ticker_symbol') or 
+            start_date != st.session_state.get('start_date') or 
+            end_date != st.session_state.get('end_date')
+        ):
+            data = get_data(ticker_symbol, start_date, end_date)
+            st.session_state['data'] = data
+            st.session_state['ticker_symbol'] = ticker_symbol
+            st.session_state['start_date'] = start_date
+            st.session_state['end_date'] = end_date
+            st.session_state['price_plot'] = create_price_plot(data, ticker_symbol)
+            st.session_state['current_fig'] = st.session_state['price_plot']
+    except Exception as e:
+        st.error(f"An error occurred while fetching data: {e}")
 
 # Check if data is fetched and stored in session state
 if 'data' in st.session_state:
     data = st.session_state['data']
     
     # Moving average method selection
-    #st.sidebar.header("Moving Average Methods")
-
-
-
     # SMA with tooltip
     with st.sidebar.expander("Simple Moving Average", expanded=False):
         use_sma = st.checkbox(
@@ -1079,21 +1083,6 @@ if 'data' in st.session_state:
             help="Specify the period (in days) for the FMA."
         )    
 
-
-    # Anchored Moving Average (AMA) with tooltip
-    #use_ama = st.sidebar.checkbox(
-    #    'Anchored Moving Average (AMA)', 
-    #    value=st.session_state.get('use_ama', False),
-    #    help="Select to apply Anchored Moving Average (AMA) to the stock price."
-    #)
-    #ama_anchor_date = st.sidebar.date_input(
-    #    'AMA Anchor Date',
-    #    value=pd.to_datetime(st.session_state.get('ama_anchor_date', '2021-01-01')),
-    #    disabled=not use_ama,
-    #    help="Select the anchor date for the AMA."
-    #)
-
-
     # Grid toggle with tooltip
     show_grid = st.sidebar.checkbox(
         "Show Grid", 
@@ -1176,11 +1165,8 @@ if 'data' in st.session_state:
         st.session_state['use_cma'] = use_cma
         st.session_state['use_mcginley_dynamic'] = use_mcginley_dynamic
         st.session_state['mcginley_dynamic_period'] = mcginley_dynamic_period
-        #st.session_state['use_ama'] = use_ama
-        #st.session_state['ama_anchor_date'] = ama_anchor_date
         st.session_state['use_fma'] = use_fma
         st.session_state['fma_period'] = fma_period
-        # (Save all previous moving average settings here)
         
         # Start with the base price plot
         fig = go.Figure(data=st.session_state['price_plot'].data)
@@ -1205,12 +1191,6 @@ if 'data' in st.session_state:
             st.session_state['McGinley_Dynamic'] = calculate_mcginley_dynamic(data['Close'].tolist(), mcginley_dynamic_period)
             fig.add_trace(go.Scatter(x=data.index, y=st.session_state['McGinley_Dynamic'], mode='lines', name=f'McGinley Dynamic (n={mcginley_dynamic_period})', line=dict(dash='dash', color='orange')))
 
-        # Add AMA if selected
-        #if use_ama:
-        #    st.session_state['AMA'] = calculate_AMA(data['Close'].tolist(), ama_anchor_date, data)
-        #    fig.add_trace(go.Scatter(x=data.index, y=st.session_state['AMA'], mode='lines', name=f'Anchored MA (Anchor Date={ama_anchor_date})', line=dict(dash='dash', color='red')))
-        #    fig.add_shape(type="line", x0=ama_anchor_date, y0=data['Close'].min(), x1=ama_anchor_date, y1=data['Close'].max(), line=dict(color="blue", width=2, dash="dash"))
-
         # Add FMA if selected
         if use_fma:
             st.session_state['FMA'] = filtered_moving_average(data['Close'].values, fma_period)
@@ -1385,7 +1365,6 @@ if 'data' in st.session_state:
             st.session_state['PWMA'] = np.concatenate([np.array([np.nan]*(pwma_period-1)), pwma_values])
             fig.add_trace(go.Scatter(x=data.index, y=st.session_state['PWMA'], mode='lines', name=f'PWMA (n={pwma_period})', line=dict(dash='dash', color='red')))
 
-
         # Update layout with grid toggle
         fig.update_layout(
             title=f'{ticker_symbol} Stock Price with Moving Averages',
@@ -1404,9 +1383,6 @@ if 'data' in st.session_state:
 if 'current_fig' in st.session_state:
     st.plotly_chart(st.session_state['current_fig'], use_container_width=True)
 
-
-
-
 st.markdown(
     """
     <style>
@@ -1419,11 +1395,10 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-
 hide_streamlit_style = """
 <style>
 #MainMenu {visibility: hidden;}
 footer {visibility: hidden;}
 </style>
 """
-st.markdown(hide_streamlit_style, unsafe_allow_html=True) 
+st.markdown(hide_streamlit_style, unsafe_allow_html=True)