Update app.py

app.py

@@ -34,7 +34,7 @@ def authenticate_user(email, password):
         customer = result['records'][0]
         reward_points = customer['Reward_Points__c']
 
-        return f"Welcome, you have {reward_points} points. Proceed to rewards."
+        return f"Welcome, you have {reward_points} points. Proceed to rewards.", customer['Id'], reward_points
     except Exception as e:
         return f"Error during authentication: {e}"
 
@@ -49,18 +49,17 @@ def handle_rewards(customer_id, bill_amount, apply_rewards):
             discount = 0.1 * bill_amount
             final_bill = bill_amount - discount + gst
             updated_points = points - 500
-
-            sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_points})
             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
-
-            sf.Customer_Login__c.update(customer_id, {'Reward_Points__c': updated_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
@@ -68,28 +67,28 @@ def handle_rewards(customer_id, bill_amount, apply_rewards):
 # 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")
-        
+
         # Action for login button
-        login_button.click(authenticate_user, inputs=[email_input, password_input], outputs=login_output)
-        
-        # Reward points section
+        login_button.click(authenticate_user, inputs=[email_input, password_input], outputs=[login_output, "customer_id", "points"])
+
+        # Reward points section - no customer ID input
         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", minimum=0)  # Use `minimum` instead of `min`
+        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], 
+                             inputs=["customer_id", bill_amount_input, apply_rewards_checkbox], 
                              outputs=[rewards_message, final_bill_output, remaining_points_output])
     
     return demo