Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, redirect
 import json
 import os
 from transformers import pipeline
@@ -82,6 +82,12 @@ try:
 except Exception as e:
     print(f"Failed to connect to Salesforce: {str(e)}")
 
+    # Function to fetch menu items from Salesforce
+def get_menu_items():
+    query = "SELECT Name, Price__c, Ingredients__c, Category__c FROM Menu_Item__c"
+    result = sf.query(query)
+    return result['records']
+
 # Function to create Salesforce record
 # API endpoint to receive data from voice bot
 @app.route('/login', methods=['POST'])
@@ -104,6 +110,8 @@ def login():
             'Phone_Number__c': phone_number
         })
         return jsonify({'success': True, 'id': customer_login['id']}), 200
+        return redirect("/menu") # Redirect to the menu page after successful login
+         return render_template("login.html")  # Render the login page for GET requests
     except Exception as e:
         return jsonify({'error': f'Failed to create record in Salesforce: {str(e)}'}), 500
 
@@ -138,6 +146,56 @@ def submit():
 def index():
     return render_template("index.html")
 
+    # Route for the menu page
+@app.route("/menu", methods=["GET"])
+def menu_page():
+    menu_items = get_menu_items()  # Fetch menu items from Salesforce
+    menu_data = []
+    for item in menu_items:
+        menu_data.append({
+            "name": item['Name'],
+            "price": item['Price__c'],
+            "ingredients": item['Ingredients__c'],
+            "category": item['Category__c'],
+        })
+
+    # Render the menu page template and pass the menu data to it
+    return render_template("menu_page.html", menu_items=menu_data)
+
+# Route for handling order (you'll save the order to Salesforce here)
+@app.route("/order", methods=["POST"])
+def place_order():
+    # Getting the item name from the POST request (you can customize this to get more data)
+    item_name = request.json.get('item_name')
+    quantity = request.json.get('quantity')
+
+    # Logic to save the order details to Salesforce
+    order_data = {
+        "Item__c": item_name,  # Assuming 'Item__c' is the field in your Order object
+        "Quantity__c": quantity,  # Assuming 'Quantity__c' is a field for the quantity
+        # Add any other fields you need to capture, like customer, price, etc.
+    }
+    
+    # Create the order record in Salesforce
+    sf.Order__c.create(order_data)
+
+    # Return a success response
+    return jsonify({"success": True, "message": f"Order for {item_name} placed successfully."})
+
+# Route to handle the cart (this can be expanded later)
+@app.route("/cart", methods=["GET"])
+def cart():
+    # Logic to fetch cart items (this could be from a session or database)
+    cart_items = []  # Placeholder for cart items
+    return render_template("cart_page.html", cart_items=cart_items)
+
+# Route for the order summary page (this can be expanded later)
+@app.route("/order-summary", methods=["GET"])
+def order_summary():
+    # Logic to fetch order summary (this could be from a session or database)
+    order_details = []  # Placeholder for order details
+    return render_template("order_summary.html", order_details=order_details)
+
 @app.route("/transcribe", methods=["POST"])
 def transcribe():
     if "audio" not in request.files: