Update app.py (#2)

- Update app.py (66af2ab49457932d9f40fbcea1dacfe5d9b0b21e)


Co-authored-by: Surendra <[email protected]>

app.py

@@ -1,88 +1,21 @@
-from flask import Flask, render_template, request, jsonify, redirect, url_for, session
-from flask_session import Session  # Import the Session class
-from flask.sessions import SecureCookieSessionInterface  # Import the class
-from salesforce import get_salesforce_connection
+from flask import Flask, render_template, request
+from simple_salesforce import Salesforce
 import os
 
-# Initialize Flask app and Salesforce connection
-print("Starting app...")
 app = Flask(__name__)
-print("Flask app initialized.")
 
-# Add debug logs in Salesforce connection setup
-sf = get_salesforce_connection()
-print("Salesforce connection established.")
+# Salesforce credentials
+SF_USERNAME = os.getenv("SF_USERNAME", "your_username")
+SF_PASSWORD = os.getenv("SF_PASSWORD", "your_password")
+SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN", "your_security_token")
 
-# Set the secret key to handle sessions securely
-app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")  # Replace with a secure key
+# Connect to Salesforce
+sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN)
 
-# Configure the session type
-app.config["SESSION_TYPE"] = "filesystem"  # Use filesystem for session storage
-#app.config["SESSION_COOKIE_NAME"] = "my_session"  # Optional: Change session cookie name
-app.config["SESSION_COOKIE_SECURE"] = True  # Ensure cookies are sent over HTTPS
-app.config["SESSION_COOKIE_SAMESITE"] = "None"  # Allow cross-site cookies
-
-# Initialize the session
-Session(app)  # Correctly initialize the Session object
-print("Session interface configured.")
-
-# Ensure secure session handling for environments like Hugging Face
-app.session_interface = SecureCookieSessionInterface()
-print("Session interface configured.")
-
-@app.route("/")
-def home():
-    #return "Welcome to Biryani Hub!"
-    return render_template("menu.html")
-
-@app.route("/signup", methods=["GET", "POST"])
-def signup():
-    if request.method == "POST":
-        name = request.form.get("name")
-        phone = request.form.get("phone")
-        email = request.form.get("email")
-        password = request.form.get("password")
-        try:
-            sf.Customer_Login__c.create({
-                "Name": name,
-                "Phone_Number__c": phone,
-                "Email__c": email,
-                "Password__c": password
-            })
-            return redirect(url_for("login"))
-        except Exception as e:
-            return render_template("signup.html", error=f"Error: {str(e)}")
-    return render_template("signup.html")
-
-@app.route("/login", methods=["GET", "POST"])
-def login():
-    if request.method == "POST":
-        email = request.form.get("email")
-        password = request.form.get("password")
-        try:
-            query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
-            result = sf.query(query)
-            if result["records"]:
-                session['user_id'] = result["records"][0]['Id']
-                session['user_email'] = email
-                return redirect(url_for("menu"))
-            else:
-                return render_template("login.html", error="Invalid credentials!")
-        except Exception as e:
-            return render_template("login.html", error=f"Error: {str(e)}")
-    return render_template("login.html")
-
-@app.route("/logout", methods=["POST"])
-def logout():
-    session.clear()  # Clears the session to log the user out
-    return redirect(url_for('login'))  # Redirect to the login page
-
-@app.route("/menu", methods=["GET", "POST"])
-@app.route("/menu")
-@app.route("/menu")
+@app.route("/menu", methods=["GET"])
 def menu():
     selected_category = request.args.get("category", "All")
-
+    
     try:
         query = "SELECT Name, Price__c, Image1__c, Category__c, Description__c FROM Menu_Item__c"
         result = sf.query(query)
@@ -95,197 +28,8 @@ def menu():
         food_items = []
         categories = []
         print(f"Error fetching data: {e}")
-
+    
     return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)
-def cart():
-    email = session.get('user_email') # Get logged-in user's email
-    if not email:
-        return redirect(url_for("login")) # Redirect to login if not logged in
-    try:
-        result = sf.query(f"""
-            SELECT Name, Price__c, Quantity__c, Add_Ons__c, Image1__c
-            FROM Cart_Item__c
-            WHERE Customer_Email__c = '{email}'
-        """)
-        cart_items = result.get("records", [])
-        subtotal = sum(item['Quantity__c'] * item['Price__c'] for item in cart_items)
-    except Exception as e:
-        print(f"Error fetching cart items: {e}")
-        cart_items = []
-        subtotal = 0
-        
-    return render_template("cart.html", cart_items=cart_items, subtotal=subtotal)
-
-
-
-@app.route('/cart/add', methods=['POST'])
-def add_to_cart():
-    data = request.json
-    item_name = data.get('itemName')
-    item_price = data.get('itemPrice')
-    item_image = data.get('itemImage')
-    addons = data.get('addons', [])
-    customer_email = session.get('user_email')
-
-    if not customer_email:
-        return jsonify({"success": False, "error": "User not logged in."}), 401
-
-    try:
-        query = f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'"
-        result = sf.query(query)
-
-        if result['totalSize'] > 0:
-            # Update existing item quantity
-            cart_item = result['records'][0]
-            sf.Cart_Item__c.update(cart_item['Id'], {
-                "Quantity__c": cart_item['Quantity__c'] + 1
-            })
-        else:
-            # Add new item to the cart
-            sf.Cart_Item__c.create({
-                "Name": item_name,
-                "Price__c": item_price,
-                "Quantity__c": 1,
-                "Add_Ons__c": ";".join(addons) if addons else None,
-                "Image1__c": item_image,
-                "Customer_Email__c": customer_email,
-            })
-        return jsonify({"success": True, "message": "Item added to cart."})
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)}), 500
-
-@app.route("/cart/add_item", methods=["POST"])
-def add_item_to_cart():
-    data = request.json  # Extract JSON data from the request
-    email = data.get('email')  # Customer email
-    item_name = data.get('item_name')  # Item name
-    quantity = data.get('quantity', 0)  # Quantity to add (default is 1) // default value is 1
-
-    try:
-        # Check if the item already exists in the cart for this customer
-        cart_items = sf.query(
-            f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Item_Name__c = '{item_name}'"
-        )['records']
-
-        if cart_items:
-            # If the item already exists, update its quantity
-            cart_item = cart_items[0]
-            new_quantity = cart_item['Quantity__c'] + quantity
-            sf.Cart_Item__c.update(cart_item['Id'], {"Quantity__c": new_quantity})
-            return jsonify({"success": True, "message": "Item quantity updated successfully."})
-        else:
-            # If the item does not exist, add it to the cart
-            sf.Cart_Item__c.create({
-                "Customer_Email__c": email,
-                "Item_Name__c": item_name,
-                "Quantity__c": quantity
-            })
 
