Update app.py

app.py

@@ -1,90 +1,106 @@
 import gradio as gr
-
-# Extended menu data with dish details
-menu_data = [
-    {"name": "Veg Burger", "category": "VEGAN",
-     "image": "https://upload.wikimedia.org/wikipedia/commons/6/6e/Veggie_burger.jpg",
-     "description": "A delicious vegan burger with plant-based patty, lettuce, and tomato."},
-    {"name": "Chicken Biryani", "category": "HALAL",
-     "image": "https://upload.wikimedia.org/wikipedia/commons/7/79/Chicken_Biryani.jpg",
-     "description": "Spicy chicken biryani with aromatic basmati rice and tender chicken pieces."},
-    {"name": "Paneer Butter Masala", "category": "VEGAN",
-     "image": "https://upload.wikimedia.org/wikipedia/commons/3/3d/Paneer_Butter_Masala.jpg",
-     "description": "Paneer cooked in a rich and creamy tomato-based gravy."},
-    {"name": "Beef Steak", "category": "HALAL",
-     "image": "https://upload.wikimedia.org/wikipedia/commons/d/d5/Beef_Steak.jpg",
-     "description": "Juicy beef steak served with mashed potatoes and grilled vegetables."},
-    {"name": "Mushroom Soup", "category": "VEGAN",
-     "image": "https://upload.wikimedia.org/wikipedia/commons/3/3c/Mushroom_Soup.jpg",
-     "description": "Creamy mushroom soup with fresh herbs."}
-]
-
-spice_options = ["🌶 Mild", "🌶🌶 Medium", "🌶🌶🌶 High"]
-portion_options = ["🧆 Small", "🍽 Medium", "🍛 Large"]
-
-# Function to filter dishes by category
-def get_dish_list(category):
-    if category == "ALL":
-        return [dish["name"] for dish in menu_data]
-    return [dish["name"] for dish in menu_data if dish["category"] == category]
-
-# Function to fetch dish details
-def get_dish_details(dish_name):
-    for dish in menu_data:
-        if dish["name"] == dish_name:
-            return dish["image"], dish["name"], dish["description"]
-    return "https://via.placeholder.com/300x200", "Unknown Dish", "No description available."
-
-# Function to confirm order
-def place_order(dish_name, spice_level, portion_size):
-    if dish_name:
-        return f"✅ Order Confirmed: '{dish_name}' with {spice_level} spice level and {portion_size} portion size."
-    return "⚠️ Please select a dish before placing an order."
-
-# Main Gradio App
-def gradio_app():
+import qrcode
+from PIL import Image
+
+# Sample Data
+CUSTOMERS = {"9876543210": {"name": "John Doe", "preferences": ["Halal", "Indian"]}}
+MENU = {
+    "ALL": [
+        {"name": "Chicken Curry", "description": "Spicy chicken curry", "image": "chicken_curry.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]},
+        {"name": "Paneer Butter Masala", "description": "Creamy paneer dish", "image": "paneer_butter.jpg", "spice_levels": ["Mild", "Medium", "Hot"], "portion_sizes": ["Small", "Medium", "Large"]}
+    ],
+    "VEGAN": [
+        {"name": "Veg Biryani", "description": "Flavourful veg biryani", "image": "veg_biryani.jpg", "spice_levels": ["Mild", "Medium"], "portion_sizes": ["Small", "Medium", "Large"]}
+    ]
+}
+
+# Generate QR Code
+
+def generate_qr(table_id):
+    qr = qrcode.QRCode(box_size=10, border=5)
+    qr.add_data(f"http://restaurant.com/table/{table_id}")
+    qr.make(fit=True)
+    img = qr.make_image(fill="black", back_color="white")
+    return img
+
+# Login / Signup
+
+def login_or_signup(phone_number):
+    if phone_number in CUSTOMERS:
+        return f"Welcome back, {CUSTOMERS[phone_number]['name']}!", True
+    else:
+        return "Please sign up to continue.", False
+
+# Save Preferences
+
+def save_preferences(phone_number, preferences):
+    if phone_number not in CUSTOMERS:
+        return "Customer not found. Sign up first."
+    CUSTOMERS[phone_number]["preferences"] = preferences
+    return f"Preferences saved: {preferences}"
+
+# Filter Menu
+
+def filter_menu(preferences):
+    filtered_menu = MENU.get(preferences[0].upper(), MENU["ALL"])
+    return filtered_menu
+
+# Dish Details
+
+def dish_details(dish_name):
+    for category in MENU.values():
+        for dish in category:
+            if dish["name"] == dish_name:
+                return {
+                    "Name": dish["name"],
+                    "Description": dish["description"],
+                    "Spice Levels": dish["spice_levels"],
+                    "Portion Sizes": dish["portion_sizes"]
+                }, dish["image"]
+
+# Gradio Interfaces
+
+def main():
     with gr.Blocks() as demo:
-        gr.HTML("<h1 style='text-align: center;'>🍛 Biryani Hub Menu 🍛</h1>")
-
-        # Category Selection
-        gr.Markdown("### Select a Category")
-        category_dropdown = gr.Radio(choices=["ALL", "VEGAN", "HALAL"], label="Categories", value="ALL")
+        gr.Markdown("# Restaurant QR Code Ordering System")
 
-        # Dish Selection
-        gr.Markdown("### Select a Dish")
-        dish_selector = gr.Radio(label="Available Dishes", choices=get_dish_list("ALL"))
-
-        # Dish Details Section
-        gr.Markdown("### Dish Details")
+        # QR Code Generator
         with gr.Row():
-            dish_image = gr.Image(label="Dish Image", height=200, width=200)
-            with gr.Column():
-                dish_name = gr.Textbox(label="Dish Name", interactive=False)
-                dish_description = gr.Textbox(label="Description", interactive=False)
-                spice_dropdown = gr.Radio(label="Select Spice Level", choices=spice_options, value="🌶 Medium")
-                portion_dropdown = gr.Radio(label="Select Portion Size", choices=portion_options, value="🍽 Medium")
-                place_order_btn = gr.Button("Place Order")
-                order_status = gr.Textbox(label="Order Status", interactive=False)
-
-        # Update dish list dynamically
-        def update_dish_list(category):
-            return gr.update(choices=get_dish_list(category))
+            table_id = gr.Textbox(label="Table ID", placeholder="Enter table ID")
+            qr_button = gr.Button("Generate QR Code")
+            qr_output = gr.Image(label="QR Code")
+            qr_button.click(lambda table_id: generate_qr(table_id), inputs=table_id, outputs=qr_output)
 
-        category_dropdown.change(fn=update_dish_list, inputs=category_dropdown, outputs=dish_selector)
+        # Login or Signup
+        with gr.Row():
+            phone_input = gr.Textbox(label="Phone Number", placeholder="Enter phone number")
+            login_button = gr.Button("Login / Signup")
+            login_output = gr.Text(label="Login Status")
+            login_button.click(login_or_signup, inputs=phone_input, outputs=login_output)
 
-        # Update dish details dynamically
-        def update_dish_details(dish_name_input):
-            return get_dish_details(dish_name_input)
+        # Preferences
+        with gr.Row():
+            preferences = gr.CheckboxGroup(["Vegan", "Halal", "Indian", "Chinese"], label="Preferences")
+            pref_button = gr.Button("Save Preferences")
+            pref_output = gr.Text(label="Preference Status")
+            pref_button.click(save_preferences, inputs=[phone_input, preferences], outputs=pref_output)
 
-        dish_selector.change(fn=update_dish_details, inputs=dish_selector,
-                             outputs=[dish_image, dish_name, dish_description])
+        # Menu Display
+        with gr.Row():
+            menu_pref = gr.Dropdown(["ALL", "VEGAN", "HALAL"], label="Menu Type")
+            menu_button = gr.Button("Show Menu")
+            menu_output = gr.JSON(label="Menu")
+            menu_button.click(filter_menu, inputs=menu_pref, outputs=menu_output)
 
-        # Place order functionality
-        place_order_btn.click(fn=place_order,
-                              inputs=[dish_name, spice_dropdown, portion_dropdown],
-                              outputs=order_status)
+        # Dish Details
+        with gr.Row():
+            dish_name = gr.Textbox(label="Dish Name", placeholder="Enter dish name")
+            dish_button = gr.Button("Get Dish Details")
+            dish_output = gr.JSON(label="Dish Details")
+            dish_image = gr.Image(label="Dish Image")
+            dish_button.click(dish_details, inputs=dish_name, outputs=[dish_output, dish_image])
 
-    return demo
+    demo.launch()
 
-# Launch the app
-gradio_app().launch()
+if __name__ == "__main__":
+    main()