Update app.py

app.py

@@ -1,6 +1,7 @@
 import bcrypt
 import gradio as gr
 from simple_salesforce import Salesforce
+from datetime import datetime
 
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
@@ -13,41 +14,118 @@ def hash_password(password):
 def verify_password(plain_password, hashed_password):
     return bcrypt.checkpw(plain_password.encode('utf-8'), hashed_password.encode('utf-8'))
 
-# Signup function
-def signup(name, email, phone, password):
+# Function to create Order in Salesforce
+def create_order_in_salesforce(user_email, cart_items, total_cost, special_instructions):
     try:
-        email = email.strip()
-        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
+        # Fetch User details from Customer_Login__c object
+        user = sf.Customer_Login__c.get_by_custom_id('Email__c', user_email)
+        user_id = user['Id']
+        
+        # Create the order record in Order__c
+        order_data = {
+            'Email__c': user_id,  # Relating the order to the logged-in user
+            'Order_Date__c': datetime.now().strftime('%Y-%m-%d'),
+            'Total_Amount__c': total_cost,
+            'Order_Status__c': 'Pending',  # Initial status
+            'Special_Instructions__c': special_instructions
+        }
+        order_response = sf.Order__c.create(order_data)
+        
+        # Create order items (order details) for each item in the cart
+        for item in cart_items:
+            item_data = {
+                'Order__c': order_response['id'],  # Linking this item to the created order
+                'Name': item['name'],
+                'Quantity__c': item['quantity'],
+                'Price__c': item['price'],
+                'Add_Ons__c': str(item['add_ons']),
+                'Special_Instructions__c': item['special_instructions']
+            }
+            sf.Order_Item__c.create(item_data)
+
+        return order_response['id']  # Returning order ID for further use
+    except Exception as e:
+        return f"Error during order creation: {str(e)}"
+
+# Function to load menu data
+def load_menu_from_salesforce():
+    try:
+        query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
         result = sf.query(query)
+        return result['records']
+    except Exception as e:
+        return []
 
-        if len(result['records']) > 0:
-            return "Email already exists! Please use a different email."
+# Function to load add-ons data
+def load_add_ons_from_salesforce():
+    try:
+        query = "SELECT Name, Price__c FROM Add_Ons__c"
+        result = sf.query(query)
+        return result['records']
+    except Exception as e:
+        return []
 
-        hashed_password = hash_password(password)
+# Function to handle cart and checkout
+def proceed_to_checkout(user_email, cart_items):
+    # Calculate total cost
+    total_cost = sum(item['price'] * item['quantity'] for item in cart_items)
+    special_instructions = ', '.join([item['special_instructions'] for item in cart_items])
+
+    # Create order in Salesforce and get order ID
+    order_id = create_order_in_salesforce(user_email, cart_items, total_cost, special_instructions)
+
+    # Return order ID to display the details page
+    return order_id
+
+# Function to display the order details after checkout
+def show_order_details(order_id):
+    try:
+        # Query the order record
+        order_query = f"SELECT Name, Order_Date__c, Total_Amount__c, Order_Status__c, Special_Instructions__c FROM Order__c WHERE Id = '{order_id}'"
+        order = sf.query(order_query)
+        order_details = order['records'][0]
+
+        # Query the order items associated with the order
+        items_query = f"SELECT Name, Quantity__c, Price__c, Add_Ons__c, Special_Instructions__c FROM Order_Item__c WHERE Order__c = '{order_id}'"
+        items = sf.query(items_query)
+
+        # Prepare the message to display the order details
+        order_message = f"Thank you for your order! Here's the summary:\n"
+        order_message += f"Order ID: {order_details['Name']}\n"
+        order_message += f"Order Date: {order_details['Order_Date__c']}\n"
+        order_message += f"Total Amount: ${order_details['Total_Amount__c']}\n"
+        order_message += f"Order Status: {order_details['Order_Status__c']}\n"
+        order_message += f"Special Instructions: {order_details['Special_Instructions__c']}\n\n"
+
+        order_message += "Items ordered:\n"
+        for item in items['records']:
+            order_message += f"\nItem Name: {item['Name']}\n"
+            order_message += f"Quantity: {item['Quantity__c']}\n"
+            order_message += f"Price: ${item['Price__c']}\n"
+            order_message += f"Add-ons: {item['Add_Ons__c']}\n"
+            order_message += f"Special Instructions: {item['Special_Instructions__c']}\n"
+
+        order_message += "\nThank you for shopping with us! Visit again soon."
+
+        return order_message
 
