Update app.py

app.py

@@ -1,22 +1,64 @@
 import gradio as gr
 import pandas as pd
+from bcrypt import hashpw, gensalt, checkpw
 
-# In-memory storage for user credentials
-users = {"admin": "password"}  # Example credentials (username: password)
+# Path to the Excel file for user storage
+USER_FILE = "users.xlsx"
 
-# Function to validate login
-def validate_login(username, password):
-    if username in users and users[username] == password:
-        return True, "Login successful! Redirecting to the menu page..."
-    return False, "Invalid username or password. Please try again."
+# Load users from Excel
+def load_users():
+    try:
+        df = pd.read_excel(USER_FILE)
+        return {row['Username']: row['Password'] for _, row in df.iterrows()}
+    except FileNotFoundError:
+        # Create the file if it doesn't exist
+        df = pd.DataFrame(columns=["Username", "Password"])
+        df.to_excel(USER_FILE, index=False)
+        return {}
 
-# Function to register a new user
+# Save users to Excel
+def save_users(users):
+    df = pd.DataFrame(list(users.items()), columns=["Username", "Password"])
+    df.to_excel(USER_FILE, index=False)
+
+# Signup user
 def signup_user(username, password):
+    users = load_users()
     if username in users:
         return "Username already exists. Please choose a different username."
-    users[username] = password
+    # Hash the password before saving
+    hashed_password = hashpw(password.encode(), gensalt()).decode()
+    users[username] = hashed_password
+    save_users(users)
     return "Signup successful! Please log in."
 
+# Validate login
+def validate_login(username, password):
+    users = load_users()
+    if username in users and checkpw(password.encode(), users[username].encode()):
+        return True, "Login successful! Redirecting to the menu page..."
+    return False, "Invalid username or password. Please try again."
+
+# Session management
+active_sessions = {}
+
+def login_user(username, password):
+    success, message = validate_login(username, password)
+    if success:
+        active_sessions[username] = True  # Add to active sessions
+        return message
+    return message
+
+def is_logged_in(username):
+    return active_sessions.get(username, False)
+
+def logout_user(username):
+    if username in active_sessions:
+        del active_sessions[username]
+        return f"{username} has been logged out successfully."
+    return "No active session found for this user."
+
+
 # Function to load menu data
 def load_menu():
     menu_file = "menu.xlsx"  # Ensure this file exists
@@ -238,10 +280,10 @@ modal_and_cart_js = """
 # Gradio app definition
 def app():
     with gr.Blocks() as demo:
-        gr.Markdown("## Dynamic Menu with Preferences")
+        gr.Markdown("## Secure Food Ordering System with Login")
 
-        # Login and Signup Tabs
         with gr.Tabs():
+            # Login Tab
             with gr.Tab("Login"):
                 gr.Markdown("### Login Page")
                 username = gr.Textbox(label="Username")
@@ -250,11 +292,12 @@ def app():
                 login_message = gr.Label()
 
                 login_btn.click(
-                    validate_login,
+                    login_user,
                     inputs=[username, password],
                     outputs=[login_message],
                 )
 
+            # Signup Tab
             with gr.Tab("Signup"):
                 gr.Markdown("### Signup Page")
                 signup_username = gr.Textbox(label="Username")
@@ -268,12 +311,23 @@ def app():
                     outputs=[signup_message],
                 )
 
+            # Menu Tab
             with gr.Tab("Menu Page"):
-                gr.Markdown("### Menu Page")
-                selected_preference = gr.Radio(
-                    choices=["All", "Vegetarian", "Halal/Non-Veg", "Guilt-Free"],
-                    value="All",
-                    label="Choose a Preference",
+                gr.Markdown("### Menu Page (Only for Logged-In Users)")
+                menu_username = gr.Textbox(label="Enter Username to Verify Login", placeholder="Enter your username")
+                menu_check = gr.Button("Check Login Status")
+                login_status = gr.Label()
+                menu_content = gr.HTML(value="")
+
+                def check_login_status(username):
+                    if is_logged_in(username):
+                        return "Welcome to the menu page!", filter_menu("All")
+                    return "You are not logged in!", ""
+
+                menu_check.click(
+                    check_login_status,
+                    inputs=[menu_username],
+                    outputs=[login_status, menu_content],
                 )
 
                 # Output area for menu items