Update app.py

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 # Set FRED API Key from environment variable
 FRED_API_KEY = os.getenv('FRED_API_KEY')
 
-# Available FRED data series and their descriptive labels
+# List of FRED data series and their descriptive labels
 series_options = {
     "UNRATE": "Unemployment Rate",
     "GDP": "Gross Domestic Product",
@@ -17,13 +17,12 @@ series_options = {
     "M1SL": "M1 Money Supply",
     "M2SL": "M2 Money Supply",
     "M3SL": "M3 Money Supply",
-    # Add more series as needed
     "HOUST": "Housing Starts",
     "PCE": "Personal Consumption Expenditures",
-    "BAA10YM": "Moody's Baa Corporate Bond Yield Spread",
+    "BAA10YM": "Moody's Baa Corporate Bond Yield Spread"
 }
 
-# Function to fetch data from FRED API
+# Function to fetch data from the FRED API
 def fetch_fred_data(series_ids):
     """
     Fetches data for a list of FRED series IDs.
@@ -33,12 +32,16 @@ def fetch_fred_data(series_ids):
     for series_id in series_ids:
         response = requests.get(
             f'https://api.stlouisfed.org/fred/series/observations',
-            params={'series_id': series_id, 'api_key': FRED_API_KEY, 'file_type': 'json'}
+            params={
+                'series_id': series_id,
+                'api_key': FRED_API_KEY,
+                'file_type': 'json'
+            }
         )
         if response.status_code == 200:
             observations = response.json().get('observations', [])
             dates = [obs['date'] for obs in observations]
-            # Safely convert values to float, handling invalid entries
+            # Convert values to float, handling invalid entries
             values = [
                 float(obs['value']) if obs['value'].replace('.', '', 1).isdigit() else float('nan')
                 for obs in observations
@@ -55,22 +58,31 @@ def standardize_data(df):
     """
     return (df - df.mean()) / df.std()
 
-# Function to create 3D correlation matrix
+# Function to create a responsive 3D correlation matrix
 def create_3d_correlation_matrix(df):
+    """
+    Creates a 3D correlation matrix graph using Plotly.
+    The graph will automatically adjust its size.
+    """
     correlation_matrix = df.corr()
-    fig = go.Figure(data=[go.Surface(z=correlation_matrix.values,
-                                     x=correlation_matrix.columns,
-                                     y=correlation_matrix.index)])
-    fig.update_layout(title='3D Correlation Matrix (Standardized)',
-                      autosize=False,
-                      width=800, height=800,
-                      scene=dict(
-                          xaxis=dict(title='Variables'),
-                          yaxis=dict(title='Variables'),
-                          zaxis=dict(title='Correlation')))
+    fig = go.Figure(data=[go.Surface(
+        z=correlation_matrix.values,
+        x=correlation_matrix.columns,
+        y=correlation_matrix.index
+    )])
+    fig.update_layout(
+        title='3D Correlation Matrix (Standardized)',
+        autosize=True,  # Enables auto-resizing
+        scene=dict(
+            xaxis=dict(title='Variables'),
+            yaxis=dict(title='Variables'),
+            zaxis=dict(title='Correlation')
+        ),
+        margin=dict(l=0, r=0, t=50, b=50)  # Adjust margins for better fit
+    )
     return fig
 
-# Gradio function
+# Gradio function to handle user interaction
 def visualize_correlation(selected_series):
     # Map descriptive labels back to FRED series IDs
     series_ids = [series for series in series_options if series_options[series] in selected_series]
@@ -96,7 +108,7 @@ with gr.Blocks() as demo:
             series_selector = gr.CheckboxGroup(
                 choices=list(series_options.values()),
                 label="Select Economic Indicators",
-                info="Choose one or more indicators to include in the correlation matrix.",
+                info="Choose one or more indicators to include in the correlation matrix."
             )
             submit_button = gr.Button("Generate Matrix")
         
@@ -104,12 +116,12 @@ with gr.Blocks() as demo:
             plot_output = gr.Plot(label="3D Correlation Matrix")
             error_message = gr.Markdown("", visible=False)
     
-    # Event handler
+    # Event handler for the submit button
     submit_button.click(
         fn=visualize_correlation,
         inputs=[series_selector],
         outputs=[plot_output, error_message],
     )
 
-# Launch the app
+# Launch the Gradio app
 demo.launch(debug=True)