-        sf.Customer_Login__c.create({
-            'Name': name.strip(),
-            'Email__c': email,
-            'Phone_Number__c': phone.strip(),
-            'Password__c': hashed_password
-        })
-        return "Signup successful! You can now login."
     except Exception as e:
-        return f"Error during signup: {str(e)}"
+        return f"Error while fetching order details: {str(e)}"
 
-# Login function
+# Function to handle login and authentication
 def login(email, password):
     try:
-        email = email.strip()
+        # Query the user by email
         query = f"SELECT Name, Password__c FROM Customer_Login__c WHERE Email__c = '{email}'"
         result = sf.query(query)
-
+        
         if len(result['records']) == 0:
             return "Invalid email or password.", None
-
+        
         user = result['records'][0]
         stored_password = user['Password__c']
-
+        
+        # Verify password
         if verify_password(password.strip(), stored_password):
             return "Login successful!", user['Name']
         else:
@@ -55,25 +133,30 @@ def login(email, password):
     except Exception as e:
         return f"Error during login: {str(e)}", None
 
-# Function to load menu data
-def load_menu_from_salesforce():
-    try:
-        query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
-        result = sf.query(query)
-        return result['records']
-    except Exception as e:
-        return []
-
-# Function to load add-ons data
-def load_add_ons_from_salesforce():
+# Function to handle signup
+def signup(name, email, phone, password):
     try:
-        query = "SELECT Name, Price__c FROM Add_Ons__c"
+        # Check if the email already exists
+        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
         result = sf.query(query)
-        return result['records']
+        
+        if len(result['records']) > 0:
+            return "Email already exists! Please use a different email."
+        
+        # Hash password and create user
+        hashed_password = hash_password(password)
+        
+        sf.Customer_Login__c.create({
+            'Name': name.strip(),
+            'Email__c': email.strip(),
+            'Phone_Number__c': phone.strip(),
+            'Password__c': hashed_password
+        })
+        return "Signup successful! You can now login."
     except Exception as e:
-        return []
+        return f"Error during signup: {str(e)}"
 
-# Function to filter menu items
+# Function to filter menu based on preference
 def filter_menu(preference):
     menu_data = load_menu_from_salesforce()
 
@@ -112,47 +195,12 @@ def filter_menu(preference):
 
     return html_content
 
