Update app.py

app.py

@@ -1,29 +1,123 @@
 import bcrypt
 import gradio as gr
+from simple_salesforce import Salesforce
 
-# Dummy data for testing (replace Salesforce queries temporarily)
+# Salesforce Connection
+sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
+
+# Function to Hash Password
+def hash_password(password):
+    return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
+
+# Function to Verify Password
+def verify_password(plain_password, hashed_password):
+    return bcrypt.checkpw(plain_password.encode('utf-8'), hashed_password.encode('utf-8'))
+
+# Signup function
+def signup(name, email, phone, password):
+    try:
+        email = email.strip()
+
+        # Check if the email already exists in Salesforce
+        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
+        result = sf.query(query)
+
+        if len(result['records']) > 0:
+            return "Email already exists! Please use a different email."
+
+        # Hash the password
+        hashed_password = hash_password(password)
+
+        # Create the new user record
+        sf.Customer_Login__c.create({
+            'Name': name.strip(),
+            'Email__c': email,
+            'Phone_Number__c': phone.strip(),
+            'Password__c': hashed_password
+        })
+        return "Signup successful! You can now login."
+    except Exception as e:
+        return f"Error during signup: {str(e)}"
+
+# Login function
+def login(email, password):
+    try:
+        email = email.strip()
+        password = password.strip()
+
+        # Debug: Print entered email and password
+        print(f"Entered Email: {email}")
+        print(f"Entered Password: {password}")
+
+        # Query Salesforce for user details
+        query = f"SELECT Name, Password__c FROM Customer_Login__c WHERE Email__c = '{email}'"
+        result = sf.query(query)
+
+        # Debug: Print query results
+        print(f"Query Result: {result}")
+
+        if len(result['records']) == 0:
+            print("No matching user found.")
+            return "Invalid email or password.", None
+
+        user = result['records'][0]
+        stored_password = user['Password__c']
+
+        # Debug: Print stored hashed password
+        print(f"Stored Hashed Password: {stored_password}")
+
+        # Verify the entered password with the stored hash
+        if verify_password(password, stored_password):
+            print("Password matched!")
+            return "Login successful!", user['Name']
+        else:
+            print("Password did not match!")
+            return "Invalid email or password.", None
+    except Exception as e:
+        print(f"Error during login: {e}")
+        return f"Error during login: {str(e)}", None
+
+# Dummy function to load menu data
 def load_menu_from_salesforce():
-    # Replace with a simple dummy dataset to test UI functionality
-    return [
-        {"Name": "Chicken Biryani", "Price__c": "8.99", "Description__c": "Delicious chicken biryani", 
-         "Image1__c": "https://via.placeholder.com/200", "Veg_NonVeg__c": "Non-Veg", "Section__c": "Biryanis"},
-        {"Name": "Veg Biryani", "Price__c": "6.99", "Description__c": "Healthy veg biryani", 
-         "Image1__c": "https://via.placeholder.com/200", "Veg_NonVeg__c": "Veg", "Section__c": "Biryanis"},
-        {"Name": "Paneer Tikka", "Price__c": "5.99", "Description__c": "Grilled paneer tikka", 
-         "Image1__c": "https://via.placeholder.com/200", "Veg_NonVeg__c": "Veg", "Section__c": "Starters"},
-        {"Name": "Chicken Wings", "Price__c": "7.99", "Description__c": "Spicy chicken wings", 
-         "Image1__c": "https://via.placeholder.com/200", "Veg_NonVeg__c": "Non-Veg", "Section__c": "Starters"}
-    ]
+    try:
+        query = "SELECT Name, Price__c, Description__c, Image1__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
+        result = sf.query(query)
+
+        # Debugging: Print fetched records
+        print("Fetched Records from Salesforce:")
+        for record in result['records']:
+            print(record)
+
+        return result['records']
+    except Exception as e:
+        print(f"Error fetching menu data: {e}")
+        return []
 
 # Function to filter menu items based on preference
 def filter_menu(preferences):
     menu_data = load_menu_from_salesforce()
 
