Update app.py

app.py

@@ -61,7 +61,7 @@ def create_3d_correlation_matrix(df):
                           xaxis=dict(title='Variables'),
                           yaxis=dict(title='Variables'),
                           zaxis=dict(title='Correlation')))
-    return fig.to_html(full_html=False)  # Return the HTML content of the plot
+    return fig
 
 # Gradio function
 def visualize_correlation(selected_series):
@@ -69,29 +69,40 @@ def visualize_correlation(selected_series):
     series_ids = [series for series in series_options if series_options[series] in selected_series]
     
     if not series_ids:
-        return "<p style='color:red;'>Please select at least one indicator.</p>"
+        return None, "Please select at least one indicator."
     
     # Fetch and process data
     df = fetch_fred_data(series_ids)
     if df.empty:
-        return "<p style='color:red;'>Failed to fetch data for the selected indicators.</p>"
+        return None, "Failed to fetch data for the selected indicators."
     
     standardized_df = standardize_data(df)
-    plot_html = create_3d_correlation_matrix(standardized_df)
-    return f"<div>{plot_html}</div>"
+    plot = create_3d_correlation_matrix(standardized_df)
+    return plot, None
 
-# Gradio Interface
-interface = gr.Interface(
-    fn=visualize_correlation,
-    inputs=gr.CheckboxGroup(
-        label="Select Economic Indicators",
-        choices=list(series_options.values()),  # Displayed names
-        value=[],
-        info="Choose one or more indicators to include in the correlation matrix.",
-    ),
-    outputs=gr.HTML(label="3D Correlation Matrix"),
-    title="3D Correlation Matrix Visualization with FRED Data"
-)
+# Gradio Blocks Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 3D Correlation Matrix Visualization with FRED Data")
+    
+    with gr.Row():
+        with gr.Column():
+            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.",
+            )
+            submit_button = gr.Button("Generate Matrix")
+        
+        with gr.Column():
+            plot_output = gr.Plot(label="3D Correlation Matrix")
+            error_message = gr.Markdown("", visible=False)
+    
+    # Event handler
+    submit_button.click(
+        fn=visualize_correlation,
+        inputs=[series_selector],
+        outputs=[plot_output, error_message],
+    )
 
 # Launch the app
-interface.launch(debug=True)
+demo.launch(debug=True)