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."
+        return "Salesforce connection failed. Please check credentials and try again.", "", 0
 
     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"
+            return "Invalid Login Details", "", 0
 
         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
     except Exception as e:
-        return f"Error during authentication: {e}"
+        return f"Error during authentication: {e}", "", 0
 
 # Function to handle reward points logic
 def handle_rewards(customer_id, bill_amount, apply_rewards):
@@ -73,10 +73,12 @@ def create_interface():
         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", "points"])
-
+        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")
         bill_amount_input = gr.Number(label="Enter Bill Amount", min=0)
@@ -88,7 +90,7 @@ def create_interface():
 
         # Action for rewards calculation
         rewards_button.click(handle_rewards, 
-                             inputs=["customer_id", bill_amount_input, apply_rewards_checkbox], 
+                             inputs=[customer_id_output, bill_amount_input, apply_rewards_checkbox], 
                              outputs=[rewards_message, final_bill_output, remaining_points_output])
     
     return demo
@@ -96,4 +98,4 @@ def create_interface():
 # Run the Gradio interface
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()