Update app.py

app.py

@@ -45,87 +45,48 @@ def login(email, password):
         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
+# Function to load menu data
 def load_menu_from_salesforce():
     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):
+# Function to filter menu items
+def filter_menu(preference):
     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"]:
-            filtered_data[item["Section__c"]].append(item)
-        elif "Non-Veg" in preferences and item["Veg_NonVeg__c"] in ["Non veg", "Both"]:
+        if preference == "All" or (preference == "Veg" and item["Veg_NonVeg__c"] in ["Veg", "Both"]) or (preference == "Non-Veg" 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:
@@ -152,11 +113,9 @@ def filter_menu(preferences):
 
 # Gradio App
 with gr.Blocks() as app:
-    # Header
     with gr.Row():
         gr.HTML("<h1 style='text-align: center;'>Welcome to Biryani Hub</h1>")
 
-    # Login Page
     with gr.Row(visible=True) as login_page:
         with gr.Column():
             login_email = gr.Textbox(label="Email")
@@ -165,7 +124,6 @@ with gr.Blocks() as app:
             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")
@@ -176,39 +134,33 @@ with gr.Blocks() as app:
             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(
-            choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value=["All"]
-        )
+        preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
         menu_output = gr.HTML()
 
-    # Footer
     with gr.Row(visible=False) as footer:
         gr.HTML("<footer style='text-align: center;'>Thank you! Welcome again!</footer>")
 
-    # Login Button Logic
     def handle_login(email, password):
         status, user = login(email, password)
         if status == "Login successful!":
-            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), status
+            content = filter_menu("All")  # Default to "All" on login
+            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(value=content), status
         else:
-            return gr.update(), gr.update(), gr.update(), status
+            return gr.update(), 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):
-        content = filter_menu(preferences)
+    def handle_menu(preference):
+        content = filter_menu(preference)
         return content
 
     login_button.click(
-        handle_login, [login_email, login_password], [login_page, menu_page, footer, login_output]
+        handle_login, [login_email, login_password], [login_page, menu_page, footer, menu_output, 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)
+    preference.change(handle_menu, [preference], menu_output)
 
 app.launch()