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
+        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
+            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
+        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
+        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):
@@ -76,11 +76,9 @@ def create_interface():
         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 - no customer ID input
-        gr.Markdown("### Reward Points Section")
+        # 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")
@@ -88,6 +86,9 @@ def create_interface():
         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], 
@@ -99,4 +100,3 @@ def create_interface():
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch()
-