Update app.py

app.py

@@ -12,19 +12,23 @@ import time
 app = Flask(__name__)
 logging.basicConfig(level=logging.DEBUG)
 
-# Salesforce connection setup
-try:
-    sf = Salesforce(
-        username='diggavalli98@gmail.com', 
-        password='Sati@1020', 
-        security_token='sSSjyhInIsUohKpG8sHzty2q', 
-        consumer_key='3MVG9WVXk15qiz1JbtW1tT9a7Wnkos2RuGamw6p1lC5uPescT5NB2nPygpo6rQ87K1T.zBEn.wR.A6JdgHnIU', 
-        consumer_secret='A75C6B7801D5D20BED0E46631CF58C4F7FF28E4DAF442FE667553D29C35C0451'
-    )
-    logging.info("Successfully connected to Salesforce")
-except Exception as e:
-    logging.error(f"Error connecting to Salesforce: {str(e)}")
-    sf = None
+# Salesforce connection setup with error handling
+def connect_to_salesforce():
+    try:
+        sf = Salesforce(
+            username='diggavalli98@gmail.com', 
+            password='Sati@1020', 
+            security_token='sSSjyhInIsUohKpG8sHzty2q', 
+            consumer_key='3MVG9WVXk15qiz1JbtW1tT9a7Wnkos2RuGamw6p1lC5uPescT5NB2nPygpo6rQ87K1T.zBEn.wR.A6JdgHnIU', 
+            consumer_secret='A75C6B7801D5D20BED0E46631CF58C4F7FF28E4DAF442FE667553D29C35C0451'
+        )
+        logging.info("Successfully connected to Salesforce")
+        return sf
+    except Exception as e:
+        logging.error(f"Error connecting to Salesforce: {str(e)}")
+        return None
+
+sf = connect_to_salesforce()
 
 # Global variables
 cart = []  # To store items, quantities, and prices
@@ -132,185 +136,4 @@ html_code = """
         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
-            })
-            logging.info("Customer registered successfully")
-            return redirect(url_for('menu'))  # Redirect to menu after successful registration
-        except Exception as e:
-            logging.error(f"Error during registration: {str(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
-            logging.info("Customer logged in successfully")
-            return redirect(url_for('menu'))  # Redirect to menu after successful login
-        except Exception as e:
-            logging.error(f"Error during login: {str(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:
-        logging.error(f"Error processing audio: {str(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)
+            let menuText = "We ha