Update app.py

app.py

@@ -1,61 +1,291 @@
+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
 import os
-from flask import Flask, render_template, request, redirect
-from simple_salesforce import Salesforce
 
-# Initialize Flask app
+# Initialize Flask app and Salesforce connection
+print("Starting app...")
 app = Flask(__name__)
+print("Flask app initialized.")
 
-# Securely store the secret key
-app.secret_key = os.environ.get('FLASK_SECRET_KEY', 'your_secret_key')
-
-# Fetch Salesforce credentials correctly (QUOTED STRINGS!)
-SF_USERNAME = os.environ.get("SF_USERNAME", "[email protected]")
-SF_PASSWORD = os.environ.get("SF_PASSWORD", "Sati@1020")
-SF_SECURITY_TOKEN = os.environ.get("SF_SECURITY_TOKEN", "sSSjyhInIsUohKpG8sHzty2q")
-
-# Debugging: Print credentials safely
-print(f"Using Salesforce Username: {SF_USERNAME}")
-
-# Establish Salesforce connection
-try:
-    sf = Salesforce(
-        username=SF_USERNAME,
-        password=SF_PASSWORD,
-        security_token=SF_SECURITY_TOKEN
-    )
-    print("✅ Salesforce connection established successfully.")
-except Exception as e:
-    print(f"❌ ERROR: Failed to connect to Salesforce: {str(e)}")
-    exit(1)
+# Add debug logs in Salesforce connection setup
+sf = get_salesforce_connection()
+print("Salesforce connection established.")
+
+# Set the secret key to handle sessions securely
+app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")  # Replace with a secure key
+
+# 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 redirect("/menu")  # Redirect home page to menu page
+    #return "Welcome to Biryani Hub!"
+    return render_template("index.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")
 def menu():
     selected_category = request.args.get("category", "All")
+
     try:
-        # Salesforce SOQL Query
         query = "SELECT Name, Price__c, Image1__c, Category__c, Description__c FROM Menu_Item__c"
         result = sf.query(query)
-
-        # Extract menu items
-        food_items = result.get('records', [])
-
-        # Extract unique categories
+        food_items = result['records'] if 'records' in result else []
         categories = {item['Category__c'] for item in food_items if 'Category__c' in item}
-
-        # Filter by category if selected
+        
         if selected_category != "All":
             food_items = [item for item in food_items if item.get("Category__c") == selected_category]
-            
     except Exception as e:
         food_items = []
         categories = []
-        print(f"❌ ERROR: Fetching data from Salesforce failed: {e}")
+        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=True)
+   app.run(host='0.0.0.0', port=7860, debug=False)
+