Create dataset_import.py

dataset_import.py

@@ -0,0 +1,28 @@
+import os
+import pandas as pd
+import streamlit as st
+
+def load_data(dataset_url):
+    # Download the dataset from the Kaggle URL provided
+    dataset_name = dataset_url.split('/')[-1]
+    os.system(f"kaggle datasets download -d {dataset_url} -p ./data --unzip")
+
+    # List available files in the data directory
+    available_files = [f for f in os.listdir('./data') if f.endswith('.csv')]
+
+    # 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")
+    dataset_url = st.text_input("Enter the Kaggle Dataset URL")
+    if dataset_url:
+        data = load_data(dataset_url)
+        st.write(data.head())
+
+if __name__ == "__main__":
+    main()