Update app.py

app.py

@@ -162,23 +162,22 @@ def main():
                 st.write()
 
             # Calculate correlation
-            if warehouse_df[warehouse_column].dtype != "object" and industry_df[industry_column].dtype != "object":
-                correlation = warehouse_df[warehouse_column].corr(industry_df[industry_column])
-                st.header("Correlation")
-                st.write(f"The correlation between {warehouse_column} in warehouse item stocks and {industry_column} in industry item stocks is: {correlation}")
-                st.write()
+            correlation = warehouse_df[warehouse_column].corr(industry_df[industry_column])
+            st.header("Correlation")
+            st.write(f"The correlation between {warehouse_column} in warehouse item stocks and {industry_column} in industry item stocks is: {correlation}")
+            st.write()
 
 
-            # Show correlation plot for each dataset
-            if st.button("Correlation for each dataset"):
+        # Show correlation plot for each dataset
+        if st.button("Correlation for each dataset"):
             
-                st.subheader("Correlation Plot for Warehouse Dataset")
-                warehouse_corr_plot = plot_correlation(warehouse_df, warehouse_column)
-                st.pyplot(warehouse_corr_plot)
+            st.subheader("Correlation Plot for Warehouse Dataset")
+            warehouse_corr_plot = plot_correlation(warehouse_df, warehouse_column)
+            st.pyplot(warehouse_corr_plot)
                 
-                st.subheader("Correlation Plot for Industry Dataset")
-                industry_corr_plot = plot_correlation(industry_df, industry_column)
-                st.pyplot(industry_corr_plot)
+            st.subheader("Correlation Plot for Industry Dataset")
+            industry_corr_plot = plot_correlation(industry_df, industry_column)
+            st.pyplot(industry_corr_plot)
 
 if __name__ == "__main__":
     main()