Update app.py

app.py

@@ -29,9 +29,7 @@ expander.write('''
     To change the app's background color to white or black, click the three-dot menu on the right-hand side of your app, go to Settings and then Choose app theme, colors and fonts.
     
     **File Handling and Errors:**
-    The app may provide an inaccurate answer, if the information is missing from the relevant cell.
-    
-    The app may display an error message if your file has errors or date values.
+    (a) The app may provide an inaccurate answer, if the information is missing from the relevant cell. (b) The app may display an error message if your file has errors or date values.
     
     For any errors or inquiries, please contact us at [email protected]
     
@@ -52,7 +50,6 @@ max_attempts = 5
 
 
 upload_file = st.file_uploader("Upload your file. Accepted file formats include: .csv, .xlsx", type=['csv', 'xlsx'])
-
 if upload_file is not None:
     file_extension = upload_file.name.split('.')[-1].lower()
     try:
@@ -64,34 +61,38 @@ if upload_file is not None:
             st.warning("Unsupported file type.")
             st.stop()
 
+        # Convert float columns to integers if they contain whole numbers
+        for col in df_original.columns:
+            if df_original[col].dtype == 'float64':
+                if all(df_original[col] == df_original[col].astype(int)):
+                    df_original[col] = df_original[col].astype(int)
+
         if df_original.isnull().values.any():
             st.error("Error: The file contains missing values.")
             st.stop()
         else:
-            st.session_state.df_original = df_original  
+            st.session_state.df_original = df_original
 
             all_columns = df_original.columns.tolist()
             st.subheader("Select columns from your data to visualize it in a tree map", divider="blue")
             parent_column = st.selectbox("Select the parent column:", all_columns)
             value_column = st.selectbox("Select the value column:", all_columns)
-
             if parent_column and value_column:
                 if parent_column == value_column:
                     st.warning("Warning: You have selected the same column for both the parent and value column. Please select two different columns from your data.")
                 else:
-                    df_treemap = df_original.copy()  
+                    df_treemap = df_original.copy()
+
                     path_columns = [px.Constant("all"), parent_column, value_column]
                     fig = px.treemap(df_treemap,
                                      path=path_columns)
                     fig.update_layout(margin=dict(t=50, l=25, r=25, b=25))
                     st.subheader("Tree map", divider="blue")
                     st.plotly_chart(fig)
-
             st.subheader("Uploaded File", divider="blue")
             st.dataframe(df_original, key="uploaded_dataframe")
             st.write(f"_Number of rows_: {df_original.shape[0]}")
             st.write(f"_Number of columns_: {df_original.shape[1]}")
-
     except pd.errors.ParserError:
         st.error("Error: The CSV file is not readable or is incorrectly formatted.")
         st.stop()
@@ -104,8 +105,9 @@ if upload_file is not None:
     except Exception as e:
         st.error(f"An unexpected error occurred: {e}")
         st.stop()
-
 st.divider()
+
+
      
 
 def clear_question():