Update pages/2_Data Cleaning and Processing .py

pages/2_Data Cleaning and Processing .py

@@ -56,13 +56,14 @@ if df is not None:
 
     # Cleaning the 'Location' column
     st.subheader("Cleaning the 'Location' Column:")
-    df['Location'] = df['Location'].apply(lambda x: re.findall(r'\w+', x))
+    df['Location'] = df['Location'].apply(lambda x: re.findall(r'\w+', str(x)) if isinstance(x, str) else [])
     df['Location'] = df['Location'].apply(lambda x: ' '.join(x))
     df['Location'] = df['Location'].apply(lambda x: re.sub(r"\d+", r",\g<0>", x))
-
+    
     st.write("Cleaned 'Location' Column:")
     st.write(df[['Location']].head())
 
+
     # Cleaning the 'Discount' column
     st.subheader("Cleaning the 'Discount' Column:")
     def f(x):