Update app.py

app.py

@@ -1,47 +1,102 @@
-from flask import Flask, render_template, request, redirect, url_for, session
-import constants
-
-app = Flask(__name__)
-app.secret_key = 'your_secret_key_here'
-
-# Route for the login page
-@app.route('/')
-def login():
-    return render_template('login.html', style=constants.LOGIN_STYLE)
-
-# Handle login logic
-@app.route('/login', methods=['POST'])
-def handle_login():
-    email = request.form.get('email')
-    password = request.form.get('password')
-
-    # Authenticate user (this part will be based on Salesforce or any authentication method)
-    if email == "[email protected]" and password == "password123":  # Example check
-        session['user'] = email
-        return redirect(url_for('next_phase'))  # Move to the next phase after successful login
-    else:
-        return render_template('login.html', style=constants.LOGIN_STYLE, error="Invalid credentials!")
-
-# Route for the next phase (e.g., entering details)
-@app.route('/next_phase', methods=['GET', 'POST'])
-def next_phase():
-    if 'user' not in session:
-        return redirect(url_for('login'))
-
-    if request.method == 'POST':
-        # Save details (can be integrated with Salesforce or any DB)
-        # Redirect to reward points or next page
-        return redirect(url_for('reward_points'))
-
-    return render_template('next_phase.html', style=constants.PHASE_STYLE)
-
-# Route for the reward points page
-@app.route('/reward_points')
-def reward_points():
-    if 'user' not in session:
-        return redirect(url_for('login'))
-
-    return render_template('reward_points.html', style=constants.REWARD_STYLE)
-
-if __name__ == '__main__':
-    app.run(debug=True)
+import gradio as gr
+from simple_salesforce import Salesforce
+from dotenv import load_dotenv
+import os
+
+# Load environment variables
+load_dotenv()
+
+# 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!")
+except Exception as e:
+    print(f"Salesforce connection failed: {e}")
+    sf = None
+
+# Function to handle user authentication and Salesforce login
+def authenticate_user(email, password):
+    if not sf:
+        return "Salesforce connection failed. Please check credentials and try again.", None, None, gr.update(visible=False), gr.update(visible=False)
+
+    try:
+        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:
+            return "Invalid Login Details", None, None, gr.update(visible=False), gr.update(visible=False)
+
+        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)
+    except Exception as e:
+        return f"Error during authentication: {e}", None, None, gr.update(visible=False), gr.update(visible=False)
+
+# 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
+
+        if points >= 500 and apply_rewards:
+            discount = 0.1 * bill_amount
+            final_bill = bill_amount - discount + gst
+            updated_points = points - 500
+            message = "You saved 10% on your total bill!"
+        else:
+            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})
+
+        return message, final_bill, updated_points
+    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()