Update app.py

app.py

@@ -86,15 +86,15 @@ def index():
 def dashboard():
     return render_template("dashboard.html")  # Render the dashboard template
 
+# Login endpoint: Authenticate the user by email and phone number
 @app.route('/login', methods=['POST'])
 def login():
     # Get data from voice bot (name, email, phone number)
     data = request.json  # Assuming voice bot sends JSON data
-    name = data.get('name')
     email = data.get('email')
     phone_number = data.get('phone_number')
 
-    if not name or not email or not phone_number:
+    if not email or not phone_number:
         return jsonify({'error': 'Missing required fields'}), 400
 
     try:
@@ -103,8 +103,9 @@ def login():
         result = sf.query(query)
 
         if result['totalSize'] > 0:
-            # User exists and credentials match, proceed to dashboard
-            return redirect("/menu")  # Redirect to the menu page after successful login
+            # User exists and credentials match, proceed to the menu
+            user_name = result['records'][0]['Name']
+            return jsonify({'success': True, 'name': user_name, 'email': email})  # Returning name for greeting
         else:
             # User not found or credentials don't match
             return jsonify({'error': 'Invalid email or phone number'}), 401  # Unauthorized
@@ -149,6 +150,7 @@ def submit():
         return jsonify({'error': str(e)}), 500
 
 
+# Menu endpoint: Fetch and display the menu
 @app.route("/menu", methods=["GET"])
 def menu_page():
     try:
@@ -156,47 +158,61 @@ def menu_page():
         query = "SELECT Name, Price__c, Ingredients__c, Category__c FROM Menu_Item__c"
         result = sf.query(query)
         
-        # Pass the menu items to the template
+        # Render the menu page template with menu items fetched from Salesforce
         return render_template("menu_page.html", menu_items=result['records'])
     
     except Exception as e:
         return jsonify({"error": f"Failed to fetch menu items from Salesforce: {str(e)}"}), 500
 
 
+# Order endpoint: Place the order and link it to the customer
 @app.route("/order", methods=["POST"])
 def place_order():
     order_data = request.json
     item_name = order_data.get('item')
     quantity = order_data.get('quantity')
+    user_email = order_data.get('email')
 
-    if not item_name or not quantity:
-        return jsonify({"error": "Item and quantity are required."}), 400
+    if not item_name or not quantity or not user_email:
+        return jsonify({"error": "Item, quantity, and email are required."}), 400
 
     try:
-        # Assuming `sf.Customer_Login__c` is where user login details are saved
+        # Query Salesforce for the customer using their email
         customer_login = sf.query(f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{user_email}' LIMIT 1")
+        if customer_login['totalSize'] == 0:
+            return jsonify({"error": "Customer not found"}), 404
+        
         customer_id = customer_login['records'][0]['Id']
 
+        # Assuming you have a method to calculate the total price based on item and quantity
+        total_price = calculate_total_price(item_name, quantity)
+
+        # Create the order in Salesforce
         order = {
             'Order_Item__c': item_name,
             'Quantity__c': quantity,
-            'Customer__c': customer_id,  # Linking the order to the customer by email
+            'Customer__c': customer_id,  # Linking the order to the customer
             'Status__c': 'Pending',
-            'Total_Amount__c': calculate_total_price(item_name, quantity),  # Implement this function to calculate the total
+            'Total_Amount__c': total_price
         }
 
         order_response = sf.Order__c.create(order)
         return jsonify({"success": True, "order_id": order_response['id']}), 200
+
     except Exception as e:
         return jsonify({"error": f"Error placing the order: {str(e)}"}), 500
 
-
-# Helper function to calculate the total price (implement as per your pricing logic)
+# Helper function to calculate the total price (implement this function as needed)
 def calculate_total_price(item_name, quantity):
-    item = sf.query(f"SELECT Price__c FROM Menu_Item__c WHERE Name = '{item_name}' LIMIT 1")
-    price = item['records'][0]['Price__c']
-    return price * quantity
-
+    try:
+        # Fetch item price from Salesforce and calculate total price
+        item = sf.query(f"SELECT Price__c FROM Menu_Item__c WHERE Name = '{item_name}' LIMIT 1")
+        if item['totalSize'] == 0:
+            raise ValueError("Item not found")
+        price = item['records'][0]['Price__c']
+        return price * quantity
+    except Exception as e:
+        raise ValueError(f"Error calculating total price: {str(e)}")
 
 # Route for the order summary page
 @app.route("/order-summary", methods=["GET"])