Update pages/2_Data_CLeaning_and_Preprocessing.py

pages/2_Data_CLeaning_and_Preprocessing.py

@@ -102,10 +102,25 @@ 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'})
-            # Correct Price Column values (round off to 2 decimal places)
-            data['Price'] = data['Price'].apply(lambda x: round(x,2))
-            print(data.head())
+            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: