Added checkboxes for cleaning options.

app.py

@@ -15,14 +15,19 @@ uploaded_file = st.file_uploader(f"Upload dataset:", type=modules.SUPPORTED_TYPE
 df, (filename, extension), result = modules.load_file(uploaded_file)
 st.text(result)
 
-st.title("Before:")
-st.dataframe(df)
+if df is not None:
+    st.title("Before:")
+    st.dataframe(df)
 
-st.title("After:")
-df = modules.data_cleaner(df)
-st.dataframe(df)
+    drop_missing = st.checkbox("Drop Missing", True)
+    remove_duplicates = st.checkbox("Remove Duplicates", True)
 
-st.download_button("Download cleaned data", modules.create_file(df, extension), file_name=filename)
+    st.title("After:")
+    df = modules.data_cleaner(df, drop_missing, remove_duplicates)
+    st.dataframe(df)
+
+    download_file = modules.create_file(df, extension)
+    st.download_button("Download cleaned data", download_file, file_name=filename)
 
 st.markdown("---")
 st.text("Created by team #2hack2furious for the hackthethreat2023")

modules.py

@@ -13,7 +13,7 @@ def load_file(file):
     @rtype: tuple
     @return: A tuple of format (pd.DataFrame, (str, str), str).
     """
-    df = pd.DataFrame()  
+    df = None
 
     if file is None: return df, ("", ""), ""