Update app.py

app.py

@@ -437,40 +437,51 @@ elif page == "Upload Receipt":
     
     uploaded_file = st.file_uploader("Upload a receipt image", type=["jpg", "jpeg", "png"])
     
+    # Create session state variables if they don't exist
+    if 'receipt_data' not in st.session_state:
+        st.session_state.receipt_data = None
+    if 'receipt_processed' not in st.session_state:
+        st.session_state.receipt_processed = False
+    
     if uploaded_file is not None:
         # Display the uploaded image
         image = Image.open(uploaded_file)
         st.image(image, caption="Uploaded Receipt", width=400)
         
         # Process button
-        if st.button("Extract Data from Receipt"):
-            with st.spinner("Processing receipt image..."):
-                transaction_data = extract_data_from_receipt(image)
+        if st.button("Extract Data from Receipt") or st.session_state.receipt_processed:
+            if not st.session_state.receipt_processed:
+                with st.spinner("Processing receipt image..."):
+                    transaction_data = extract_data_from_receipt(image)
+                    st.session_state.receipt_data = transaction_data
+                    st.session_state.receipt_processed = True
+            
+            if st.session_state.receipt_data:
+                # Show extracted data
+                st.success("Successfully extracted data from receipt!")
+                st.json(st.session_state.receipt_data)
                 
-                if transaction_data:
-                    # Show extracted data and confirmation
-                    st.success("Successfully extracted data from receipt!")
-                    st.json(transaction_data)
+                # Ask for confirmation
+                if st.button("Confirm and Add Transaction"):
+                    add_expense_to_df(st.session_state.receipt_data)
+                    st.success("Transaction added to your records!")
+                    # Reset the receipt processing state
+                    st.session_state.receipt_processed = False
+                    st.session_state.receipt_data = None
                     
-                    # Ask for confirmation
-                    if st.button("Confirm and Add Transaction"):
-                        st.write("Transaction data before adding:", transaction_data)
-                        add_expense_to_df(transaction_data)
-                        st.write("Transaction added!")
-                        st.success("Transaction added to your records!")
-                        
-                        # Show updated recent transactions
-                        st.subheader("Recent Transactions")
-                        display_df = st.session_state.df.copy()
-                        display_df['Credit'] = display_df['Amount'].apply(lambda x: x if x > 0 else 0)
-                        display_df['Debit'] = display_df['Amount'].apply(lambda x: abs(x) if x < 0 else 0)
-                        st.dataframe(
-                            display_df[['Date', 'Description', 'Category', 'Credit', 'Debit', 'Balance']]
-                            .sort_values(by='Date', ascending=False),
-                            use_container_width=True
-                        )
-                else:
-                    st.error("Could not extract transaction data from image. Please try a clearer image or enter the details manually.")
+                    # Show updated recent transactions
+                    st.subheader("Recent Transactions")
+                    display_df = st.session_state.df.copy()
+                    display_df['Credit'] = display_df['Amount'].apply(lambda x: x if x > 0 else 0)
+                    display_df['Debit'] = display_df['Amount'].apply(lambda x: abs(x) if x < 0 else 0)
+                    st.dataframe(
+                        display_df[['Date', 'Description', 'Category', 'Credit', 'Debit', 'Balance']]
+                        .sort_values(by='Date', ascending=False),
+                        use_container_width=True
+                    )
+            else:
+                st.error("Could not extract transaction data from image. Please try a clearer image or enter the details manually.")
+                st.session_state.receipt_processed = False
     
     # Manual entry option if image processing fails
     st.subheader("Manual Receipt Entry")