Update main.py

main.py

@@ -221,21 +221,47 @@ 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])
         transactions_ref = db.reference('transactions')
         transactions = transactions_ref.order_by_child('uid').equal_to(uid).get()
+
+        if transactions is None:
+            return jsonify({
+                'daily_spending': [],
+                'recent_transactions': []
+            })
         
-        df = pd.DataFrame(transactions.values())
+        # Create dataframe
+        df = pd.DataFrame.from_dict(transactions, orient='index')
+        
+        # Check if dataframe is empty
         if df.empty:
-            return jsonify({'error': 'No transactions found'}), 404
-
+            return jsonify({
+                'daily_spending': [],
+                'recent_transactions': []
+            })
+        
+        # Check if all the data is here, otherwise we delete the data.
+        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')
-        daily_spending = df.groupby(df['date'].dt.date)['total'].sum().reset_index()
+        df.dropna(subset=['date'], inplace=True)
+        if df.empty:
+            return jsonify({
+                'daily_spending': [],
+                'recent_transactions': []
+            })
         
+        daily_spending = df.groupby(df['date'].dt.date)['total'].sum().reset_index()
+
         return jsonify({
             'daily_spending': daily_spending.to_dict(orient='records'),
             'recent_transactions': df.sort_values(by='timestamp', ascending=False)
@@ -243,10 +269,8 @@ def get_spending_overview():
                 .to_dict(orient='records')
         })
     except Exception as e:
-        uid = verify_token(request.headers.get('Authorization', '').split(' ')[1])
-        transactions_ref = db.reference('transactions')
-        transactions = transactions_ref.order_by_child('uid').equal_to(uid).get()
-        return jsonify({'error': str(e) + f'transactions:{transactions}'}), 500
+        return jsonify({'error': str(e)}), 500
+
 
 # ... (previous imports remain the same)