Update main.py

main.py

@@ -225,12 +225,13 @@ def handle_manual_entry(uid, user_ref, user_data):
     except Exception as e:
         return jsonify({'error': str(e)}), 400
 
+
 def validate_and_save_transaction(uid, user_data, data, file_hash, image_bytes, manual=False):
     transactions_ref = db.reference('transactions')
     receipt_number = data.get('receipt_number')
     items = data.get('items', [])
 
-    # Check for duplicate transactions based on receipt number and items.
+    # Check for duplicate transactions based on receipt number and items
     existing_transactions_with_receipt = transactions_ref.order_by_child('receipt_number').equal_to(receipt_number).get()
 
     if existing_transactions_with_receipt:
@@ -242,20 +243,21 @@ def validate_and_save_transaction(uid, user_data, data, file_hash, image_bytes,
 
     total = float(data.get('total', 0))
     
-    # Removed the check for insufficient funds to allow negative balances
+    # Initialize image_url from data (could be None)
+    image_url = data.get('image_url')
     
-    # Upload image if available
-    image_url = None
+    # Upload image if image_bytes are provided and not manual
     if image_bytes and not manual:
         timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
         blob = bucket.blob(f'receipts/{uid}/{timestamp}_{file_hash}.jpg')
         blob.upload_from_string(image_bytes, content_type='image/jpeg')
         image_url = blob.public_url
 
-    # Update user cash - now allowing negative values
+    # Update user cash - allowing negative values
     new_remaining = user_data['remaining_cash'] - total
     db.reference(f'users/{uid}').update({'remaining_cash': new_remaining})
 
+    # Build transaction data without image_url initially
     transaction_data = {
         'uid': uid,
         'total': total,
@@ -264,10 +266,14 @@ def validate_and_save_transaction(uid, user_data, data, file_hash, image_bytes,
         'receipt_number': receipt_number,
         'timestamp': datetime.now(pytz.UTC).isoformat(),
         'hash': file_hash,
-        'image_url': image_url,
         'manual_entry': manual
     }
 
+    # Only add image_url if it is not None
+    if image_url is not None:
+        transaction_data['image_url'] = image_url
+
+    # Save the transaction
     new_transaction_ref = transactions_ref.push(transaction_data)
     return jsonify({
         'success': True,