Update main.py

main.py

@@ -288,7 +288,12 @@ def validate_and_save_transaction(uid, user_data, data, file_hash, image_bytes,
 @app.route('/api/user/spending-overview', methods=['GET'])
 def get_spending_overview():
     try:
-        uid = verify_token(request.headers.get('Authorization', '').split(' ')[1])
+        # Get UID from auth token.
+        auth_header = request.headers.get('Authorization', '')
+        token = auth_header.split(' ')[1] if len(auth_header.split(' ')) > 1 else ''
+        uid = verify_token(token)
+        
+        # Get transactions for this user.
         transactions_ref = db.reference('transactions')
         transactions = transactions_ref.order_by_child('uid').equal_to(uid).get()
 
@@ -298,6 +303,7 @@ def get_spending_overview():
                 'recent_transactions': []
             })
         
+        # Create DataFrame from the transactions.
         df = pd.DataFrame.from_dict(transactions, orient='index')
         if df.empty:
             return jsonify({
@@ -305,14 +311,16 @@ def get_spending_overview():
                 'recent_transactions': []
             })
         
-        df.dropna(subset=['uid','total','items','date', 'receipt_number', 'timestamp','hash', 'manual_entry'], inplace=True)
+        # Drop rows missing essential fields (image_url is optional).
+        df.dropna(subset=['uid', 'total', 'items', 'date', 'receipt_number', 'timestamp', 'hash', 'manual_entry'], inplace=True)
         if df.empty:
             return jsonify({
                 'daily_spending': [],
                 'recent_transactions': []
             })
         
-        df['date'] = pd.to_datetime(df['date'], format='%d/%m/%Y', errors='coerce')
+        # Parse the date column without a fixed format (allows "Fri, 12 Jun 2020 00:00:00 GMT").
+        df['date'] = pd.to_datetime(df['date'], errors='coerce')
         df.dropna(subset=['date'], inplace=True)
         if df.empty:
             return jsonify({
@@ -320,17 +328,24 @@ def get_spending_overview():
                 'recent_transactions': []
             })
         
+        # Calculate daily spending.
         daily_spending = df.groupby(df['date'].dt.date)['total'].sum().reset_index()
-
+        # Replace any NaN with None.
+        daily_spending = daily_spending.replace({np.nan: None})
+        
+        # Get the 10 most recent transactions.
+        recent_transactions = df.sort_values(by='timestamp', ascending=False).head(10)
+        # Replace any NaN values (e.g. for image_url) with None.
+        recent_transactions = recent_transactions.replace({np.nan: None})
+        
         return jsonify({
             'daily_spending': daily_spending.to_dict(orient='records'),
-            'recent_transactions': df.sort_values(by='timestamp', ascending=False)
-                .head(10)
-                .to_dict(orient='records')
+            'recent_transactions': recent_transactions.to_dict(orient='records')
         })
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+
 # ========================================
 # Modified verify_admin function (now checks database is_admin flag)
 # ========================================