Update flask-salesforce-app/app.py

flask-salesforce-app/app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, render_template, redirect, url_for
+import gradio as gr
 from simple_salesforce import Salesforce
 from dotenv import load_dotenv
 import os
@@ -6,9 +6,6 @@ import os
 # Load environment variables
 load_dotenv()
 
-# Initialize Flask app
-app = Flask(__name__)
-
 # Salesforce connection
 try:
     sf = Salesforce(
@@ -22,21 +19,12 @@ except Exception as e:
     print(f"Salesforce connection failed: {e}")
     sf = None
 
-@app.route('/')
-def login():
-    return render_template('login.html')
-
-
-@app.route('/auth', methods=['POST'])
-def auth():
-    email = request.form['email']
-    password = request.form['password']
-
+# 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."
 
     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}'"
         result = sf.query(query)
 
@@ -46,28 +34,12 @@ def auth():
         customer = result['records'][0]
         reward_points = customer['Reward_Points__c']
 
-        # Redirect to rewards page
-        return redirect(url_for('rewards', customer_id=customer['Id'], points=reward_points))
+        return f"Welcome, you have {reward_points} points. Proceed to rewards."
     except Exception as e:
         return f"Error during authentication: {e}"
 
-
-@app.route('/rewards/<customer_id>')
-def rewards(customer_id):
-    try:
-        customer = sf.Customer_Login__c.get(customer_id)
-        points = customer['Reward_Points__c']
-        return render_template('rewards.html', points=points, customer_id=customer_id)
-    except Exception as e:
-        return f"Error fetching rewards: {e}"
-
-
-@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']
@@ -78,7 +50,6 @@ def apply_rewards():
             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:
@@ -87,24 +58,43 @@ def apply_rewards():
             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 {earned_points:.2f} reward 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
-        )
+        return message, final_bill, updated_points
     except Exception as e:
-        return f"Error applying rewards: {e}"
-
-
-if __name__ == '__main__':
-    from waitress import serve
-    serve(app, host="0.0.0.0", port=8080)
+        return f"Error applying rewards: {e}", 0, 0
+
+# Define the Gradio interface
+def create_interface():
+    with gr.Blocks() as demo:
+        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")
+        
+        # Action for login button
+        login_button.click(authenticate_user, inputs=[email_input, password_input], outputs=login_output)
+        
+        # Reward points section
+        gr.Markdown("### Reward Points Section")
+        customer_id_input = gr.Textbox(label="Customer ID", placeholder="Enter Customer ID")
+        bill_amount_input = gr.Number(label="Enter Bill Amount", min=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 rewards calculation
+        rewards_button.click(handle_rewards, 
+                             inputs=[customer_id_input, 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()