Update app.py

app.py

@@ -73,7 +73,78 @@ def load_add_ons_from_salesforce():
     except Exception as e:
         return []
 
-# Function to filter menu items
+# Cart Management Functions
+cart = []
+total_cart_cost = 0
+cart_id = None  # Store the Cart__c record ID
+
+# Function to add item to the cart and Salesforce
+def add_to_cart(item_name, item_price, quantity, special_instructions, selected_addons):
+    global total_cart_cost, cart_id
+    total_cost = item_price * quantity
+    addons_cost = sum([addon['Price__c'] for addon in selected_addons])
+
+    # If this is the first item, create a new Cart__c record in Salesforce
+    if not cart_id:
+        cart_record = sf.Cart__c.create({
+            'Total_Cost__c': total_cost + addons_cost
+        })
+        cart_id = cart_record['Id']
+    else:
+        # Update the total cost of the cart
+        sf.Cart__c.update(cart_id, {
+            'Total_Cost__c': total_cart_cost + total_cost + addons_cost
+        })
+
+    # Create the Cart_Item__c record and link it to Cart__c
+    cart_item = sf.Cart_Item__c.create({
+        'Name': item_name,
+        'Quantity__c': quantity,
+        'Price__c': item_price,
+        'Special_Instructions__c': special_instructions,
+        'Cart__c': cart_id,  # Link this item to the Cart__c
+        'Add_ons__c': [addon['Id'] for addon in selected_addons]  # Link selected add-ons
+    })
+    
+    # Add the item to the cart list (for local tracking)
+    cart.append({
+        'name': item_name,
+        'price': item_price,
+        'quantity': quantity,
+        'instructions': special_instructions,
+        'total_cost': total_cost + addons_cost,
+        'addons': selected_addons
+    })
+    
+    total_cart_cost += total_cost + addons_cost
+    return cart, total_cart_cost
+
+# Function to remove item from the cart
+def remove_from_cart(index):
+    global total_cart_cost
+    total_cart_cost -= cart[index]['total_cost']
+    cart.pop(index)
+    
+    # Update the Cart__c record in Salesforce to reflect the new total cost
+    sf.Cart__c.update(cart_id, {
+        'Total_Cost__c': total_cart_cost
+    })
+    
+    return cart, total_cart_cost
+
+# Checkout Summary
+def proceed_to_checkout():
+    summary = ""
+    for item in cart:
+        summary += f"{item['name']} (x{item['quantity']}) - ${item['total_cost']:.2f}\n"
+        if item['instructions']:
+            summary += f"Instructions: {item['instructions']}\n"
+        if item['addons']:
+            summary += f"Add-ons: {', '.join([addon['Name'] for addon in item['addons']])}\n"
+    summary += f"Total Bill: ${total_cart_cost:.2f}"
+    return summary
+
+# Function to filter menu items based on user preference
 def filter_menu(preference):
     menu_data = load_menu_from_salesforce()
 
@@ -96,7 +167,7 @@ def filter_menu(preference):
             html_content += f"""
             <div style="border: 1px solid #ddd; border-radius: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1); overflow: hidden; height: 350px;">
                 <img src="{item.get('Image1__c', '')}" style="width: 100%; height: 200px; object-fit: cover;" 
-                onclick="openModal('{item['Name']}', '{item.get('Image2__c', '')}', '{item['Description__c']}', '{item['Price__c']}')">
+                onclick="openModal('{item['Name']}', '{item.get('Image2__c', '')}', '{item['Description__c']}', '${item['Price__c']}')">
                 <div style="padding: 10px;">
                     <h3 style='font-size: 1.2em; text-align: center;'>{item['Name']}</h3>
                     <p style='font-size: 1.1em; color: green; text-align: center;'>${item['Price__c']}</p>
@@ -111,19 +182,7 @@ def filter_menu(preference):
         return "<p>No items match your filter.</p>"
     return html_content
 
-# Function to add item to cart and Salesforce
-def add_to_cart(item_name, item_price, quantity, special_instructions, selected_addons):
-    cart_item = sf.Cart_Item__c.create({
-        'Name': item_name,
-        'Quantity__c': quantity,
-        'Price__c': item_price,
-        'Special_Instructions__c': special_instructions,
-        'Add_ons__c': [addon['Id'] for addon in selected_addons]
-    })
-    # Additional code for cart management
-    return "Item added to cart!"
-
-# Function to handle modal popup HTML and content
+# Modal Simulation Function (handled in Python)
 def open_modal(item_name, item_image, item_description, item_price):
     add_ons = load_add_ons_from_salesforce()
     add_ons_html = ""
@@ -152,15 +211,64 @@ def open_modal(item_name, item_image, item_description, item_price):
     """
     return gr.HTML(modal_content)
 
+
 # Gradio app definition
 with gr.Blocks() as app:
     with gr.Row():
         gr.HTML("<h1 style='text-align: center;'>Welcome to Biryani Hub</h1>")
 
-    with gr.Row(visible=True) as menu_page:
+    with gr.Row(visible=True) as login_page:
+        with gr.Column():
+            login_email = gr.Textbox(label="Email")
+            login_password = gr.Textbox(label="Password", type="password")
+            login_button = gr.Button("Login")
+            signup_button = gr.Button("Go to Signup")
+            login_output = gr.Textbox(label="Status")
+
+    with gr.Row(visible=False) as signup_page:
+        with gr.Column():
+            signup_name = gr.Textbox(label="Name")
+            signup_email = gr.Textbox(label="Email")
+            signup_phone = gr.Textbox(label="Phone")
+            signup_password = gr.Textbox(label="Password", type="password")
+            submit_signup = gr.Button("Signup")
+            login_redirect = gr.Button("Go to Login")
+            signup_output = gr.Textbox(label="Status")
+
+    with gr.Row(visible=False) as menu_page:
         with gr.Column():
             preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
-            menu_output = gr.HTML()  # Dynamically updated based on the filter
-            preference.change(lambda pref: filter_menu(pref), [preference], menu_output)
+            menu_output = gr.HTML()
+            gr.HTML("<div id='cart-button' style='position: fixed; top: 20px; right: 20px; background: #28a745; color: white; padding: 10px 20px; border-radius: 30px; cursor: pointer;'>View Cart</div>")
+            cart_summary = gr.Textbox(label="Cart Summary", interactive=False)
+            checkout_button = gr.Button("Proceed to Checkout")
+
+            checkout_button.click(proceed_to_checkout, [], cart_summary)
+
+    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]
+    )
+
+    submit_signup.click(
+        lambda name, email, phone, password: signup(name, email, phone, password),
+        inputs=[signup_name, signup_email, signup_phone, signup_password],
+        outputs=signup_output
+    )
+
+    signup_button.click(
+        lambda: (gr.update(visible=False), gr.update(visible=True)),
+        inputs=[],
+        outputs=[login_page, signup_page]
+    )
+
+    login_redirect.click(
+        lambda: (gr.update(visible=True), gr.update(visible=False)),
+        inputs=[],
+        outputs=[login_page, signup_page]
+    )
+
+    preference.change(lambda pref: filter_menu(pref), [preference], menu_output)
 
-    app.launch()
+app.launch()