Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 
-def generate_email(customer_name, out_of_stock, dispatched, template_choice):
-    # Define the two email templates with placeholders.
+def generate_email(customer_name, out_of_stock, dispatched, customer_address, tracking_number, template_choice):
+    # Define the three 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:
 
@@ -24,11 +24,30 @@ Out of Stock:
 Dispatched:
 [variable: dispatched product]
 
-Since we do not have an estimated time of when the outstanding item will be back in stock at this moment, a a solution, we would like to offer you either a refund or an exchange for an alternative model. We understand that this situation may have caused inconvenience and we sincerely apologize for that.
+Since we do not have an estimated time of when the outstanding item will be back in stock at this moment, as a solution, we would like to offer you either a refund or an exchange for an alternative model. We understand that this situation may have caused inconvenience and we sincerely apologize for that.
 Please let us know your preference by replying to this email and we will take care of the rest. If you have any further questions or concerns, please do not hesitate to reach out to us.
 
 Sincerely,
 
+Winson Lai
+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:
+
+[variable: tracking number]
+
+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.
+
+If you have any additional questions or need further support, feel free to reach out to us.
+
+Best regards,
+
 Winson Lai
 Trendy Sweet Shop"""
 
@@ -37,14 +56,17 @@ Trendy Sweet Shop"""
         email_template = template1
     elif template_choice == "Partial Out of Stock":
         email_template = template2
+    elif template_choice == "US not received":
+        email_template = template3
     else:
         email_template = ""
 
     # Replace the placeholders with the input values.
-    # If no value is provided, the placeholder will be replaced with an empty string.
     email_template = email_template.replace("[variable: customer name]", customer_name if customer_name else "")
     email_template = email_template.replace("[variable: out of stock product]", out_of_stock if out_of_stock else "")
     email_template = email_template.replace("[variable: dispatched product]", dispatched if dispatched else "")
+    email_template = email_template.replace("[variable: customer address]", customer_address if customer_address else "")
+    email_template = email_template.replace("[variable: tracking number]", tracking_number if tracking_number else "")
 
     return email_template
 
@@ -57,9 +79,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")
         dispatched_input = gr.Textbox(label="Dispatched Product", placeholder="Enter dispatched product (if any)")
     
+    # New input fields for customer address and tracking number.
+    with gr.Row():
+        customer_address_input = gr.Textbox(label="Customer Address", placeholder="Enter customer address", lines=2)
+        tracking_number_input = gr.Textbox(label="Tracking Number", placeholder="Enter tracking number", lines=1)
+    
     # Dropdown for selecting the email template.
     template_choice = gr.Dropdown(
-        choices=["Standard out of stock", "Partial Out of Stock"],
+        choices=["Standard out of stock", "Partial Out of Stock", "US not received"],
         label="Select Email Template"
     )
     
@@ -69,8 +96,8 @@ 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, template_choice], 
+        inputs=[customer_name_input, out_of_stock_input, dispatched_input, customer_address_input, tracking_number_input, template_choice], 
         outputs=email_output
     )
     
-demo.launch()
+demo.launch()