-# Create Modal Window HTML
-def create_modal_window():
-    add_ons = load_add_ons_from_salesforce()
-    add_ons_html = ""
-    for add_on in add_ons:
-        add_ons_html += f"""
-        <label>
-            <input type="checkbox" name="biryani-extra" value="{add_on['Name']}" data-price="{add_on['Price__c']}" />
-            {add_on['Name']} + ${add_on['Price__c']}
-        </label>
-        <br>
-        """
-
-    modal_html = f"""
-    <div id="modal" style="display: none; position: fixed; background: white; border-radius: 8px; box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2); padding: 20px; z-index: 1000;">
-        <div style="text-align: right;">
-            <button onclick="closeModal()" style="background: none; border: none; font-size: 18px; cursor: pointer;">&times;</button>
-        </div>
-        <img id="modal-image" style="width: 100%; height: 300px; border-radius: 8px; margin-bottom: 20px;" />
-        <h2 id="modal-name"></h2>
-        <p id="modal-description"></p>
-        <p id="modal-price"></p>
-        <label for="biryani-extras"><strong>Add-ons :</strong></label>
-        <div id="biryani-extras-options" style="display: flex; flex-wrap: wrap; gap: 10px; margin: 10px 0;">
-            {add_ons_html}
-        </div>
-        <label for="quantity">Quantity:</label>
-        <input type="number" id="quantity" value="1" min="1" style="width: 50px;" />
-        <textarea id="special-instructions" placeholder="Add your special instructions here..." style="width: 100%; height: 60px;"></textarea>
-        <button style="background-color: #28a745; color: white; border: none; padding: 10px 20px; font-size: 14px; border-radius: 5px; cursor: pointer;" onclick="addToCart()">Add to Cart</button>
-    </div>
-    """
-    return modal_html
-
 # JavaScript for Modal and Cart
 def modal_js():
     modal_script = """
     <script>
         let cart = [];
         let totalCartCost = 0;
-
         function openModal(name, image2, description, price) {
             const modal = document.getElementById('modal');
             modal.style.display = 'block';
@@ -169,11 +217,9 @@ def modal_js():
             document.getElementById('special-instructions').value = '';
             resetAddOns();  // Reset add-ons when opening the modal
         }
-
         function closeModal() {
             document.getElementById('modal').style.display = 'none';
         }
-
         function addToCart() {
             const name = document.getElementById('modal-name').innerText;
             const price = parseFloat(document.getElementById('modal-price').innerText.replace('$', ''));
@@ -187,7 +233,6 @@ def modal_js():
             }));
             const extrasCost = extras.reduce((total, extra) => total + (extra.price * extra.quantity), 0);
             const totalCost = (price * quantity) + extrasCost;
-
             // Add the item to the cart with its specific add-ons
             cart.push({ name, price, quantity, extras, instructions, totalCost });
             totalCartCost += totalCost;  // Update the total cost of the cart
@@ -195,17 +240,14 @@ def modal_js():
             updateCartTotalCost();  // Update total cost displayed
             closeModal();
         }
-
         function updateCartButton() {
             const cartButton = document.getElementById('cart-button');
             cartButton.innerText = `View Cart (${cart.length} items)`;
         }
-
         function openCartModal() {
             const cartModal = document.getElementById('cart-modal');
             const cartItemsContainer = document.getElementById('cart-items');
             cartItemsContainer.innerHTML = "";
-
             cart.forEach((item, index) => {
                 const extrasList = item.extras.map(extra => `${extra.name} x<input type="number" value="${extra.quantity}" min="1" style="width: 50px;" onchange="updateCartItem(${index}, 'extra', this.value)" /> (+$${(extra.price * extra.quantity).toFixed(2)})`).join(', ');
                 cartItemsContainer.innerHTML += `
@@ -219,14 +261,11 @@ def modal_js():
                     </div>
                 `;
             });
-
             cartModal.style.display = 'block';
         }
-
         function closeCartModal() {
             document.getElementById('cart-modal').style.display = 'none';
         }
-
         function removeFromCart(index) {
             totalCartCost -= cart[index].totalCost;  // Deduct the cost of the removed item from total cost
             cart.splice(index, 1);
@@ -234,14 +273,12 @@ def modal_js():
             updateCartTotalCost();  // Update total cost displayed
             openCartModal();
         }
-
         function updateCartItem(index, type, value) {
             if (type === 'item') {
                 cart[index].quantity = parseInt(value);
             } else if (type === 'extra') {
                 cart[index].extras[0].quantity = parseInt(value); // Assuming one add-on for simplicity
             }
-
             const item = cart[index];
             const price = item.price;
             const extrasCost = item.extras.reduce((total, extra) => total + (extra.price * extra.quantity), 0);
@@ -249,17 +286,14 @@ def modal_js():
             document.getElementById(`item-${index}-total`).innerText = item.totalCost.toFixed(2);
             updateCartTotalCost();  // Update total cost displayed
         }
-
         function updateCartTotalCost() {
             const totalCostElement = document.getElementById('cart-total-cost');
             totalCartCost = cart.reduce((total, item) => total + item.totalCost, 0);
             totalCostElement.innerText = `Total Cart Cost: $${totalCartCost.toFixed(2)}`;
         }
-
         function proceedToCheckout() {
             alert("Proceeding to checkout...");
         }
-
         // Reset all selected add-ons when opening a new item modal
         function resetAddOns() {
             const checkboxes = document.querySelectorAll('input[name="biryani-extra"]');
@@ -301,11 +335,19 @@ with gr.Blocks() as app:
             gr.HTML(create_modal_window())
             gr.HTML(modal_js())
 
+    with gr.Row(visible=False) as order_page:
+        with gr.Column():
+            order_id_input = gr.Textbox(label="Order ID")
+            show_order_button = gr.Button("Show Order Details")
+            order_details_output = gr.Textbox(label="Order Details")
+            show_order_button.click(show_order_details, [order_id_input], order_details_output)
+
     login_button.click(
         lambda email, password: (gr.update(visible=False), gr.update(visible=True), gr.update(value=filter_menu("All")), "Login successful!")
         if login(email, password)[0] == "Login successful!" else (gr.update(), gr.update(), gr.update(), "Invalid email or password."),
         [login_email, login_password], [login_page, menu_page, menu_output, login_output]
     )
+
     preference.change(lambda pref: filter_menu(pref), [preference], menu_output)
 
 app.launch()