Update main.py

main.py

@@ -5,6 +5,7 @@ from visualization import Visualization  # Import the data viz class
 from ai_analysis import AIAnalysis  # Import the ai analysis class
 from sidebar import Sidebar  # Import the Sidebar class
 
+
 def main():
     # Initialize the app configuration
     app_config = AppConfig()
@@ -24,6 +25,26 @@ def main():
 
     st.title("Intervention Program Analysis")
 
+    # Date selection option
+    date_option = st.radio(
+        "Select data range:",
+        ("All Data", "Date Range")
+    )
+
+    # Initialize start and end date variables
+    start_date = None
+    end_date = None
+
+    if date_option == "Date Range":
+        # Prompt user to enter start and end dates
+        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:
+            st.error("Start date must be before end date.")
+            return
+
     # File uploader
     uploaded_file = st.file_uploader("Upload your Excel file", type=["xlsx"])
 
@@ -38,22 +59,8 @@ 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")
-            )
-
+            # Filter data if date range is selected
             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")