Update app.py

app.py

@@ -1,44 +1,71 @@
 import gradio as gr
+from fpdf import FPDF
+from PIL import Image, ImageDraw, ImageFont
 
-# Function to generate a customized invoice with a wish message
-def generate_invoice(customer_name, items, total_amount):
-    # Customized message
-    wish_message = f"Thank you for visiting, {customer_name}! We appreciate your business."
-
-    # Sample invoice structure
-    invoice = f"""
-    ----------------------------------------
-    INVOICE
-    Restaurant Name: Example Restaurant
-    Address: 123 Main Street, City, State
-    Phone: (555) 123-4567
-    ----------------------------------------
-    Customer: {customer_name}
-    ----------------------------------------
-    Items:
-    {items}
+# 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.")
     
-    Total Amount: ${total_amount}
-    ----------------------------------------
-    {wish_message}
-    ----------------------------------------
-    """
+    # Save PDF to a file
+    pdf_output = "/tmp/invoice.pdf"
+    pdf.output(pdf_output)
+    return pdf_output
+
 
-    return invoice
+# Gradio Interface
+def invoice_interface(customer_name, items, logo_path):
+    return generate_invoice(customer_name, items, logo_path)
 
-# Gradio interface
-iface = gr.Interface(
-    fn=generate_invoice, 
+
+# Example Gradio Interface
+interface = gr.Interface(
+    fn=invoice_interface,
     inputs=[
         gr.Textbox(label="Customer Name"),
-        gr.Textbox(label="Items (comma separated)"),
-        gr.Number(label="Total Amount")
-    ], 
-    outputs="text",
-    live=True
+        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
-iface.launch()
-
+interface.launch()