Update pages/2_Data_CLeaning_and_Preprocessing.py

pages/2_Data_CLeaning_and_Preprocessing.py

@@ -10,7 +10,6 @@ st.markdown("""
 ### Perform EDA and Clean Data
 Upload a CSV file to begin. This app will provide basic insights into the dataset, 
 highlight missing values, and visualize numeric and categorical columns.
-
 ---
 """)
 
@@ -89,6 +88,26 @@ if uploaded_file is not None:
                 sns.heatmap(corr_matrix, annot=True, cmap='coolwarm', ax=ax)
                 st.pyplot(fig)
 
+            # Check the columns before renaming
+            st.write("### Dataset Columns:")
+            st.write(data.columns)
+
+            # Renaming columns if they exist
+            if 'ProductCategory' in data.columns and 'ProductBrand' in data.columns and 'ProductPrice' in data.columns:
+                data = data.rename(columns={'ProductCategory': 'Category', 'ProductBrand': 'Brand', 'ProductPrice': 'Price'})
+                st.success("Columns renamed successfully!")
+            else:
+                st.warning("Columns 'ProductCategory', 'ProductBrand', or 'ProductPrice' not found in the dataset.")
+
+            # Now check if 'Category' exists and plot
+            if 'Category' in data.columns:
+                st.write("### Bar Plot for Category")
+                fig, ax = plt.subplots()
+                sns.countplot(x='Category', data=data, palette='viridis', ax=ax)
+                st.pyplot(fig)
+            else:
+                st.warning("'Category' column not found for plotting.")
+
             # Data Cleaning Section
             st.write("### Cleaned Dataset")
             cleaned_data = data.drop_duplicates()
@@ -102,25 +121,7 @@ if uploaded_file is not None:
                 file_name="cleaned_dataset.csv",
                 mime="text/csv"
             )
-            # Check the columns before renaming
-        st.write("### Dataset Columns:")
-        st.write(data.columns)
-        
-        # Renaming columns if they exist
-        if 'ProductCategory' in data.columns and 'ProductBrand' in data.columns and 'ProductPrice' in data.columns:
-            data = data.rename(columns={'ProductCategory': 'Category', 'ProductBrand': 'Brand', 'ProductPrice': 'Price'})
-            st.success("Columns renamed successfully!")
-        else:
-            st.warning("Columns 'ProductCategory', 'ProductBrand', or 'ProductPrice' not found in the dataset.")
-        
-        # Now check if 'Category' exists and plot
-        if 'Category' in data.columns:
-            st.write("### Bar Plot for Category")
-            fig, ax = plt.subplots()
-            sns.countplot(x='Category', data=data, palette='viridis', ax=ax)
-            st.pyplot(fig)
-        else:
-            st.warning("'Category' column not found for plotting.")
+
         except pd.errors.EmptyDataError:
             st.error("The uploaded CSV file is empty. Please upload a valid file.")
         except pd.errors.ParserError: