Update app.py

app.py

@@ -161,11 +161,13 @@ def main():
                 st.write(f"____________________")
                 st.write()
 
-            # Calculate correlation
-            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()
+            if warehouse_df[warehouse_column].dtype != "object" and industry_df[industry_column].dtype != "object":
+
+                # Calculate correlation
+                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