Update main.py

main.py

@@ -38,6 +38,24 @@ def main():
             # Replace student names with initials
             df = data_processor.replace_student_names_with_initials(df)
 
+            # Date selection option
+            date_option = st.radio(
+                "Select data range:",
+                ("All Data", "Date Range")
+            )
+
+            if date_option == "Date Range":
+                # Get the min and max dates from the data
+                min_date = df['Date of Session'].min()
+                max_date = df['Date of Session'].max()
+
+                # Date input for start and end date
+                start_date = st.date_input("Start Date", min_value=min_date, max_value=max_date, value=min_date)
+                end_date = st.date_input("End Date", min_value=min_date, max_value=max_date, value=max_date)
+
+                # Filter the DataFrame based on the selected date range
+                df = df[(df['Date of Session'] >= pd.to_datetime(start_date)) & (df['Date of Session'] <= pd.to_datetime(end_date))]
+
             st.subheader("Uploaded Data")
             st.write(df)