Update app.py

app.py

@@ -10,32 +10,8 @@ from langchain.schema import Document
 from datasets import load_dataset
 import os
 
-
-# Function to load dataset into session
-def load_dataset_into_session():
-    input_option = st.sidebar.radio("Select Dataset Input:", ["Use Hugging Face Dataset", "Upload CSV File"])
-    if input_option == "Use Hugging Face Dataset":
-        dataset_name = st.sidebar.text_input("Enter Hugging Face Dataset Name:", value="HUPD/hupd")
-        if st.sidebar.button("Load Dataset"):
-            try:
-                # Ensure `uniform_split=True` is provided for compatibility
-                dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True, uniform_split=True)
-                st.session_state.df = pd.DataFrame(dataset)
-                st.sidebar.success(f"Dataset '{dataset_name}' loaded successfully!")
-            except Exception as e:
-                st.sidebar.error(f"Error loading dataset: {e}")
-    elif input_option == "Upload CSV File":
-        uploaded_file = st.sidebar.file_uploader("Upload CSV File:", type=["csv"])
-        if uploaded_file:
-            try:
-                st.session_state.df = pd.read_csv(uploaded_file)
-                st.sidebar.success("File uploaded successfully!")
-            except Exception as e:
-                st.sidebar.error(f"Error loading file: {e}")
-
-
 # Title
-st.title("temp")
+st.title("Dataset Analysis and Visualization")
 
 # Fetch API keys from environment variables
 api_key = os.getenv("OPENAI_API_KEY")
@@ -45,8 +21,28 @@ pandasai_api_key = os.getenv("PANDASAI_API_KEY")
 if "df" not in st.session_state:
     st.session_state.df = None
 
-# Call the function to load the dataset
-load_dataset_into_session()
+# Dataset loading section
+st.subheader("Load Dataset")
+input_option = st.radio("Select Dataset Input:", ["Use Hugging Face Dataset", "Upload CSV File"])
+
+if input_option == "Use Hugging Face Dataset":
+    dataset_name = st.text_input("Enter Hugging Face Dataset Name:", value="HUPD/hupd")
+    if st.button("Load Dataset"):
+        try:
+            # Ensure `uniform_split=True` is provided for compatibility
+            dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True, uniform_split=True)
+            st.session_state.df = pd.DataFrame(dataset)
+            st.success(f"Dataset '{dataset_name}' loaded successfully!")
+        except Exception as e:
+            st.error(f"Error loading dataset: {e}")
+elif input_option == "Upload CSV File":
+    uploaded_file = st.file_uploader("Upload CSV File:", type=["csv"])
+    if uploaded_file:
+        try:
+            st.session_state.df = pd.read_csv(uploaded_file)
+            st.success("File uploaded successfully!")
+        except Exception as e:
+            st.error(f"Error loading file: {e}")
 
 # Show the loaded dataframe preview
 if st.session_state.df is not None:
@@ -119,10 +115,10 @@ if st.session_state.df is not None:
                 st.write(f"An error occurred: {str(e)}")
                 st.write("Please try rephrasing your question.")
 else:
-    st.warning("No dataset loaded. Please select a dataset input option from the sidebar.")
+    st.warning("No dataset loaded. Please select a dataset input option above.")
 
 # Error handling for missing API keys
 if not api_key:
     st.error("Missing OpenAI API Key in environment variables.")
 if not pandasai_api_key:
-    st.error("Missing PandasAI API Key in environment variables.")
+    st.error("Missing PandasAI API Key in environment variables.")