Update app.py

app.py

@@ -2,10 +2,6 @@ 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()
@@ -20,14 +16,14 @@ SF_DOMAIN = os.getenv('SF_DOMAIN')
 
 # Salesforce connection
 try:
-    app.logger.debug("Attempting Salesforce connection...")
+    print("Attempting Salesforce connection...")
     sf = Salesforce(username=SF_USERNAME, 
                     password=SF_PASSWORD, 
                     security_token=SF_SECURITY_TOKEN, 
                     domain=SF_DOMAIN)
-    app.logger.debug("Salesforce connection successful!")
+    print("Salesforce connection successful!")
 except Exception as e:
-    app.logger.error(f"Salesforce connection failed: {e}")
+    print(f"Salesforce connection failed: {e}")
     sf = None
 
 # Route for login page
@@ -42,26 +38,22 @@ 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}' LIMIT 1"
+        query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}' AND Password__c = '{password}'"
         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
@@ -70,11 +62,10 @@ 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
@@ -88,7 +79,6 @@ 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
@@ -97,6 +87,7 @@ 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
@@ -107,9 +98,10 @@ 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
+        # Render the summary page with Dollar symbol
         return render_template(
             'apply_rewards.html',
             original_bill=bill_amount,
@@ -120,7 +112,6 @@ 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__':