Update menu.py

menu.py

@@ -52,6 +52,7 @@ def menu():
         user_name = session.get('user_name')
 
     first_letter = user_name[0].upper() if user_name else "A"
+    user_image = session.get('user_image')  # Add avatar image from session
 
     # Fetch user referral and reward points
     user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
@@ -150,9 +151,35 @@ def menu():
         reward_points=reward_points,
         user_name=user_name,
         first_letter=first_letter,
-        cart_item_count=cart_item_count
+        cart_item_count=cart_item_count,
+        user_image=user_image  # Pass user_image to template
     )
 
+@menu_blueprint.route('/upload_avatar', methods=['POST'])
+def upload_avatar():
+    data = request.get_json()
+    if not data or 'image' not in data:
+        return jsonify({'success': False, 'error': 'No image data provided'})
+    
+    image_data = data['image']
+    # Validate base64 image
+    if not image_data.startswith('data:image/'):
+        return jsonify({'success': False, 'error': 'Invalid image format'})
+    
+    # Limit size to ~1MB (base64 encoded size)
+    if len(image_data) > 1_400_000:
+        return jsonify({'success': False, 'error': 'Image too large (max 1MB)'})
+    
+    session['user_image'] = image_data
+    return jsonify({'success': True, 'image': image_data})
+
+@menu_blueprint.route('/delete_avatar', methods=['POST'])
+def delete_avatar():
+    if 'user_image' in session:
+        session.pop('user_image', None)
+        return jsonify({'success': True})
+    return jsonify({'success': False, 'error': 'No image to delete'})
+
 @menu_blueprint.route('/api/addons', methods=['GET'])
 def get_addons():
     item_name = request.args.get('item_name')