Update app.py

app.py

@@ -1,71 +1,65 @@
 import gradio as gr
-from fpdf import FPDF
-from PIL import Image, ImageDraw, ImageFont
+import json
 
-# Function to generate a themed invoice
-def generate_invoice(customer_name, items, logo_path):
-    # Create a PDF document
-    pdf = FPDF()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.add_page()
-
-    # Title and Basic Styling
-    pdf.set_font("Arial", 'B', 16)
-    pdf.cell(200, 10, txt="Restaurant Invoice", ln=True, align='C')
-
-    # Add logo
-    if logo_path:
-        pdf.image(logo_path, x=10, y=20, w=30)  # Adjust the size and position of the logo
-
-    # Customer Details
-    pdf.ln(30)  # Add space after title
-    pdf.set_font("Arial", size=12)
-    pdf.cell(100, 10, f"Customer: {customer_name}", ln=True)
-
-    # Itemized list
-    pdf.cell(100, 10, "Item", ln=True, border=1, align="C")
-    pdf.cell(30, 10, "Price", ln=True, border=1, align="C")
-    total_price = 0
-
-    for item, price in items.items():
-        pdf.cell(100, 10, item, ln=True, border=1)
-        pdf.cell(30, 10, f"${price:.2f}", ln=True, border=1, align="R")
-        total_price += price
-
-    # Add Total
-    pdf.ln(10)
-    pdf.cell(100, 10, "Total", ln=True, align="C", border=1)
-    pdf.cell(30, 10, f"${total_price:.2f}", ln=True, border=1, align="R")
-
-    # Wish Message
-    pdf.ln(20)
-    pdf.set_font("Arial", 'I', 12)
-    pdf.multi_cell(0, 10, "Thank you for choosing us! We hope to serve you again soon.")
-    
-    # Save PDF to a file
-    pdf_output = "/tmp/invoice.pdf"
-    pdf.output(pdf_output)
-    return pdf_output
+# Sample JSON data of dishes
+dishes = [
+    {
+        "dish_id": 1,
+        "name": "Spaghetti Bolognese",
+        "description": "Classic Italian pasta with a rich, meaty tomato sauce.",
+        "price": 12.99
+    },
+    {
+        "dish_id": 2,
+        "name": "Margherita Pizza",
+        "description": "Simple and delicious pizza with fresh mozzarella, basil, and tomato sauce.",
+        "price": 9.99
+    },
+    {
+        "dish_id": 3,
+        "name": "Caesar Salad",
+        "description": "Crisp romaine lettuce, parmesan, croutons, and Caesar dressing.",
+        "price": 8.99
+    },
+    {
+        "dish_id": 4,
+        "name": "Chicken Alfredo",
+        "description": "Grilled chicken breast served with creamy Alfredo sauce over fettuccine.",
+        "price": 14.99
+    },
+    {
+        "dish_id": 5,
+        "name": "Tiramisu",
+        "description": "Classic Italian dessert with layers of coffee-soaked ladyfingers and mascarpone cream.",
+        "price": 5.99
+    }
+]
 
+# Function to generate an invoice based on selected dishes
+def generate_invoice(selected_dishes):
+    total_price = sum(dish["price"] for dish in selected_dishes)
+    invoice_text = "----- Invoice -----\n"
+    for dish in selected_dishes:
+        invoice_text += f"{dish['name']}: ${dish['price']}\n"
+    invoice_text += f"Total: ${total_price:.2f}\n"
+    invoice_text += "\nThank you for dining with us! Have a great day!"
+    return invoice_text
 
 # Gradio Interface
-def invoice_interface(customer_name, items, logo_path):
-    return generate_invoice(customer_name, items, logo_path)
-
+def invoice_interface(selected_dishes_ids):
+    selected_dishes = [dishes[i] for i in selected_dishes_ids]
+    return generate_invoice(selected_dishes)
 
-# Example Gradio Interface
-interface = gr.Interface(
-    fn=invoice_interface,
-    inputs=[
-        gr.Textbox(label="Customer Name"),
-        gr.JSON(label="Items (JSON format like {'item1': 10.5, 'item2': 7.25})"),
-        gr.Image(label="Upload Logo", type="filepath")
-    ],
-    outputs=gr.File(label="Download Invoice"),
-    title="Customized Restaurant Invoice Generator",
-    description="Generate an invoice with your restaurant's custom theme, logo, and message!"
-)
-
-# Launch the app
-interface.launch()
+# Create Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("### Select Dishes for Your Invoice")
+    
+    dish_choices = [gr.Checkbox(label=dish['name'], value=False) for dish in dishes]
+    
+    btn = gr.Button("Generate Invoice")
+    
+    output = gr.Textbox(label="Invoice")
+    
+    btn.click(invoice_interface, inputs=dish_choices, outputs=output)
 
+demo.launch()