Update app.py

app.py

@@ -2,33 +2,6 @@
 import gradio as gr
 from datetime import datetime
 
-def generate_invoice(customer_name, table_number, items, wishes):
-    total = sum(item['price'] for item in items)
-    invoice = f"""
-    -----------------------------------------
-                Restaurant Invoice
-    -----------------------------------------
-    Date: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
-    Customer Name: {customer_name}
-    Table Number: {table_number}
-    -----------------------------------------
-    Items Ordered:
-    """
-    for item in items:
-        invoice += f"- {item['name']} (${item['price']})\n"
-
-    invoice += f"""
-    -----------------------------------------
-    Total Amount: ${total:.2f}
-    -----------------------------------------
-    Customer Wishes:
-    {wishes}
-    -----------------------------------------
-    Thank you for dining with us!
-    Have a great day!
-    """
-    return invoice
-
 # Dynamic menu data
 menu = {
     "Bread": [
@@ -57,13 +30,41 @@ def render_menu():
             rendered_menu += f"- {item['name']} (${item['price']})\n"
     return rendered_menu
 
-def place_order(customer_name, table_number, selected_items, wishes):
-    selected_items = eval(selected_items)  # Convert string to list
+def generate_invoice(customer_name, table_number, selected_items, wishes):
+    selected_items = selected_items.split(', ')
     items = [item for category in menu.values() for item in category if item['name'] in selected_items]
-    return generate_invoice(customer_name, table_number, items, wishes)
+    total = sum(item['price'] for item in items)
+    invoice = f"""
+    -----------------------------------------
+                Restaurant Invoice
+    -----------------------------------------
+    Date: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+    Customer Name: {customer_name}
+    Table Number: {table_number}
+    -----------------------------------------
+    Items Ordered:
+    """
+    for item in items:
+        invoice += f"- {item['name']} (${item['price']})\n"
 
-def show_menu():
-    return render_menu()
+    invoice += f"""
+    -----------------------------------------
+    Total Amount: ${total:.2f}
+    -----------------------------------------
+    Customer Wishes:
+    """
+    for wish in wishes.split(', '):
+        invoice += f"- {wish}\n"
+
+    invoice += """
+    -----------------------------------------
+    Thank you for dining with us!
+    Have a great day!
+    """
+    return invoice
+
+def place_order(customer_name, table_number, selected_items, wishes):
+    return generate_invoice(customer_name, table_number, selected_items, wishes)
 
 with gr.Blocks() as demo:
     gr.Markdown("""# Restaurant Invoice Generator
@@ -78,7 +79,7 @@ with gr.Blocks() as demo:
     menu_display = gr.Textbox(show_label=False, interactive=False, value=render_menu())
 
     selected_items = gr.Textbox(label="Selected Items (comma-separated)", placeholder="e.g., Roti, Paneer Butter Masala")
-    wishes = gr.Textbox(label="Special Wishes", placeholder="e.g., Extra spicy, Less salt, etc.")
+    wishes = gr.Textbox(label="Special Wishes (comma-separated)", placeholder="e.g., Extra spicy, Less salt")
 
     generate_invoice_btn = gr.Button("Generate Invoice")
 
@@ -91,3 +92,4 @@ with gr.Blocks() as demo:
     )
 
 demo.launch()
+