Update dataset_import.py

dataset_import.py

@@ -17,17 +17,17 @@ def load_data():
     # List available files in the data directory
     available_files = [f for f in os.listdir('./data') if f.endswith('.csv')]
 
-    if len(available_files) == 0:
-        st.error("No CSV files found in the dataset folder.")
-        return None
-
     # Ask user to select which file to load
     selected_file = st.selectbox('Select a file to load', available_files)
 
-    if selected_file:
-        # Load the selected dataset
-        data = pd.read_csv(f'./data/{selected_file}')
-        return data
-    else:
-        st.error("Please select a valid file.")
-        return None
+    # Load the selected dataset
+    data = pd.read_csv(f'./data/{selected_file}')
+    return data
+
+def main():
+    st.title("Kaggle Dataset Loader")
+    data = load_data()
+    st.write(data.head())
+
+if __name__ == "__main__":
+    main()