Update app.py

app.py

@@ -3,7 +3,6 @@ import pandas_datareader.data as web
 import yfinance as yf
 import datetime
 import plotly.graph_objs as go
-import pandas as pd
 
 # Set up the layout
 st.set_page_config(layout="wide")
@@ -14,12 +13,11 @@ st.markdown(
     <style>
     /* Increase the width of the sidebar */
     .css-1d391kg {
-        width: 350px;  /* Adjust this value to your desired width */
+        width: 350px;
     }
-
     /* Ensure main content is still centered properly */
     .css-1d391kg ~ .css-1d391kg {
-        left: 350px;  /* This value should match the width above */
+        left: 350px;
     }
     </style>
     """,
@@ -49,8 +47,8 @@ st.sidebar.header("Select Indicators")
 with st.sidebar.expander("Indicators", expanded=True):
     indicators = {
         'Sahm Recession Indicator': 'SAHMREALTIME',
-        'U.S. Recession Probabilities': 'Recession_Probabilities',
-        'Yield Spread (10Y - 2Y)': 'Yield_Spread',
+        'U.S. Recession Probabilities': 'RECPROUSM156N',  # Corrected to actual FRED series ID
+        'Yield Spread (10Y - 2Y)': 'Yield_Spread',  # Kept as placeholder, calculated later
         'Stock Market (S&P 500)': 'SP500',
         'VIX': 'VIX',
         'Treasury Rates': ('GS10', 'DGS2', 'DGS1MO', 'TB3MS'),
@@ -95,9 +93,9 @@ crash_periods = {
 if st.sidebar.button('Run Analysis'):
     # Initialize combined_data as an empty DataFrame
     combined_data = pd.DataFrame()
-
-    # Fetch FRED data with error handling
     fred_data = {}
+
+    # Fetch FRED data with error handling, excluding derived columns
     for key, column in indicators.items():
         if selected_indicators[key]:
             if isinstance(column, tuple):
@@ -108,7 +106,7 @@ if st.sidebar.button('Run Analysis'):
                         except Exception as e:
                             st.warning(f"Failed to fetch {col} from FRED: {e}")
                             fred_data[col] = pd.DataFrame()
-            else:
+            elif column != 'Yield_Spread':  # Skip Yield_Spread as it's calculated
                 try:
                     fred_data[column] = web.DataReader(column, 'fred', start_date, end_date)
                 except Exception as e:
@@ -138,11 +136,6 @@ if st.sidebar.button('Run Analysis'):
             st.warning(f"Failed to fetch VIX data: {e}")
             fred_data['VIX'] = pd.DataFrame()
 
-    # Rename recession probabilities for clarity
-    if 'RECPROUSM156N' in fred_data:
-        fred_data['RECPROUSM156N'].columns = ['Recession_Probabilities']
-        fred_data['Recession_Probabilities'] = fred_data.pop('RECPROUSM156N')
-
     # Combine all FRED and Yahoo Finance data
     if fred_data:
         combined_data = pd.concat(fred_data.values(), axis=1, join='outer')
@@ -190,7 +183,6 @@ if st.sidebar.button('Run Analysis'):
                 
                 if isinstance(column, tuple):
                     if column == ('INDPRO', 'INDPRO_PCT') and 'INDPRO' in combined_data:
-                        # Plot industrial production and its percentage change on dual y-axes
                         fig.add_trace(go.Scatter(x=combined_data.index, y=combined_data['INDPRO'], mode='lines', name='Industrial Production'))
                         if 'INDPRO_PCT' in combined_data:
                             fig.add_trace(go.Scatter(x=combined_data.index, y=combined_data['INDPRO_PCT'], mode='lines', name='Industrial Production % Change', yaxis='y2'))
@@ -209,7 +201,6 @@ if st.sidebar.button('Run Analysis'):
                             )
                         )
                     elif column == ('CPIAUCSL', 'CPIAUCSL_PCT') and 'CPIAUCSL' in combined_data:
-                        # Plot inflation and its percentage change on dual y-axes
                         fig.add_trace(go.Scatter(x=combined_data.index, y=combined_data['CPIAUCSL'], mode='lines', name='Inflation (CPI)'))
                         if 'CPIAUCSL_PCT' in combined_data:
                             fig.add_trace(go.Scatter(x=combined_data.index, y=combined_data['CPIAUCSL_PCT'], mode='lines', name='Inflation % Change', yaxis='y2'))
@@ -228,7 +219,6 @@ if st.sidebar.button('Run Analysis'):
                             )
                         )
                     elif column == ('GS10', 'DGS2', 'DGS1MO', 'TB3MS'):
-                        # Plot multiple Treasury rates in the same subplot
                         for col in column:
                             if col in combined_data:
                                 fig.add_trace(go.Scatter(x=combined_data.index, y=combined_data[col], mode='lines', name=col))
@@ -244,7 +234,6 @@ if st.sidebar.button('Run Analysis'):
                 else:
                     if column in combined_data:
                         fig.add_trace(go.Scatter(x=combined_data.index, y=combined_data[column], mode='lines', name=key))
-                        # Add horizontal threshold line for Sahm Recession Indicator
                         if column == 'SAHMREALTIME':
                             fig.add_hline(y=0.5, line=dict(color="red", dash="dash"), annotation_text="Recession Threshold", annotation_position="bottom right")