Update app.py

app.py

@@ -31,9 +31,8 @@ def authenticate_user(email, password):
         if result['totalSize'] == 0:
             return "Invalid Login Details", None, None, gr.update(visible=False), gr.update(visible=False)
 
-        # Ensure we get the correct record
         customer = result['records'][0]
-        reward_points = customer.get('Reward_Points__c', 0)  # Default to 0 if not available
+        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:
@@ -42,9 +41,8 @@ def authenticate_user(email, password):
 # Function to handle reward points logic
 def handle_rewards(customer_id, bill_amount, apply_rewards):
     try:
-        # Ensure the customer exists in Salesforce
         customer = sf.Customer_Login__c.get(customer_id)
-        points = customer.get('Reward_Points__c', 0)
+        points = customer['Reward_Points__c']
         gst = 0.18 * bill_amount
 
         if points >= 500 and apply_rewards:
@@ -66,9 +64,9 @@ def handle_rewards(customer_id, bill_amount, apply_rewards):
     except Exception as e:
         return f"Error applying rewards: {e}", 0, 0
 
-# Define the Gradio interface with page-like transitions using gr.update()
+# Define the Gradio interface with automatic page transitions
 def create_interface():
-    with gr.Blocks() as demo:
+    with gr.Page() as demo:
         # Login page
         with gr.Column(visible=True) as login_page:
             gr.Markdown("### Login to your account")
@@ -97,6 +95,18 @@ def create_interface():
                                  inputs=[customer_id_output, bill_amount_input, apply_rewards_checkbox], 
                                  outputs=[rewards_message, final_bill_output, remaining_points_output])
 
+        # Final summary page (After calculation)
+        with gr.Column(visible=False) as summary_page:
+            gr.Markdown("### Bill Summary")
+            summary_message = gr.Textbox(label="Summary Message")
+            final_amount_output = gr.Textbox(label="Final Amount")
+            remaining_points = gr.Textbox(label="Remaining Points")
+
+            # This page will show the message and final bill when handled in rewards calculation
+            rewards_button.click(handle_rewards, 
+                                 inputs=[customer_id_output, bill_amount_input, apply_rewards_checkbox], 
+                                 outputs=[summary_message, final_amount_output, remaining_points])
+
     return demo
 
 # Run the Gradio interface