Update app.py

app.py

@@ -2,6 +2,10 @@ from flask import Flask, render_template, redirect, request, url_for
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
 
 # Load environment variables from .env file
 load_dotenv()
@@ -16,13 +20,14 @@ SF_DOMAIN = os.getenv('SF_DOMAIN')
 
 # Salesforce connection
 try:
+    app.logger.debug("Attempting Salesforce connection...")
     sf = Salesforce(username=SF_USERNAME, 
                     password=SF_PASSWORD, 
                     security_token=SF_SECURITY_TOKEN, 
                     domain=SF_DOMAIN)
-    print("Salesforce connection successful!")
+    app.logger.debug("Salesforce connection successful!")
 except Exception as e:
-    print(f"Salesforce connection failed: {e}")
+    app.logger.error(f"Salesforce connection failed: {e}")
     sf = None
 
 # Route for login page
@@ -37,22 +42,26 @@ def auth():
     password = request.form['password']
 
     if not sf:
+        app.logger.error("Salesforce connection failed. Please check credentials and try again.")
         return "Salesforce connection failed. Please check credentials and try again."
 
     try:
         # Query Salesforce for user authentication
-        query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}' AND Password__c = '{password}'"
+        query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}' AND Password__c = '{password}' LIMIT 1"
         result = sf.query(query)
 
         if result['totalSize'] == 0:
+            app.logger.error("Invalid Login Details")
             return "Invalid Login Details"
 
         customer = result['records'][0]
         reward_points = customer['Reward_Points__c']
 
         # Redirect to rewards page
+        app.logger.debug(f"Redirecting to rewards page with customer_id: {customer['Id']} and points: {reward_points}")
         return redirect(url_for('rewards', customer_id=customer['Id'], points=reward_points))
     except Exception as e:
+        app.logger.error(f"Error during authentication: {e}")
         return f"Error during authentication: {e}"
 
 # Route to display rewards page
@@ -61,10 +70,11 @@ def rewards(customer_id):
     try:
         customer = sf.Customer_Login__c.get(customer_id)
         points = customer['Reward_Points__c']
-
+        app.logger.debug(f"Fetched reward points: {points} for customer_id: {customer_id}")
         # Render the rewards page
         return render_template('rewards.html', points=points, customer_id=customer_id)
     except Exception as e:
+        app.logger.error(f"Error fetching rewards: {e}")
         return f"Error fetching rewards: {e}"
 
 # Route to apply rewards
@@ -78,6 +88,7 @@ def apply_rewards():
         customer = sf.Customer_Login__c.get(customer_id)
         points = customer['Reward_Points__c']
         gst = 0.18 * bill_amount
+        app.logger.debug(f"Processing bill amount: {bill_amount}, GST: {gst}, Points: {points}")
 
         if points >= 500 and apply_rewards:
             discount = 0.1 * bill_amount
@@ -86,7 +97,6 @@ def apply_rewards():
 
             # 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
@@ -97,7 +107,6 @@ def apply_rewards():
 
             # 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
@@ -111,7 +120,8 @@ def apply_rewards():
             message=message
         )
     except Exception as e:
+        app.logger.error(f"Error applying rewards: {e}")
         return f"Error applying rewards: {e}"
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True, host="0.0.0.0", port=5000)