Update app.py

app.py

@@ -32,8 +32,9 @@ if uploaded_file:
     st.write("Here are the first few entries in your file for format verification:")
     st.write(df.head())
 
-    # Ensure 'Amount' is numeric
+    # Ensure 'Amount' is numeric and 'Date' is in datetime format
     df['Amount'] = pd.to_numeric(df['Amount'], errors='coerce')
+    df['Date'] = pd.to_datetime(df['Date'])
 
     # Initialize Hugging Face model for zero-shot classification
     classifier = pipeline('zero-shot-classification', model='roberta-large-mnli')
@@ -66,14 +67,15 @@ if uploaded_file:
         "Transportation": transportation_budget
     }
 
-    # Add a date slider for start and end date (default is the last month)
-    today = datetime.date.today()
-    last_month = today - pd.DateOffset(months=1)
-    start_date = st.sidebar.date_input("Start Date", last_month)
-    end_date = st.sidebar.date_input("End Date", today)
+    # Get the minimum and maximum dates from the uploaded file
+    min_date = df['Date'].min()
+    max_date = df['Date'].max()
+
+    # Add a date slider for start and end date based on data from the uploaded file
+    start_date = st.sidebar.date_input("Start Date", min_date)
+    end_date = st.sidebar.date_input("End Date", max_date)
 
     # Filter data by date range
-    df['Date'] = pd.to_datetime(df['Date'])
     df_filtered = df[(df['Date'] >= pd.to_datetime(start_date)) & (df['Date'] <= pd.to_datetime(end_date))]
 
     # Track if any category exceeds its budget