Update lab/interim.py

lab/interim.py

@@ -35,7 +35,7 @@ llm = initialize_llm(model_choice)
 def load_dataset_into_session():
     input_option = st.radio(
         "Select Dataset Input:",
-        ["Use Repo Directory Dataset", "Use Hugging Face Dataset", "Upload CSV File"], index=0, horizontal=True
+        ["Use Repo Directory Dataset", "Use Hugging Face Dataset", "Upload CSV File"], index=1, horizontal=True
     )
 
     # Option 1: Load dataset from the repo directory
@@ -53,14 +53,13 @@ def load_dataset_into_session():
         dataset_name = st.text_input(
             "Enter Hugging Face Dataset Name:", value="HUPD/hupd"
         )
-        if st.button("Load Hugging Face Dataset"):
+        if st.button("Load Dataset"):
             try:
-                dataset = load_dataset(dataset_name, name="all", split="train", trust_remote_code=True, uniform_split=True)
+                dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True, uniform_split=True)
                 if hasattr(dataset, "to_pandas"):
                     st.session_state.df = dataset.to_pandas()
                 else:
                     st.session_state.df = pd.DataFrame(dataset)
-                st.session_state.df = validate_and_clean_dataset(st.session_state.df)
                 st.success(f"Hugging Face Dataset '{dataset_name}' loaded successfully!")
             except Exception as e:
                 st.error(f"Error loading Hugging Face dataset: {e}")
@@ -71,7 +70,6 @@ def load_dataset_into_session():
         if uploaded_file:
             try:
                 st.session_state.df = pd.read_csv(uploaded_file)
-                st.session_state.df = validate_and_clean_dataset(st.session_state.df)
                 st.success("File uploaded successfully!")
             except Exception as e:
                 st.error(f"Error reading uploaded file: {e}")
@@ -130,4 +128,5 @@ with st.sidebar:
     st.header("References:")
     st.markdown(
         "1. [Chat With Your CSV File With PandasAI - Prince Krampah](https://medium.com/aimonks/chat-with-your-csv-file-with-pandasai-22232a13c7b7)"
-    )
+    )
+