Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, redirect, url_for
+from flask import Flask, render_template, redirect, request, url_for
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
@@ -29,7 +29,7 @@ except Exception as e:
 # Route for login page
 @app.route('/')
 def login():
-    return render_template('login.html', error_message=None)
+    return render_template('login.html')
 
 # Route to process login
 @app.route('/auth', methods=['POST'])
@@ -38,7 +38,7 @@ def auth():
     password = request.form['password']
 
     if not sf:
-        return render_template('login.html', error_message="Salesforce connection failed. Please try again.")
+        return "Salesforce connection failed. Please check credentials and try again."
 
     try:
         # Query Salesforce for user authentication
@@ -46,16 +46,15 @@ def auth():
         result = sf.query(query)
 
         if result['totalSize'] == 0:
-            return render_template('login.html', error_message="Invalid login details. Please check your email and password.")
+            return "Invalid Login Details"
 
         customer = result['records'][0]
         reward_points = customer['Reward_Points__c']
 
-        # Redirect to rewards page or any other page
+        # Redirect to rewards page
         return redirect(url_for('rewards', customer_id=customer['Id'], points=reward_points))
-
     except Exception as e:
-        return render_template('login.html', error_message=f"Error during authentication: {e}")
+        return f"Error during authentication: {e}"
 
 # Route to display rewards page
 @app.route('/rewards/<customer_id>')
@@ -69,5 +68,51 @@ def rewards(customer_id):
     except Exception as e:
         return f"Error fetching rewards: {e}"
 
+# Route to apply rewards
+@app.route('/apply_rewards', methods=['POST'])
+def apply_rewards():
+    customer_id = request.form['customer_id']
+    bill_amount = float(request.form['bill_amount'])
+    apply_rewards = request.form.get('apply_rewards')
+
+    try:
+        customer = sf.Customer_Login__c.get(customer_id)
+        points = customer['Reward_Points__c']
+        gst = 0.18 * bill_amount
+
+        if points >= 500 and apply_rewards:
+            discount = 0.1 * bill_amount
+            final_bill = bill_amount - discount + gst
+            updated_points = points - 500
+
+            # Update the customer's reward points in Salesforce
+            sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_points})
+
+            message = "You saved 10% on your total bill!"
+        else:
+            # Customers with below 500 points earn 10% of their bill amount as reward points
+            discount = 0
+            earned_points = 0.1 * bill_amount
+            final_bill = bill_amount + gst
+            updated_points = points + earned_points
+
+            # Update the customer's reward points in Salesforce
+            sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_points})
+
+            message = f"You earned 10% of your bill amount ({earned_points} points) as reward points!"
+
+        # Render the summary page with Dollar symbol
+        return render_template(
+            'apply_rewards.html',
+            original_bill=bill_amount,
+            discount=discount,
+            gst=gst,
+            final_bill=final_bill,
+            updated_points=updated_points,
+            message=message
+        )
+    except Exception as e:
+        return f"Error applying rewards: {e}"
+
 if __name__ == '__main__':
-    app.run(debug=True, host="0.0.0.0", port=7860)  # Ensure it runs on port 7860
+    app.run(debug=True, host="0.0.0.0", port=7860)