-        return jsonify({"success": True, "message": "Item added/updated successfully.", "redirect": "/menu"})
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)}), 500
-
-
-@app.route('/cart/remove/<item_name>', methods=['POST'])
-def remove_cart_item(item_name):
-    try:
-        customer_email = session.get('user_email')
-        if not customer_email:
-            return jsonify({'success': False, 'message': 'User email not found. Please log in again.'}), 400
-        query = f"""
-            SELECT Id FROM Cart_Item__c 
-            WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
-        """
-        result = sf.query(query)
-        if result['totalSize'] == 0:
-            return jsonify({'success': False, 'message': 'Item not found in cart.'}), 400
-        cart_item_id = result['records'][0]['Id']
-        sf.Cart_Item__c.delete(cart_item_id)
-        return jsonify({'success': True, 'message': f"'{item_name}' removed successfully!"}), 200
-    except Exception as e:
-        print(f"Error: {str(e)}")
-        return jsonify({'success': False, 'message': f"An error occurred: {str(e)}"}), 500
-
-@app.route('/api/addons', methods=['GET'])
-def get_addons():
-    item_name = request.args.get('item_name')
-    if not item_name:
-        return jsonify({"success": False, "error": "Item name is required."})
-
-    try:
-        # Salesforce query to get the add-ons related to the item name
-        query = f"SELECT Name, Price__c FROM Add_Ons__c"
-        addons = sf.query(query)['records']# Get records of add-ons
-        return jsonify({"success": True, "addons": addons})
-    except Exception as e:
-        print(f"Error fetching add-ons: {e}")
-        return jsonify({"success": False, "error": "Unable to fetch add-ons. Please try again later."})
-
-@app.route("/cart/update_quantity", methods=["POST"])
-def update_quantity():
-    data = request.json  # Extract JSON data from the request
-    email = data.get('email')  # Customer email
-    item_name = data.get('item_name')  # Item name (Cart Item Name in Salesforce)
-    quantity = data.get('quantity')  # New quantity
-
-    # Validate inputs
-    if not email or not item_name:
-        return jsonify({"success": False, "error": "Email and item name are required."}), 400
-
-    try:
-        # Query the cart item using the correct field names
-        cart_items = sf.query(
-            f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name__c = '{item_name}'"
-        )['records']
-
-        if not cart_items:
-            return jsonify({"success": False, "error": "Cart item not found."}), 404
-
-        # Get the first matching record ID
-        cart_item_id = cart_items[0]['Id']
-
-        # Update the quantity in Salesforce
-        sf.Cart_Item__c.update(cart_item_id, {"Quantity__c": quantity})
-
-        return jsonify({"success": True, "new_quantity": quantity})
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)}), 500
-
-@app.route("/checkout", methods=["POST"])
-def checkout():
-    email = session.get('user_email')
-    user_id = session.get('user_id')
-    if not email or not user_id:
-        return jsonify({"success": False, "message": "User not logged in"})
-    try:
-        result = sf.query(f"""
-            SELECT Id, Name, Price__c, Quantity__c, Add_Ons__c
-            FROM Cart_Item__c
-            WHERE Customer_Email__c = '{email}'
-        """)
-        cart_items = result["records"]
-        if not cart_items:
-            return jsonify({"success": False, "message": "Cart is empty"})
-        total_price = sum(item['Quantity__c'] * item['Price__c'] for item in cart_items)
-        order_data = {
-            "Customer_Name__c": user_id,
-            "Customer_Email__c": email,
-            "Total_Amount__c": total_price,
-            "Order_Status__c": "Pending",
-            "Order_Items__c": "\n".join(
-                [f"{item['Name']} (Qty: {item['Quantity__c']})" for item in cart_items]
-            ),
-            "Add_Ons__c": "\n".join(
-                [f"{item['Add_Ons__c']}" if item['Add_Ons__c'] else "None" for item in cart_items]
-            ),
-        }
-        sf.Order__c.create(order_data)
-        for item in cart_items:
-            sf.Cart_Item__c.delete(item["Id"])
-        return jsonify({"success": True, "message": "Order placed successfully!"})
-    except Exception as e:
-        return jsonify({"success": False, "error": str(e)})
-        
 if __name__ == "__main__":
-   app.run(host='0.0.0.0', port=7860, debug=False)
-
+    app.run(debug=True)