Update main.py

main.py

@@ -387,26 +387,56 @@ def update_user_limit(uid):
 
 
 # User management endpoint for profile
-@app.route('/api/user/profile', methods=['GET'])
-def get_user_profile():
+@app.route('/api/user/spending-overview', methods=['GET'])
+def get_spending_overview():
     try:
         uid = verify_token(request.headers.get('Authorization', '').split(' ')[1])
-        user = auth.get_user(uid)
-        user_data = db.reference(f'users/{uid}').get()
+        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': []
+            })
+        
+        # Create dataframe
+        df = pd.DataFrame.from_dict(transactions, orient='index')
+        
+        # Check if dataframe is empty
+        if df.empty:
+            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')
+        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({
-            'uid': uid,
-            'email': user.email,
-            'daily_cash': user_data.get('daily_cash', 0),
-            'remaining_cash': user_data.get('remaining_cash', 0),
-            'last_reset': user_data.get('last_reset'),
-            'is_admin': user_data.get('is_admin', False)
+            'daily_spending': daily_spending.to_dict(orient='records'),
+            'recent_transactions': df.sort_values(by='timestamp', ascending=False)
+                .head(10)
+                .to_dict(orient='records')
         })
     except Exception as e:
-        uid = verify_token(request.headers.get('Authorization', '').split(' ')[1])
-        user = auth.get_user(uid)
-        user_data = db.reference(f'users/{uid}').get()
-        return jsonify({'error': str(e)+ f'user data: {user_data}'}), 500
+        return jsonify({'error': str(e)}), 500
+
 
 # Receipt media endpoints