Update app.py

app.py

@@ -397,6 +397,12 @@ def main():
     # Initialize session state for last error
     if 'last_error' not in st.session_state:
         st.session_state['last_error'] = None
+    
+    # Initialize session state for transaction date range
+    if 'min_date' not in st.session_state:
+        st.session_state['min_date'] = date(2024, 1, 1)  # Default min date
+    if 'max_date' not in st.session_state:
+        st.session_state['max_date'] = date(2024, 12, 31)  # Default max date
 
     # Sidebar: Select input type: Bulk PDF or CSV Upload
     input_type = st.sidebar.radio("Select Input Type", ("Bulk Bank Statement Upload", "CSV Upload"))
@@ -475,7 +481,7 @@ def main():
             except Exception as e:
                 st.error(f"Error processing CSV file: {str(e)}")
 
-    # If transactions are loaded, show DataFrame
+    # If transactions are loaded, show DataFrame and update date ranges
     if all_transactions:
         df = pd.DataFrame(all_transactions)
         # Drop 'Unnamed:' columns from the final DataFrame
@@ -484,9 +490,26 @@ def main():
             try:
                 df['Amount'] = pd.to_numeric(df['Amount'], errors='coerce')
                 df['Amount'] = df['Amount'].apply(lambda x: f"R {x:,.2f}" if x >= 0 else f"R ({abs(x):,.2f})")
-                df['Date'] = pd.to_datetime(df['Date'], format='%d/%m/%Y', errors='coerce').dt.strftime('%d/%m/%Y')
+                
+                # Process dates and extract min/max dates for date range inputs
+                df['Date'] = pd.to_datetime(df['Date'], format='%d/%m/%Y', errors='coerce')
+                
+                # Get min and max dates from transactions
+                if not df['Date'].isna().all():
+                    min_date = df['Date'].min().date()
+                    max_date = df['Date'].max().date()
+                    
+                    # Update session state with actual transaction date range
+                    st.session_state['min_date'] = min_date
+                    st.session_state['max_date'] = max_date
+                
+                # Format dates for display
+                df['Date'] = df['Date'].dt.strftime('%d/%m/%Y')
+                
             except Exception as e:
                 st.warning("Some data could not be formatted correctly.")
+                st.exception(e)
+            
             st.success("Transactions loaded successfully!")
             st.write("### Extracted Transactions")
             st.dataframe(df)
@@ -499,9 +522,11 @@ def main():
     st.write("### Generate Financial Report")
     col1, col2 = st.columns(2)
     with col1:
-        start_date = st.date_input("Start Date", date(2024, 1, 1))
+        # Use the min_date from transactions if available, otherwise use default
+        start_date = st.date_input("Start Date", st.session_state['min_date'])
     with col2:
-        end_date = st.date_input("End Date", date(2024, 12, 31))
+        # Use the max_date from transactions if available, otherwise use default
+        end_date = st.date_input("End Date", st.session_state['max_date'])
 
     statement_type = st.selectbox("Select Financial Statement", ["Income Statement", "Cashflow Statement", "Balance Sheet"])