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
+    return fig.to_html(full_html=False)  # Return the HTML content of the plot
 
 # Gradio function
 def visualize_correlation(selected_series):
@@ -77,11 +77,8 @@ def visualize_correlation(selected_series):
         return "<p style='color:red;'>Failed to fetch data for the selected indicators.</p>"
     
     standardized_df = standardize_data(df)
-    fig = create_3d_correlation_matrix(standardized_df)
-    
-    # Save figure as HTML and return for display
-    fig.write_html("3d_correlation_matrix.html")
-    return "3d_correlation_matrix.html"
+    plot_html = create_3d_correlation_matrix(standardized_df)
+    return f"<div>{plot_html}</div>"
 
 # Gradio Interface
 interface = gr.Interface(