Update main.py

main.py

@@ -38,8 +38,8 @@ def main():
 
     if date_option == "Date Range":
         # Prompt user to enter start and end dates
-        start_date = st.date_input("Start Date").date()
-        end_date = st.date_input("End Date").date()
+        start_date = st.date_input("Start Date")
+        end_date = st.date_input("End Date")
 
         # Ensure start date is before end date
         if start_date > end_date:
@@ -63,8 +63,8 @@ def main():
             # Filter data if date range is selected
             if date_option == "Date Range":
                 # Convert start_date and end_date to datetime
-                start_date = pd.to_datetime(start_date)
-                end_date = pd.to_datetime(end_date)
+                start_date = pd.to_datetime(start_date).date()
+                end_date = pd.to_datetime(end_date).date()
 
                 # Filter the DataFrame based on the selected date range
                 df = df[(df['Date of Session'] >= start_date) & (df['Date of Session'] <= end_date)]