Update app.py

app.py

@@ -6,14 +6,15 @@ def process_multiline(variable):
         return "\n".join(variable)
     return variable
 
-def generate_email(customer_name, out_of_stock, dispatched, customer_address, tracking_number, template_choice):
+def generate_email(customer_name, out_of_stock, dispatched, customer_address, delivered_date, tracking_link, tracking_number, template_choice):
     # Process variables that may contain multiple rows.
     out_of_stock = process_multiline(out_of_stock) if out_of_stock else ""
     dispatched = process_multiline(dispatched) if dispatched else ""
     customer_address = process_multiline(customer_address) if customer_address else ""
-    tracking_number = process_multiline(tracking_number) if tracking_number else ""
+    tracking_link = process_multiline(tracking_link) if tracking_link else ""
+    # delivered_date and tracking_number are single row variables, so we use them as is.
     
-    # Define the three email templates with placeholders.
+    # Define the email templates with placeholders.
     template1 = """Dear [variable: customer name],
 Thank you for your recent order. We regret to inform you that, according to our vendor, the below product you have ordered are currently out of stock:
 
@@ -28,7 +29,7 @@ Winson Lai
 Trendy Sweet Shop"""
 
     template2 = """Dear [variable: customer name],
-Thanks for your order. We would like to update you your order status as following :
+Thanks for your order. We would like to update you your order status as following:
 
 Out of Stock:
 [variable: out of stock product]
@@ -46,15 +47,15 @@ Trendy Sweet Shop"""
 
     template3 = """Dear [variable: customer name],
 
-Thank you for your recent order. According to tracking information from both DHL and USPS, your package was successfully delivered on February 20. You can verify the delivery status using the following tracking links:
+Thank you for your recent order. According to tracking information from both DHL and USPS, your package was successfully delivered on [variable: delivered date]. You can verify the delivery status using the following tracking links:
 
-[variable: tracking number]
+[variable: tracking link]
 
 The package was sent to the following address:
 
 [variable: customer address]
 
-Since the tracking details indicate a successful delivery, we recommend checking the usual delivery locations at your residence. If you're unable to locate the package, please contact USPS directly with the tracking number 9261290339726603626251 for further assistance.
+Since the tracking details indicate a successful delivery, we recommend checking the usual delivery locations at your residence. If you're unable to locate the package, please contact USPS directly with the tracking number [variable: tracking number] for further assistance.
 
 If you have any additional questions or need further support, feel free to reach out to us.
 
@@ -78,7 +79,9 @@ Trendy Sweet Shop"""
     email_template = email_template.replace("[variable: out of stock product]", out_of_stock)
     email_template = email_template.replace("[variable: dispatched product]", dispatched)
     email_template = email_template.replace("[variable: customer address]", customer_address)
-    email_template = email_template.replace("[variable: tracking number]", tracking_number)
+    email_template = email_template.replace("[variable: delivered date]", delivered_date if delivered_date else "")
+    email_template = email_template.replace("[variable: tracking link]", tracking_link)
+    email_template = email_template.replace("[variable: tracking number]", tracking_number if tracking_number else "")
 
     return email_template
 
@@ -91,10 +94,14 @@ with gr.Blocks(title="Email Generator") as demo:
         out_of_stock_input = gr.Textbox(label="Out of Stock Product", placeholder="Enter out-of-stock product (one per row)", lines=3)
         dispatched_input = gr.Textbox(label="Dispatched Product", placeholder="Enter dispatched product (if any, one per row)", lines=3)
     
-    # New input fields for customer address and tracking number.
+    # New input fields for customer address, delivered date, tracking link, and tracking number.
     with gr.Row():
         customer_address_input = gr.Textbox(label="Customer Address", placeholder="Enter customer address (one per row)", lines=3)
-        tracking_number_input = gr.Textbox(label="Tracking Number", placeholder="Enter tracking number (one per row)", lines=3)
+    with gr.Row():
+        delivered_date_input = gr.Textbox(label="Delivered Date", placeholder="Enter delivered date (single row)")
+        tracking_number_input = gr.Textbox(label="Tracking Number", placeholder="Enter tracking number (single row)")
+    with gr.Row():
+        tracking_link_input = gr.Textbox(label="Tracking Link", placeholder="Enter tracking link (one per row)", lines=3)
     
     # Dropdown for selecting the email template.
     template_choice = gr.Dropdown(
@@ -108,8 +115,17 @@ with gr.Blocks(title="Email Generator") as demo:
     # When the button is clicked, generate_email() is called.
     generate_button.click(
         fn=generate_email, 
-        inputs=[customer_name_input, out_of_stock_input, dispatched_input, customer_address_input, tracking_number_input, template_choice], 
+        inputs=[
+            customer_name_input, 
+            out_of_stock_input, 
+            dispatched_input, 
+            customer_address_input, 
+            delivered_date_input, 
+            tracking_link_input, 
+            tracking_number_input, 
+            template_choice
+        ], 
         outputs=email_output
     )
     
-demo.launch() 
+demo.launch()