Update app.py

app.py

@@ -1,51 +1,44 @@
 import gradio as gr
-from PIL import Image, ImageDraw, ImageFont
 
+# Function to generate a customized invoice with a wish message
 def generate_invoice(customer_name, items, total_amount):
-    # Create a blank image with white background
-    width, height = 400, 600
-    img = Image.new('RGB', (width, height), color=(255, 255, 255))
-    
-    # Initialize drawing context
-    draw = ImageDraw.Draw(img)
-    font = ImageFont.load_default()
-    
-    # Add restaurant name and header
-    draw.text((10, 10), "Restaurant XYZ", font=font, fill=(0, 0, 0))
-    draw.text((10, 40), f"Customer: {customer_name}", font=font, fill=(0, 0, 0))
-    
-    # Add items and prices
-    y_offset = 70
-    for item in items:
-        draw.text((10, y_offset), item, font=font, fill=(0, 0, 0))
-        y_offset += 20
-    
-    # Add total amount
-    draw.text((10, y_offset), f"Total: ${total_amount}", font=font, fill=(0, 0, 0))
-    
-    # Add a wish message
-    y_offset += 30
-    draw.text((10, y_offset), "Thank you for dining with us!", font=font, fill=(0, 0, 0))
-    
-    # Save the image to a temporary file
-    img_path = "/tmp/invoice.png"
-    img.save(img_path)
+    # 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}
     
-    return img_path
+    Total Amount: ${total_amount}
+    ----------------------------------------
+    {wish_message}
+    ----------------------------------------
+    """
+
+    return invoice
 
 # Gradio interface
 iface = gr.Interface(
     fn=generate_invoice, 
     inputs=[
-        gr.Textbox(label="Customer Name"), 
-        gr.Textbox(label="Items (comma-separated)"), 
-        gr.Textbox(label="Total Amount")
+        gr.Textbox(label="Customer Name"),
+        gr.Textbox(label="Items (comma separated)"),
+        gr.Number(label="Total Amount")
     ], 
-    outputs=gr.Image(label="Generated Invoice"),
-    title="Custom Invoice Generator",
-    description="Enter the customer name, items, and total amount to generate a customized restaurant invoice with a wish message."
+    outputs="text",
+    live=True
 )
 
 # Launch the app
 iface.launch()
 
+