Update app.py

app.py

@@ -1,102 +1,127 @@
-import gradio as gr
+from flask import Flask, render_template, redirect, request, url_for
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
+import logging
 
-# Load environment variables
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+
+# Load environment variables from .env file
 load_dotenv()
 
+app = Flask(__name__)
+
+# Get Salesforce credentials from environment variables
+SF_USERNAME = os.getenv('SF_USERNAME')
+SF_PASSWORD = os.getenv('SF_PASSWORD')
+SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
+SF_DOMAIN = os.getenv('SF_DOMAIN')
+
 # Salesforce connection
 try:
-    sf = Salesforce(
-        username=os.getenv("SF_USERNAME"),
-        password=os.getenv("SF_PASSWORD"),
-        security_token=os.getenv("SF_SECURITY_TOKEN"),
-        domain="login"  # Use "test" for sandbox
-    )
-    print("Salesforce connection successful!")
+    app.logger.debug("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!")
 except Exception as e:
-    print(f"Salesforce connection failed: {e}")
+    app.logger.error(f"Salesforce connection failed: {e}")
     sf = None
 
-# Function to handle user authentication and Salesforce login
-def authenticate_user(email, password):
+# Route for login page
+@app.route('/')
+def login():
+    return render_template('login.html')
+
+# Route to process login
+@app.route('/auth', methods=['POST'])
+def auth():
+    email = request.form['email']
+    password = request.form['password']
+
     if not sf:
-        return "Salesforce connection failed. Please check credentials and try again.", None, None, gr.update(visible=False), gr.update(visible=False)
+        app.logger.error("Salesforce connection failed. Please check credentials and try again.")
+        return "Salesforce connection failed. Please check credentials and try again."
 
     try:
-        query = f"SELECT Id, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{email}' AND Password__c = '{password}'"
+        # 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"
         result = sf.query(query)
 
         if result['totalSize'] == 0:
-            return "Invalid Login Details", None, None, gr.update(visible=False), gr.update(visible=False)
+            app.logger.error("Invalid Login Details")
+            return "Invalid Login Details"
 
         customer = result['records'][0]
         reward_points = customer['Reward_Points__c']
 
-        return f"Welcome, you have {reward_points} points. Proceed to rewards.", customer['Id'], reward_points, gr.update(visible=True), gr.update(visible=True)
+        # 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:
-        return f"Error during authentication: {e}", None, None, gr.update(visible=False), gr.update(visible=False)
+        app.logger.error(f"Error during authentication: {e}")
+        return f"Error during authentication: {e}"
+
+# Route to display rewards page
+@app.route('/rewards/<customer_id>')
+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
+@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')
 
-# Function to handle reward points logic
-def handle_rewards(customer_id, bill_amount, apply_rewards):
     try:
         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
             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
-            message = f"You earned {earned_points:.2f} reward points!"
 
-        # Update the customer's reward points in Salesforce
-        sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_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!"
 
-        return message, final_bill, updated_points
+        # Render the summary page
+        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}", 0, 0
-
-# Define the Gradio interface
-def create_interface():
-    with gr.Blocks() as demo:
-        # Login section
-        gr.Markdown("### Login to your account")
-        email_input = gr.Textbox(label="Email", placeholder="Enter your email")
-        password_input = gr.Textbox(label="Password", placeholder="Enter your password", type="password")
-        login_button = gr.Button("Login")
-        login_output = gr.Textbox(label="Status")
-        customer_id_output = gr.Textbox(label="Customer ID", visible=False)  # Hidden
-        reward_points_output = gr.Textbox(label="Available Reward Points", visible=False)  # Hidden
-        
-        # Reward points section (Initially hidden)
-        reward_section = gr.Column(visible=False)
-        gr.Markdown("### Reward Points Section", elem_id="reward_section")
-        bill_amount_input = gr.Number(label="Enter Bill Amount", value=0)
-        apply_rewards_checkbox = gr.Checkbox(label="Apply Reward Points", value=True)
-        rewards_button = gr.Button("Calculate Bill")
-        rewards_message = gr.Textbox(label="Message")
-        final_bill_output = gr.Textbox(label="Final Bill Amount")
-        remaining_points_output = gr.Textbox(label="Remaining Points")
-
-        # Action for login button
-        login_button.click(authenticate_user, inputs=[email_input, password_input], outputs=[login_output, customer_id_output, reward_points_output, reward_section, bill_amount_input])
-
-        # Action for rewards calculation
-        rewards_button.click(handle_rewards, 
-                             inputs=[customer_id_output, bill_amount_input, apply_rewards_checkbox], 
-                             outputs=[rewards_message, final_bill_output, remaining_points_output])
-    
-    return demo
-
-# Run the Gradio interface
-if __name__ == "__main__":
-    demo = create_interface()
-    demo.launch()
+        app.logger.error(f"Error applying rewards: {e}")
+        return f"Error applying rewards: {e}"
+
+if __name__ == '__main__':
+    app.run(debug=True, host="0.0.0.0", port=5000)