Update app.py

app.py

@@ -1,49 +1,123 @@
+import bcrypt
+import gradio as gr
 from simple_salesforce import Salesforce
-
 # Salesforce Connection
 sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-# Function to fetch menu items from Salesforce
-def load_menu_from_salesforce():
+
+# 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 with email validation
+def signup(name, email, phone, password):
     try:
-        query = "SELECT Name, Price__c, Description__c, Image1__c, Ingredients__c FROM Menu_Item__c"
+        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)
-        return result['records']
+
+        if len(result['records']) > 0:
+            # Email already exists
+            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  # Use Password3__c field for hashed passwords
+        })
+        return "Signup successful! You can now login."
     except Exception as e:
-        raise ValueError(f"Error loading menu data from Salesforce: {e}")
-
-# Function to filter menu items based on preference
-def filter_menu_from_salesforce(preference):
-    menu_data = load_menu_from_salesforce()
-    filtered_data = []
-    
-    for item in menu_data:
-        if preference == "Halal/Non-Veg":
-            if any(x in item.get("Ingredients__c", "").lower() for x in ["chicken", "mutton", "fish", "prawns", "goat"]):
-                filtered_data.append(item)
-        elif preference == "Vegetarian":
-            if not any(x in item.get("Ingredients__c", "").lower() for x in ["chicken", "mutton", "fish", "prawns", "goat"]):
-                filtered_data.append(item)
-        elif preference == "Guilt-Free":
-            if "fat:" in item.get("Description__c", "").lower():
-                filtered_data.append(item)
+        return f"Error during signup: {str(e)}"
+
+# Login function
+def login(email, password):
+    try:
+        email = email.strip()
+        password = password.strip()
+
+        # Query Salesforce for user details
+        query = f"SELECT Name, Email__c, Phone_Number__c, Password__c FROM Customer_Login__c WHERE Email__c = '{email}'"
+        result = sf.query(query)
+
+        if len(result['records']) == 0:
+            return "Invalid email or password.", None, None, None
+
+        user = result['records'][0]
+        stored_password = user['Password__c']
+
+        print(f"Entered Email: {email}")
+        print(f"Entered Password: {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'], user['Email__c'], user['Phone_Number__c']
         else:
-            filtered_data = menu_data
-
-    # Generate HTML content
-    html_content = ""
-    for item in filtered_data:
-        html_content += f"""
-        <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 15px; margin-bottom: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
-            <div style="flex: 1; margin-right: 15px;">
-                <h3 style="margin: 0; font-size: 18px;">{item['Name']}</h3>
-                <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price__c']}</p>
-                <p style="margin: 5px 0; font-size: 14px; color: #555;">{item['Description__c']}</p>
-            </div>
-            <div style="flex-shrink: 0; text-align: center;">
-                <img src="{item['Image1__c']}" alt="{item['Name']}" style="width: 100px; height: 100px; border-radius: 8px; object-fit: cover; margin-bottom: 10px;">
-                <button style="background-color: #28a745; color: white; border: none; padding: 8px 15px; font-size: 14px; border-radius: 5px; cursor: pointer;" onclick="openModal('{item['Name']}', '{item['Image1__c']}', '{item['Description__c']}', '{item['Price__c']}')">Add</button>
-            </div>
-        </div>
-        """
-    return html_content
+            print("Password did not match!")
+            return "Invalid email or password.", None, None, None
+    except Exception as e:
+        print(f"Error during login: {str(e)}")
+        return f"Error during login: {str(e)}", None, None, None
+
+# Gradio Signup Page
+def signup_page(name, email, phone, password):
+    response = signup(name, email, phone, password)
+    return response
+
+signup_interface = gr.Interface(
+    fn=signup_page,
+    inputs=[
+        gr.Textbox(label="Name"),
+        gr.Textbox(label="Email"),
+        gr.Textbox(label="Phone"),
+        gr.Textbox(label="Password", type="password"),
+    ],
+    outputs=gr.Textbox(label="Signup Status"),
+    title="Signup Page"
+)
+
+# Gradio Login Page
+def login_page(email, password):
+    login_status, name, email, phone = login(email, password)
+    if login_status == "Login successful!":
+        return login_status, name, email, phone
+    else:
+        return login_status, "Error", "Error", "Error"
+
+login_interface = gr.Interface(
+    fn=login_page,
+    inputs=[
+        gr.Textbox(label="Email"),
+        gr.Textbox(label="Password", type="password"),
+    ],
+    outputs=[
+        gr.Textbox(label="Login Status"),
+        gr.Textbox(label="Name"),
+        gr.Textbox(label="Email"),
+        gr.Textbox(label="Phone"),
+    ],
+    title="Login Page"
+)
+
+# Combine Pages in Gradio
+with gr.Blocks() as app:
+    with gr.Tab("Signup"):
+        signup_interface.render()
+
+    with gr.Tab("Login"):
+        login_interface.render()
+
+# Launch Gradio App
+app.launch()