Update dataset_import.py

dataset_import.py

@@ -8,7 +8,7 @@ def download_dataset():
         os.makedirs('./data')
 
     # Download the dataset using Kaggle API
-    os.system("kaggle datasets download -d mohaideenabuthahir/analyzeyt")
+    os.system("kaggle datasets download -d mohaideenabuthahir/analyzeyt -p ./data --unzip")
 
 def load_data():
     # Ensure dataset is downloaded
@@ -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)
 
-    # 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()
+    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