debut merge

frontend/exploration.py

@@ -9,13 +9,24 @@ st.set_page_config(
     layout="wide"
 )
 
+st.title("Home")
+
 ### Exploration
 uploaded_file = st.file_uploader("Upload your CSV file", type=["csv"])
 
-if uploaded_file:
+if uploaded_file is not None:
-    data = pd.read_csv(uploaded_file)
+    st.session_state.data = pd.read_csv(uploaded_file)
     st.success("File loaded successfully!")
 
+if "data" not in st.session_state:
+    st.session_state.data = None
+
+
+if st.session_state.data is not None:
+    data = st.session_state.data
+    st.write(data.head(10))
+    st.write(data.tail(10))
+
     st.header("Data Preview")
     
     st.subheader("First 5 Rows")
@@ -39,26 +50,4 @@ if uploaded_file:
     
     st.subheader("Statistical Summary")
     st.write(data.describe())
-    
+    
-    ### Visualization
-
-    st.header("Data Visualization")
-
-    st.subheader("Histogram")
-    column_to_plot = st.selectbox("Select Column for Histogram", data.columns)
-    if column_to_plot:
-        fig, ax = plt.subplots()
-        ax.hist(data[column_to_plot].dropna(), bins=20, edgecolor='k')
-        ax.set_title(f'Histogram of {column_to_plot}')
-        ax.set_xlabel(column_to_plot)
-        ax.set_ylabel('Frequency')
-        st.pyplot(fig)
-    
-    st.subheader("Boxplot")
-    dataNumeric = data.select_dtypes(include='number')
-    column_to_plot = st.selectbox("Select Column for Boxplot", dataNumeric.columns)
-    if column_to_plot:
-        fig, ax = plt.subplots()
-        sns.boxplot(data=data, x=column_to_plot, ax=ax)
-        ax.set_title(f'Boxplot of {column_to_plot}')
-        st.pyplot(fig)

frontend/pages/visualization.py

@@ -0,0 +1,30 @@
+import streamlit as st
+import matplotlib.pyplot as plt
+import seaborn as sns
+
+st.header("Data Visualization")
+
+
+if "data" in st.session_state:
+    data = st.session_state.data
+
+    st.subheader("Histogram")
+    column_to_plot = st.selectbox("Select Column for Histogram", data.columns)
+    if column_to_plot:
+        fig, ax = plt.subplots()
+        ax.hist(data[column_to_plot].dropna(), bins=20, edgecolor='k')
+        ax.set_title(f"Histogram of {column_to_plot}")
+        ax.set_xlabel(column_to_plot)
+        ax.set_ylabel("Frequency")
+        st.pyplot(fig)
+    
+    st.subheader("Boxplot")
+    dataNumeric = data.select_dtypes(include="number")
+    column_to_plot = st.selectbox("Select Column for Boxplot", dataNumeric.columns)
+    if column_to_plot:
+        fig, ax = plt.subplots()
+        sns.boxplot(data=data, x=column_to_plot, ax=ax)
+        ax.set_title(f"Boxplot of {column_to_plot}")
+        st.pyplot(fig)
+else:
+    st.error("file not loaded")