logindirectmenu2

Sleeping

App Files Files Community

geethareddy commited on Feb 15, 2025

Commit

6927368

verified ·

1 Parent(s): bd8aaca

Update app.py

Browse files

Files changed (1) hide show

app.py +249 -35

app.py CHANGED Viewed

@@ -1,8 +1,16 @@
-from flask import Flask, render_template, request, redirect, url_for, jsonify
 from simple_salesforce import Salesforce
 import os
 app = Flask(__name__)
 # Salesforce connection setup
 sf = Salesforce(
@@ -17,74 +25,280 @@ sf = Salesforce(
 cart = []  # To store items, quantities, and prices
 MENU = {}
 # Fetching menu items dynamically from Salesforce
 def get_menu_items():
     menu_items = {}
     query = "SELECT Name, Category__c, Price__c, Ingredients__c FROM Menu_Item__c"
     results = sf.query_all(query)
     for record in results['records']:
         category = record['Category__c']
         item_name = record['Name']
         price = record['Price__c']
         if category not in menu_items:
             menu_items[category] = {}
         menu_items[category][item_name] = price
     return menu_items
-# Fetch menu on app load
 MENU = get_menu_items()
-# Welcome page for registration/login
-@app.route("/", methods=["GET", "POST"])
-def welcome():
-    if request.method == "POST":
-        customer_type = request.form["customer_type"]
-        if customer_type == 'new':
-            return redirect(url_for('register'))  # Redirect to registration
-        elif customer_type == 'existing':
-            return redirect(url_for('login'))  # Redirect to login
-    return render_template("welcome.html")  # Render Welcome page
-# Register new customer
 @app.route("/register", methods=["GET", "POST"])
 def register():
     if request.method == "POST":
-        name = request.form["name"]
-        email = request.form["email"]
-        phone = request.form["phone"]
         if not name or not email or not phone:
-            return jsonify({"error": "All fields are required"}), 400
-        # Create Salesforce record for new customer
         try:
             customer_login = sf.Customer_Login__c.create({
                 'Name': name,
                 'Email__c': email,
                 'Phone_Number__c': phone
             })
-            return redirect(url_for('ai_dining_assistant'))  # Redirect to AI Dining Assistant page
         except Exception as e:
-            return jsonify({"error": f"Failed to create record: {str(e)}"}), 500
-    return render_template("register.html")  # Render registration page
-# Login existing customer
 @app.route("/login", methods=["GET", "POST"])
 def login():
     if request.method == "POST":
-        email = request.form["email"]
-        phone = request.form["phone"]
-        # You can implement logic to check the email and phone number with Salesforce records
-        return redirect(url_for('ai_dining_assistant'))  # Redirect to AI Dining Assistant page
-    return render_template("login.html")  # Render login page
-# AI Dining Assistant page for menu interaction
-@app.route("/ai_dining_assistant", methods=["GET"])
-def ai_dining_assistant():
-    return render_template("ai_dining_assistant.html", menu=MENU)
-# Run the Flask app
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)  # Enable debug mode

 from simple_salesforce import Salesforce
+import logging
+from flask import Flask, render_template_string, request, jsonify, redirect, url_for
+import speech_recognition as sr
+from tempfile import NamedTemporaryFile
 import os
+import ffmpeg
+from werkzeug.exceptions import BadRequest
+from gtts import gTTS
+import time
 app = Flask(__name__)
+logging.basicConfig(level=logging.INFO)
 # Salesforce connection setup
 sf = Salesforce(
 cart = []  # To store items, quantities, and prices
 MENU = {}
+current_category = None
+current_item = None
+awaiting_quantity = False
 # Fetching menu items dynamically from Salesforce
 def get_menu_items():
     menu_items = {}
+    # Querying the Menu_Item__c object to fetch details
     query = "SELECT Name, Category__c, Price__c, Ingredients__c FROM Menu_Item__c"
     results = sf.query_all(query)
+    # Organizing the menu items by category
     for record in results['records']:
         category = record['Category__c']
         item_name = record['Name']
         price = record['Price__c']
         if category not in menu_items:
             menu_items[category] = {}
         menu_items[category][item_name] = price
     return menu_items
+# Update global MENU variable
 MENU = get_menu_items()
+# HTML Template for Frontend (for register, login, and menu pages)
+html_code = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Register / Login</title>
+    <style>
+        body {
+            font-family: Arial, sans-serif;
+            text-align: center;
+            background-color: #f4f4f9;
+        }
+        h1 {
+            color: #333;
+        }
+        .status, .response {
+            margin-top: 20px;
+        }
+    </style>
+</head>
+<body>
+    <h1>Welcome to Biryani Hub</h1>
+    <div id="welcomePage">
+        <button onclick="startRegistration()">New Customer</button>
+        <button onclick="startLogin()">Existing Customer</button>
+    </div>
+    <div id="registrationPage" style="display: none;">
+        <h2>Register</h2>
+        <form action="/register" method="POST">
+            <label for="name">Name</label><br>
+            <input type="text" id="name" name="name" required><br>
+            <label for="email">Email</label><br>
+            <input type="email" id="email" name="email" required><br>
+            <label for="phone">Phone</label><br>
+            <input type="text" id="phone" name="phone" required><br>
+            <button type="submit">Register</button>
+        </form>
+    </div>
+    <div id="loginPage" style="display: none;">
+        <h2>Login</h2>
+        <form action="/login" method="POST">
+            <label for="email">Email</label><br>
+            <input type="email" id="email" name="email" required><br>
+            <label for="phone">Phone</label><br>
+            <input type="text" id="phone" name="phone" required><br>
+            <button type="submit">Login</button>
+        </form>
+    </div>
+    <div id="menuPage" style="display: none;">
+        <h1>AI Dining Assistant</h1>
+        <div class="status" id="status">Starting interaction...</div>
+        <div class="response" id="response" style="display: none;">Response will appear here...</div>
+        <button onclick="showMenu()">Show Menu</button>
+    </div>
+    <script>
+        function startRegistration() {
+            document.getElementById('welcomePage').style.display = 'none';
+            document.getElementById('registrationPage').style.display = 'block';
+        }
+        function startLogin() {
+            document.getElementById('welcomePage').style.display = 'none';
+            document.getElementById('loginPage').style.display = 'block';
+        }
+        function showMenu() {
+            document.getElementById('menuPage').style.display = 'block';
+            const categories = {{ MENU | tojson }};
+            let menuText = "We have the following categories: ";
+            for (let category in categories) {
+                menuText += category + " ";
+            }
+            document.getElementById('status').innerText = menuText;
+        }
+    </script>
+</body>
+</html>
+"""
+@app.route("/")
+def index():
+    return render_template_string(html_code)
+# Register Page
 @app.route("/register", methods=["GET", "POST"])
 def register():
     if request.method == "POST":
+        name = request.form.get('name')
+        email = request.form.get('email')
+        phone = request.form.get('phone')
         if not name or not email or not phone:
+            return jsonify({"error": "All fields are required."}), 400
+        # Create a record in Salesforce
         try:
             customer_login = sf.Customer_Login__c.create({
                 'Name': name,
                 'Email__c': email,
                 'Phone_Number__c': phone
             })
+            return redirect(url_for('menu'))  # Redirect to menu after successful registration
         except Exception as e:
+            return jsonify({'error': f'Failed to create record in Salesforce: {str(e)}'}), 500
+    return render_template_string(html_code)
+# Login Page
 @app.route("/login", methods=["GET", "POST"])
 def login():
     if request.method == "POST":
+        email = request.form.get('email')
+        phone = request.form.get('phone')
+        if not email or not phone:
+            return jsonify({"error": "Email and phone are required."}), 400
+        # Check if the customer exists in Salesforce
+        try:
+            customer_query = sf.query(f"SELECT Id, Name FROM Customer_Login__c WHERE Email__c = '{email}' AND Phone_Number__c = '{phone}'")
+            if customer_query['totalSize'] == 0:
+                return jsonify({"error": "Customer not found."}), 404
+            return redirect(url_for('menu'))  # Redirect to menu after successful login
+        except Exception as e:
+            return jsonify({'error': f'Failed to query Salesforce: {str(e)}'}), 500
+    return render_template_string(html_code)
+# Menu Page
+@app.route("/menu")
+def menu():
+    return render_template_string(html_code)
+@app.route("/process-audio", methods=["POST"])
+def process_audio():
+    global current_category, current_item, awaiting_quantity
+    try:
+        audio_file = request.files.get("audio")
+        if not audio_file:
+            raise BadRequest("No audio file provided.")
+        temp_file = NamedTemporaryFile(delete=False, suffix=".webm")
+        audio_file.save(temp_file.name)
+        if os.path.getsize(temp_file.name) == 0:
+            raise BadRequest("Uploaded audio file is empty.")
+        converted_file = NamedTemporaryFile(delete=False, suffix=".wav")
+        ffmpeg.input(temp_file.name).output(
+            converted_file.name, acodec="pcm_s16le", ac=1, ar="16000"
+        ).run(overwrite_output=True)
+        recognizer = sr.Recognizer()
+        with sr.AudioFile(converted_file.name) as source:
+            audio_data = recognizer.record(source)
+            try:
+                command = recognizer.recognize_google(audio_data)
+                logging.info(f"Recognized command: {command}")
+                response = process_command(command)
+            except sr.UnknownValueError:
+                response = "Sorry, I couldn't understand your command. Could you please repeat?"
+            except sr.RequestError as e:
+                response = f"Error with the speech recognition service: {e}"
+        return jsonify({"response": response})
+    except BadRequest as br:
+        return jsonify({"response": f"Bad Request: {str(br)}"}), 400
+    except Exception as e:
+        return jsonify({"response": f"An error occurred: {str(e)}"}), 500
+    finally:
+        os.unlink(temp_file.name)
+        os.unlink(converted_file.name)
+def process_command(command):
+    global cart, MENU, current_category, current_item, awaiting_quantity
+    command = command.lower()
+    # Handle quantity input
+    if awaiting_quantity:
+        quantity = extract_quantity(command)
+        if quantity:
+            cart.append((current_item, MENU[current_category][current_item], quantity))
+            awaiting_quantity = False
+            item = current_item
+            current_item = None
+            total = sum(i[1] * i[2] for i in cart)
+            cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
+            return f"Added {quantity} x {item} to your cart. Your current cart: {cart_summary}. Total: ₹{total}. Would you like to add more items?"
+        else:
+            return "Sorry, I couldn't understand the quantity. Please provide a valid quantity."
+    # Handle category selection
+    for category, items in MENU.items():
+        if category.lower() in command:
+            current_category = category
+            item_list = ", ".join([f"{item} (₹{price})" for item, price in items.items()])
+            return f"{category} menu: {item_list}. What would you like to order?"
+    # Handle item selection with dynamic matching
+    if current_category:
+        for item in MENU[current_category].keys():
+            if item.lower().startswith(command) or command in item.lower():
+                current_item = item
+                awaiting_quantity = True
+                return f"How many quantities of {current_item} would you like?"
+    # Handle item removal
+    if "remove" in command:
+        for item in cart:
+            if item[0].lower() in command:
+                cart.remove(item)
+                total = sum(i[1] * i[2] for i in cart)
+                cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
+                return f"Removed {item[0]} from your cart. Updated cart: {cart_summary}. Total: ₹{total}."
+        return "The item you are trying to remove is not in your cart."
+    # Handle final order
+    if "final order" in command or "submit" in command or "Proceed" in command or "place the order" in command:
+        if cart:
+            order_details = ", ".join([f"{item[0]} x{item[2]} (₹{item[1] * item[2]})" for item in cart])
+            total = sum(item[1] * item[2] for item in cart)
+            cart.clear()
+            return f"Your final order is: {order_details}. Total price: ₹{total}. Your order will arrive soon, Thank you for visiting Biryani Hub!"
+        else:
+            return "Your cart is empty. Please add items before placing the final order."
+    # Handle cart details
+    if "cart details" in command:
+        if cart:
+            cart_summary = "\n".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
+            total = sum(i[1] * i[2] for i in cart)
+            return f"Your cart contains:\n{cart_summary}\nTotal: ₹{total}."
+        else:
+            return "Your cart is empty."
+    # Handle menu request
+    if "menu" in command or "yes" in command or "yeah" in command:
+        categories = ", ".join(MENU.keys())
+        return f"We have the following categories: {categories}. Please select a category to proceed."
+    # Default response
+    return "Sorry, I didn't understand that. Please try again."
 if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)