Update app.py

app.py

@@ -22,21 +22,21 @@ except Exception as e:
 # 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)
+        return "Salesforce connection failed. Please check credentials and try again.", None, None
 
     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)
+            return "Invalid Login Details", None, None
 
         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)
+        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}", None, None, gr.update(visible=False), gr.update(visible=False)
+        return f"Error during authentication: {e}", None, None
 
 # Function to handle reward points logic
 def handle_rewards(customer_id, bill_amount, apply_rewards):
@@ -67,37 +67,49 @@ 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")
-        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])
-    
+        # Login page
+        with gr.Tab("Login"):
+            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
+
+            # Action for login button
+            login_button.click(authenticate_user, inputs=[email_input, password_input], outputs=[login_output, customer_id_output, reward_points_output])
+
+        # Reward points section (in next tab after login)
+        with gr.Tab("Reward Points"):
+            gr.Markdown("### Reward Points 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 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])
+
+        # Final summary page (after calculation)
+        with gr.Tab("Bill Summary"):
+            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
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch()
-