Update combined_summary.py

combined_summary.py

@@ -1,10 +1,5 @@
-from flask import Blueprint, render_template, session, redirect, url_for
+from flask import Blueprint, render_template, session, redirect, url_for, request, jsonify
 from salesforce import get_salesforce_connection
-from flask import Blueprint, render_template, request, session, jsonify, redirect, url_for
-import os
-import re
-from salesforce import get_salesforce_connection
-
 
 combined_summary_blueprint = Blueprint('combined_summary', __name__)
 
@@ -76,7 +71,7 @@ def combined_summary():
         order = order_result["records"][0]
         order_details = order.get("Order_Details__c", "")
         order_items = []
-        sector_names = set()  # Use a set to ensure sector names are unique
+        sector_names = set()
 
         for line in order_details.split('\n'):
             item_parts = line.split('|')
@@ -86,7 +81,7 @@ def combined_summary():
                 safe_item_name = escape_soql(item_name)
 
                 menu_query = f"""
-                    SELECT Name, Price__c, Image1__c,
+                    SELECT Id, Name, Price__c, Image1__c,
                         Ingredient_1__r.Ingredient_Name__c, Ingredient_1__r.Ingredient_Image__c, 
                         Ingredient_1__r.Health_Benefits__c, Ingredient_1__r.Fun_Facts__c,
                         Ingredient_2__r.Ingredient_Name__c, Ingredient_2__r.Ingredient_Image__c, 
@@ -121,10 +116,11 @@ def combined_summary():
 
                     # Process the Sector__c field from Menu_Item__c
                     if menu_item.get('Sector__c'):
-                        sector_names.update(menu_item['Sector__c'].split(','))  # Add sectors to the set
+                        sector_names.update(menu_item['Sector__c'].split(','))
 
-                    # Only add the item if ingredients are present
+                    # Add item with Salesforce Id
                     order_items.append({
+                        "id": menu_item.get("Id", ''),
                         "name": item_name,
                         "price": menu_item.get("Price__c", 0),
                         "image_url": menu_item.get("Image1__c", ''),
@@ -158,9 +154,69 @@ def combined_summary():
             progress_percentage=round(progress_percentage),
             points_needed_for_next_tier=round(points_needed_for_next_tier),
             order_items=order_items,
-            sector_details=sector_details
+            sector_details=sector_details,
+            validity_year=2025  # Add validity_year for template
         )
 
     except Exception as e:
         print(f"Error in combined_summary: {str(e)}")
-        return f"Error: {str(e)}", 500
+        return f"Error: {str(e)}", 500
+
+@combined_summary_blueprint.route('/reorder', methods=['POST'])
+def reorder():
+    email = session.get('user_email')
+    if not email:
+        return jsonify({'success': False, 'message': 'Please log in to reorder.'}), 401
+
+    try:
+        data = request.get_json()
+        item_id = data.get('item_id')
+        if not item_id:
+            return jsonify({'success': False, 'message': 'Item ID is required.'}), 400
+
+        # Sanitize item_id for SOQL
+        safe_item_id = escape_soql(item_id)
+
+        # Verify item exists in Menu_Item__c
+        item_query = f"""
+            SELECT Id, Name, Price__c
+            FROM Menu_Item__c
+            WHERE Id = '{safe_item_id}'
+        """
+        item_result = sf.query_all(item_query)
+        if not item_result.get("records"):
+            return jsonify({'success': False, 'message': 'Item not found.'}), 404
+
+        item = item_result["records"][0]
+        item_name = item.get("Name", "Item")
+
+        # Check if item is already in cart
+        safe_email = escape_soql(email)
+        cart_query = f"""
+            SELECT Id, Quantity__c
+            FROM Cart__c
+            WHERE Customer_Email__c = '{safe_email}' AND Menu_Item_Id__c = '{safe_item_id}'
+        """
+        cart_result = sf.query_all(cart_query)
+
+        if cart_result.get("records"):
+            # Update existing cart item
+            cart_id = cart_result["records"][0]["Id"]
+            current_quantity = cart_result["records"][0].get("Quantity__c", 1)
+            sf.Cart__c.update(cart_id, {'Quantity__c': current_quantity + 1})
+        else:
+            # Create new cart item
+            sf.Cart__c.create({
+                'Customer_Email__c': email,
+                'Menu_Item_Id__c': item_id,
+                'Quantity__c': 1
+            })
+
+        # Set session variable for menu page popup
+        session['reorder_success'] = f"{item_name} added to cart!"
+
+        return jsonify({'success': True, 'message': f'{item_name} added to cart!'}), 200
+
+    except Exception as e:
+        print(f"Error in reorder: {str(e)}")
+        return jsonify({'success': False, 'message': 'An error occurred while reordering.'}), 500