Update pages/Hotel Data.py

pages/Hotel Data.py

@@ -1,6 +1,6 @@
-import streamlit as st
-import pandas as pd
 import os
+import pandas as pd
+import streamlit as st
 
 # Define a persistent file path for the dataset
 DATA_FILE_PATH = "dataset.csv"
@@ -8,22 +8,16 @@ DATA_FILE_PATH = "dataset.csv"
 # Page Title
 st.markdown("<h1 style='text-align:center; color:yellow;'>Hotel Data Set</h1>", unsafe_allow_html=True)
 
-# Check if a dataset already exists on disk
-if os.path.exists(DATA_FILE_PATH):
-    # If dataset exists, load it
-    df = pd.read_csv(DATA_FILE_PATH)
-
-    # Display dataset
-    st.subheader("Current Dataset (Previously Uploaded):")
-    st.write(df)  # Display the first 5 rows of the dataset
-
-    st.subheader("Dataset Description:")
-    st.write(df.describe())
+# Function to load the dataset
+def load_dataset():
+    if os.path.exists(DATA_FILE_PATH):
+        return pd.read_csv(DATA_FILE_PATH)
+    else:
+        return None
 
-    st.subheader("Dataset Shape (Rows, Columns):")
-    st.write(df.shape)
-else:
-    st.info("No dataset found. Please upload a CSV file.")
+# Check if dataset is in session state or load it from disk
+if "dataset" not in st.session_state:
+    st.session_state["dataset"] = load_dataset()
 
 # File uploader widget to upload a new dataset
 uploaded_file = st.file_uploader("Choose a CSV file", type=["csv"])
@@ -35,9 +29,23 @@ if uploaded_file is not None:
     # Save the dataset permanently to disk
     df.to_csv(DATA_FILE_PATH, index=False)
 
-    # Display the uploaded dataset
-    st.subheader("Uploaded Dataset Preview:")
-    st.write(df.head())  # Display the first 5 rows
+    # Store in session state
+    st.session_state["dataset"] = df
 
     # Display success message
     st.success(f"Dataset uploaded and saved permanently as {DATA_FILE_PATH}!")
+
+# Access the dataset from session state
+df = st.session_state.get("dataset")
+
+if df is not None:
+    st.subheader("Dataset Preview:")
+    st.write(df.head())  # Display the first 5 rows
+
+    st.subheader("Dataset Description:")
+    st.write(df.describe())
+
+    st.subheader("Dataset Shape (Rows, Columns):")
+    st.write(df.shape)
+else:
+    st.info("No dataset found. Please upload a CSV file.")