Update app.py

app.py

@@ -1,12 +1,128 @@
-from simple_salesforce import Salesforce
 
+
+import bcrypt
+import gradio as gr
+from simple_salesforce import Salesforce
 # Authenticate with Salesforce
 sf = Salesforce(username='[email protected]', 
                 password='Lavanyanaga@123', 
                 security_token='z7Wvk6mys7n8XjqbYKf3bwBh7')
-import bcrypt
 
+
+# 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'))
+
+# Function for Signup
+def signup(name, email, phone, password):
+    hashed_password = hash_password(password)  # Hash the plain text password
+    try:
+        sf.User_Login__c.create({
+            'Name__c': name,
+            'Email__c': email,
+            'Phone__c': phone,
+            'Password__c': hashed_password
+        })
+        return "Signup successful! You can now login."
+    except Exception as e:
+        return f"Error during signup: {str(e)}"
+
+# Function for Login
+def login(email, password):
+    try:
+        # Query Salesforce for the user's hashed password
+        query = f"SELECT Name__c, Email__c, Phone__c, Password__c FROM User_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]
+        hashed_password = user['Password__c']
+
+        # Compare the entered password with the stored hash
+        if verify_password(password, hashed_password):
+            return "Login successful!", user['Name__c'], user['Email__c'], user['Phone__c']
+        else:
+            return "Invalid email or password.", None, None, None
+    except Exception as e:
+        return f"Error during login: {str(e)}", None, None, None
+
+# Function for Home Page
+def home_page(name, email, phone):
+    return f"Welcome, {name}! Your email is {email} and your phone number is {phone}."
+
+# Gradio Signup Interface
+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 Interface
+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"
+)
+
+# Gradio Home Interface
+home_interface = gr.Interface(
+    fn=home_page,
+    inputs=[
+        gr.Textbox(label="Name"),
+        gr.Textbox(label="Email"),
+        gr.Textbox(label="Phone"),
+    ],
+    outputs=gr.Textbox(label="Welcome Message"),
+    title="Home Page"
+)
+
+# Combine All Pages in Gradio
+with gr.Blocks() as app:
+    with gr.Tab("Signup"):
+        signup_interface.render()
+
+    with gr.Tab("Login"):
+        login_interface.render()
+
+    with gr.Tab("Home"):
+        home_interface.render()
+
+# Launch the Gradio App
+app.launch()
+
+"""def hash_password(password):
     # This will generate a hash that may exceed 60 characters
     return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
 
@@ -24,17 +140,25 @@ def signup(name, email, phone, password):
         return "Signup successful! You can now login."
     except Exception as e:
         return f"Error during signup: {str(e)}"
+
+
 def login(email, password):
-    query = f"SELECT Password__c FROM User_Login__c WHERE Email__c = '{email}'"
+    # Query Salesforce for the user's hashed password
+    query = f"SELECT Name__c, Email__c, Phone__c, Password__c FROM User_Login__c WHERE Email__c = '{email}'"
     result = sf.query(query)
+    
     if len(result['records']) == 0:
-        return "Invalid email or password."
+        return "Invalid email or password.", None, None, None
     
-    hashed_password = result['records'][0]['Password__c']
-    if verify_password(password, hashed_password):
-        return "Login successful!"
+    user = result['records'][0]
+    hashed_password = user['Password__c']
+    
+    # Compare the entered password with the stored hash
+    if bcrypt.checkpw(password.encode('utf-8'), hashed_password.encode('utf-8')):
+        return "Login successful!", user['Name__c'], user['Email__c'], user['Phone__c']
     else:
-        return "Invalid email or password."
+        return "Invalid email or password.", None, None, None
+
 
 import gradio as gr
 def signup_page(name, email, phone, password):
@@ -93,4 +217,4 @@ with gr.Blocks() as app:
         home_interface.render()
 
 # Launch the app
-app.launch()
+app.launch()"""