Update app.py

app.py

@@ -4,84 +4,74 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-# Sample menu items from Salesforce
+# Fetch menu items from Salesforce
 def fetch_menu_items():
-    try:
-        query = "SELECT Id, Name, Price__c, Image1__c, Description__c, Section__c FROM Menu_Item__c"
-        menu_items = sf.query(query)['records']
-        return menu_items
-    except Exception as e:
-        print(f"Error fetching menu items: {e}")
-        return []
+    query = "SELECT Id, Name, Price__c, Description__c, Image1__c FROM Menu_Item__c"
+    records = sf.query(query)["records"]
+    return records
 
-# Cart management
-cart = {}
+# Fetch cart items (local data structure for demonstration)
+cart_items = []
 
-def update_cart(item_id, name, price, quantity):
-    if item_id in cart:
-        cart[item_id]['quantity'] += quantity
-        if cart[item_id]['quantity'] <= 0:
-            del cart[item_id]
-    else:
-        if quantity > 0:
-            cart[item_id] = {'name': name, 'price': price, 'quantity': quantity}
-    return cart_summary()
+# Add item to cart
+def add_to_cart(item_id, name, price):
+    for item in cart_items:
+        if item["id"] == item_id:
+            item["quantity"] += 1
+            return f"{name} added to the cart!", cart_items
+    cart_items.append({"id": item_id, "name": name, "price": price, "quantity": 1})
+    return f"{name} added to the cart!", cart_items
 
-def cart_summary():
-    total_items = sum(item['quantity'] for item in cart.values())
-    total_cost = sum(item['quantity'] * item['price'] for item in cart.values())
-    return f"{total_items} items | Total: ₹{total_cost:.2f}"
+# Display cart items
+def view_cart():
+    cart_html = ""
+    total_items = 0
+    total_price = 0
+    for item in cart_items:
+        cart_html += f"<p>{item['name']} - ₹{item['price']} x {item['quantity']}</p>"
+        total_items += item["quantity"]
+        total_price += item["price"] * item["quantity"]
+    cart_html += f"<h3>Total: ₹{total_price} ({total_items} items)</h3>"
+    return cart_html
 
-# Cart Page
-def cart_page():
-    items_html = "".join(
-        f"""
-        <div>
-            <h4>{item['name']} (₹{item['price']})</h4>
-            <p>Quantity: {item['quantity']}</p>
-        </div>
-        """
-        for item in cart.values()
-    )
-    total_cost = sum(item['quantity'] * item['price'] for item in cart.values())
-    return f"""
-    <h3>Your Cart</h3>
-    <div>{items_html}</div>
-    <h4>Total Cost: ₹{total_cost:.2f}</h4>
-    <button onclick="alert('Checkout functionality to be implemented')">Proceed to Checkout</button>
-    """
-
-# Generate HTML for Menu Items
-def generate_menu_html(menu_items):
-    html = ""
+# Generate menu page HTML
+def generate_menu_page(menu_items):
+    menu_html = ""
     for item in menu_items:
-        html += f"""
-        <div style="border: 1px solid #ddd; padding: 10px; margin: 10px; border-radius: 8px; display: flex; align-items: center;">
-            <img src="{item['Image1__c']}" alt="{item['Name']}" style="width: 100px; height: 100px; margin-right: 10px;">
+        menu_html += f"""
+        <div style="display: flex; align-items: center; margin-bottom: 20px;">
+            <img src="{item['Image1__c']}" style="width: 100px; height: 100px; margin-right: 20px;">
             <div>
                 <h4>{item['Name']}</h4>
                 <p>₹{item['Price__c']}</p>
-                <button onclick="updateCart('{item['Id']}', '{item['Name']}', {item['Price__c']}, -1)">-</button>
-                <span>0</span>
-                <button onclick="updateCart('{item['Id']}', '{item['Name']}', {item['Price__c']}, 1)">+</button>
+                <button onclick="add_to_cart('{item['Id']}', '{item['Name']}', {item['Price__c']})">Add</button>
             </div>
         </div>
         """
-    return html
+    return menu_html
 
-# Gradio App
-with gr.Blocks() as app:
-    menu_items = fetch_menu_items()
+# Fetch menu items for display
+menu_items = fetch_menu_items()
+menu_html = generate_menu_page(menu_items)
 
-    with gr.Row():
-        gr.HTML("<h1>Welcome to Biryani Hub</h1>")
+# Gradio interface for menu page
+def menu_interface():
+    return menu_html, gr.update(visible=False)
+
+def cart_interface():
+    return view_cart(), gr.update(visible=True)
 
+# Gradio setup
+with gr.Blocks() as demo:
     with gr.Row():
-        gr.HTML(generate_menu_html(menu_items))
+        gr.Markdown("# Welcome to Biryani Hub")
 
-    # Floating Cart Button
     with gr.Row():
-        cart_display = gr.HTML("0 items | Total: ₹0.00", elem_id="cart-summary")
-        gr.Button("View Cart", elem_id="view-cart-btn").click(cart_page, [], cart_display)
+        menu_display = gr.HTML(value=menu_html, interactive=False)
+        cart_button = gr.Button("View Cart", visible=True)
+        cart_display = gr.HTML(value=view_cart(), visible=False)
+    
+    cart_button.click(cart_interface, inputs=[], outputs=[menu_display, cart_display])
 
-app.launch()
+# Launch the app
+demo.launch()