Update app.py

app.py

@@ -1,7 +1,6 @@
 import bcrypt
 import gradio as gr
 from simple_salesforce import Salesforce
-from datetime import datetime
 
 # Salesforce Connection
 sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
@@ -14,118 +13,41 @@ def hash_password(password):
 def verify_password(plain_password, hashed_password):
     return bcrypt.checkpw(plain_password.encode('utf-8'), hashed_password.encode('utf-8'))
 
-# Function to create Order in Salesforce
-def create_order_in_salesforce(user_email, cart_items, total_cost, special_instructions):
-    try:
-        # 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 []
-
-# Function to load add-ons data
-def load_add_ons_from_salesforce():
+# Signup function
+def signup(name, email, phone, password):
     try:
-        query = "SELECT Name, Price__c FROM Add_Ons__c"
+        email = email.strip()
+        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
         result = sf.query(query)
-        return result['records']
-    except Exception as e:
-        return []
-
-# 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
+        if len(result['records']) > 0:
+            return "Email already exists! Please use a different email."
 
-# 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
+        hashed_password = hash_password(password)
 
+        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 while fetching order details: {str(e)}"
+        return f"Error during signup: {str(e)}"
 
-# Function to handle login and authentication
+# Login function
 def login(email, password):
     try:
-        # Query the user by email
+        email = email.strip()
         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:
@@ -133,30 +55,25 @@ def login(email, password):
     except Exception as e:
         return f"Error during login: {str(e)}", None
 
-# Function to handle signup
-def signup(name, email, phone, password):
+# Function to load menu data
+def load_menu_from_salesforce():
     try:
-        # Check if the email already exists
-        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
+        query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
         result = sf.query(query)
-        
-        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."
+        return result['records']
     except Exception as e:
-        return f"Error during signup: {str(e)}"
+        return []
 
-# Function to filter menu based on preference
+# 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 []
+
+# Function to filter menu items
 def filter_menu(preference):
     menu_data = load_menu_from_salesforce()
 
@@ -195,39 +112,45 @@ def filter_menu(preference):
 
     return html_content
 
-# Modal Window to display item details and add-ons
-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
+# Function to create order in Salesforce
+def create_order_in_salesforce(cart, customer_email):
+    try:
+        # Fetch customer details
+        query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Email__c = '{customer_email}'"
+        result = sf.query(query)
+        
+        if len(result['records']) == 0:
+            return "Customer not found."
+
+        customer = result['records'][0]
+        customer_id = customer['Id']
+        customer_name = customer['Name']
+        
+        # Create Order record
+        order = sf.Order.create({
+            'AccountId': customer_id,  # Assuming Account is linked to Customer
+            'Status': 'Draft',  # You can set this as per your requirement
+            'OrderNumber': 'Auto-generated',  # Use your custom logic if required
+            'Name': f"Order for {customer_name}",
+            'Customer_Email__c': customer_email  # Assuming a custom field for email in Order
+        })
+        
+        # Create Order Items for each cart item
+        for item in cart:
+            sf.OrderItem.create({
+                'OrderId': order['id'],
+                'Product2Id': item['product_id'],  # Assuming you have Product2 Id for each item
+                'Quantity': item['quantity'],
+                'UnitPrice': item['price'],
+                'TotalPrice': item['totalCost'],
+                'Special_Instructions__c': item['instructions'],
+            })
+
+        return "Order successfully placed and saved to Salesforce."
+    
+    except Exception as e:
+        return f"Error during order creation: {str(e)}"
+
 # JavaScript for Modal and Cart
 def modal_js():
     modal_script = """
@@ -262,7 +185,7 @@ def modal_js():
             const extras = selectedAddOns.map(extra => ({
                 name: extra.value,
                 price: parseFloat(extra.getAttribute('data-price')),
-                quantity: 1  // Default quantity for add-ons is 1
+                quantity: 1
             }));
             const extrasCost = extras.reduce((total, extra) => total + (extra.price * extra.quantity), 0);
             const totalCost = (price * quantity) + extrasCost;
@@ -277,60 +200,39 @@ def modal_js():
             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 += `
-                    <div style="border: 1px solid #ddd; padding: 10px; margin-bottom: 10px; border-radius: 8px;">
-                        <h3>${item.name}</h3>
-                        <p>Quantity: <input type="number" value="${item.quantity}" min="1" style="width: 50px;" onchange="updateCartItem(${index}, 'item', this.value)" /></p>
-                        <p>Extras: ${extrasList || 'None'}</p>
-                        <p>Special Instructions: ${item.instructions || 'None'}</p>
-                        <p>Total Cost: $<span id="item-${index}-total">${item.totalCost.toFixed(2)}</span></p>
-                        <button onclick="removeFromCart(${index})" style="color: red;">Remove</button>
-                    </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);
-            updateCartButton();
-            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);
-            item.totalCost = (price * item.quantity) + extrasCost;
-            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"]');
-            checkboxes.forEach(checkbox => checkbox.checked = false);  // Uncheck all add-ons
+            // Collect cart details
+            const cartData = cart.map(item => ({
+                product_id: item.product_id,  // Assuming you have a product_id field
+                name: item.name,
+                quantity: item.quantity,
+                price: item.price,
+                totalCost: item.totalCost,
+                instructions: item.instructions
+            }));
+
+            const email = 'user@example.com';  // Replace with dynamic email
+
+            fetch('/create_order', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({
+                    cart: cartData,
+                    email: email
+                })
+            })
+            .then(response => response.json())
+            .then(data => {
+                alert(data.message);
+                if (data.success) {
+                    // Optionally redirect to a confirmation page or clear cart
+                }
+            })
+            .catch(error => {
+                alert("Error during checkout: " + error);
+            });
         }
     </script>
     """
@@ -368,19 +270,11 @@ 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()
+app.launch()