Update app.py

app.py

@@ -63,9 +63,7 @@ if upload_file is not None:
             else:
                 new_columns = [f'column_{i+1}' for i in range(len(df.columns))]
                 df.columns = new_columns
-                st.dataframe(df, key="csv_dataframe")
-                st.write("_number of rows_", df.shape[0])
-                st.write("_number of columns_", df.shape[1])
+                
                 all_columns = df.columns.tolist()
                 st.subheader("Select columns for the Tree Map", divider="blue")
                 parent_column = st.selectbox("Select the parent column:", all_columns)
@@ -104,9 +102,7 @@ if upload_file is not None:
             else:
                 new_columns = [f'column_{i+1}' for i in range(len(df.columns))]
                 df.columns = new_columns
-                st.dataframe(df, key="csv_dataframe")
-                st.write("_number of rows_", df.shape[0])
-                st.write("_number of columns_", df.shape[1])
+                
                 all_columns = df.columns.tolist()
                 st.subheader("Select columns for the Tree Map", divider="blue")
                 parent_column = st.selectbox("Select the parent column:", all_columns)
@@ -143,6 +139,49 @@ if upload_file is not None:
  
 st.divider()
 
+if upload_file is not None:
+    file_extension = upload_file.name.split('.')[-1].lower()
+    if file_extension == 'csv':
+        try:
+            df = pd.read_csv(upload_file, na_filter=False)
+            if df.isnull().values.any():
+                st.error("Error: The CSV file contains missing values.")
+                st.stop()
+            else:
+                st.dataframe(df, key="csv_dataframe")
+                st.write("_number of rows_", df.shape[0])
+                st.write("_number of columns_", df.shape[1])
+                st.session_state.df = df  
+        except pd.errors.ParserError:
+            st.error("Error: The CSV file is not readable or is incorrectly formatted.")
+            st.stop()
+        except UnicodeDecodeError:
+            st.error("Error: The CSV file could not be decoded.")
+            st.stop()
+        except Exception as e:
+            st.error(f"An unexpected error occurred while reading CSV: {e}")
+            st.stop()
+    elif file_extension == 'xlsx':
+        try:
+            df = pd.read_excel(upload_file, na_filter=False)
+            
+            if df.isnull().values.any():
+                st.error("Error: The Excel file contains missing values.")
+                st.stop()
+            else:
+                st.dataframe(df, key="excel_dataframe")
+                st.write("_number of rows_", df.shape[0])
+                st.write("_number of columns_", df.shape[1])
+                st.session_state.df = df  
+        except ValueError:
+            st.error("Error: The Excel file is not readable or is incorrectly formatted.")
+            st.stop()
+        except Exception as e:
+            st.error(f"An unexpected error occurred while reading Excel: {e}")
+            st.stop()
+    else:
+        st.warning("Unsupported file type.")
+        st.stop()
 
 def clear_question():
     st.session_state["question"] = ""