Update app.py

app.py

@@ -113,4 +113,18 @@ import matplotlib.pyplot as plt
 fig, ax = plt.subplots()
 ax.scatter(df['column_x'], df['column_y'])
 st.pyplot(fig)
+import seaborn as sns
+import matplotlib.pyplot as plt
+
+# Generate a correlation matrix
+corr = df.corr()
+
+# Create a heatmap
+plt.figure(figsize=(10, 8))
+sns.heatmap(corr, annot=True, cmap='coolwarm', fmt='.2f')
+plt.title('Correlation Heatmap')
+
+# Display the heatmap
+plt.show()
+