+    # Debugging: Print the fetched menu data
+    print("Menu Data Fetched:")
+    for item in menu_data:
+        print(item)
+
+    # Check if data exists
+    if not menu_data:
+        return "<p style='text-align: center;'>No menu data available. Please try again later.</p>"
+
     filtered_data = {}
+
     for item in menu_data:
+        # Ensure the required fields exist
+        if "Section__c" not in item or "Veg_NonVeg__c" not in item:
+            continue
+
+        # Add the section to filtered data if not already present
         if item["Section__c"] not in filtered_data:
             filtered_data[item["Section__c"]] = []
 
+        # Apply filters based on preferences
         if "All" in preferences:
             filtered_data[item["Section__c"]].append(item)
         elif "Veg" in preferences and item["Veg_NonVeg__c"] in ["Veg", "Both"]:
@@ -31,6 +125,7 @@ def filter_menu(preferences):
         elif "Non-Veg" in preferences and item["Veg_NonVeg__c"] in ["Non-Veg", "Both"]:
             filtered_data[item["Section__c"]].append(item)
 
+    # Generate HTML content
     html_content = '<div style="padding: 20px; max-width: 1200px; margin: auto;">'
     for section, items in filtered_data.items():
         if items:
@@ -39,7 +134,7 @@ def filter_menu(preferences):
             for item in items:
                 html_content += f"""
                 <div style="width: 300px; border: 1px solid #ddd; border-radius: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
-                    <img src="{item['Image1__c']}" style="width: 100%; height: 200px; object-fit: cover;">
+                    <img src="{item.get('Image1__c', '')}" style="width: 100%; height: 200px; object-fit: cover;">
                     <div style="padding: 10px;">
                         <h3>{item['Name']}</h3>
                         <p>${item['Price__c']}</p>
@@ -67,8 +162,20 @@ with gr.Blocks() as app:
             login_email = gr.Textbox(label="Email")
             login_password = gr.Textbox(label="Password", type="password")
             login_button = gr.Button("Login")
+            signup_button = gr.Button("Go to Signup")
             login_output = gr.Textbox(label="Status")
 
+    # Signup Page
+    with gr.Row(visible=False) as signup_page:
+        with gr.Column():
+            signup_name = gr.Textbox(label="Name")
+            signup_email = gr.Textbox(label="Email")
+            signup_phone = gr.Textbox(label="Phone")
+            signup_password = gr.Textbox(label="Password", type="password")
+            submit_signup = gr.Button("Signup")
+            login_redirect = gr.Button("Go to Login")
+            signup_output = gr.Textbox(label="Status")
+
     # Menu Page
     with gr.Row(visible=False) as menu_page:
         preferences = gr.CheckboxGroup(
@@ -82,10 +189,15 @@ with gr.Blocks() as app:
 
     # Login Button Logic
     def handle_login(email, password):
-        if email == "[email protected]" and password == "password":
-            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), "Login successful!"
+        status, user = login(email, password)
+        if status == "Login successful!":
+            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), status
         else:
-            return gr.update(), gr.update(), gr.update(), "Invalid login credentials."
+            return gr.update(), gr.update(), gr.update(), status
+
+    # Signup Button Logic
+    def handle_signup(name, email, phone, password):
+        return signup(name, email, phone, password)
 
     # Menu Logic
     def handle_menu(preferences):
@@ -95,6 +207,8 @@ with gr.Blocks() as app:
     login_button.click(
         handle_login, [login_email, login_password], [login_page, menu_page, footer, login_output]
     )
+    signup_button.click(lambda: (gr.update(visible=False), gr.update(visible=True)), None, [login_page, signup_page])
+    submit_signup.click(handle_signup, [signup_name, signup_email, signup_phone, signup_password], signup_output)
     preferences.change(handle_menu, [preferences], menu_output)
 
 app.launch()