Update dataset_import.py

dataset_import.py

@@ -2,10 +2,17 @@ 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")
+def download_dataset():
+    # Ensure the data directory exists
+    if not os.path.exists('./data'):
+        os.makedirs('./data')
+
+    # Download the dataset using Kaggle API
+    os.system("kaggle datasets download -d mohaideenabuthahir/analyzeyt -p ./data --unzip")
+
+def load_data():
+    # Ensure dataset is downloaded
+    download_dataset()
 
     # List available files in the data directory
     available_files = [f for f in os.listdir('./data') if f.endswith('.csv')]
@@ -19,10 +26,8 @@ def load_data(dataset_url):
 
 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())
+    data = load_data()
+    st.write(data.head())
 
 if __name__ == "__main__":
     main()