Update app.py

app.py

@@ -15,6 +15,7 @@ def generate_email(
     template_choice,
     available_date,
     solution_choice,
+    status_choice,
 ):
     # Process variables that may contain multiple rows.
     out_of_stock = process_multiline(out_of_stock) if out_of_stock else ""
@@ -130,14 +131,42 @@ Sincerely,
 Winson Lai
 Trendy Sweet Shop"""
 
-    # --- Order Status template ---
-
-    template_status = """Dear [variable: customer name],
+    # --- Order Status templates with "status" dropdown ---
 
+    # Status – Option 1: Sent with tracking provided
+    template_status_1 = """Dear [variable: customer name], 
+    
 Thanks for your order. The tracking number of your package is [variable: tracking number] and below is the tracking link:
 
 [variable: tracking link]
 
+Your package is on its way, and you can monitor the delivery status from the tracking link above.
+
+Should you have any further question, please feel free to contact us again. Thanks.
+
+Regards,
+
+Winson Lai
+Trendy Sweet Shop"""
+
+    # Status – Option 2: Sent without tracking provided
+    template_status_2 = """Dear [variable: customer name], 
+    
+Thank you for your order. We confirm that your package has been dispatched, and the tracking link has been sent to you via our system. If you cannot locate the email, please check your junk or spam folder.
+You can monitor its delivery status using the tracking link provided.
+
+Should you have any further question, please feel free to contact us again. Thanks.
+
+Regards,
+
+Winson Lai
+Trendy Sweet Shop"""
+
+    # Status – Option 3: To be sent
+    template_status_3 = """Dear [variable: customer name], 
+    
+Thanks for your order. Your order is still under processing and we will release the tracking code once available.
+
 Should you have any further question, please feel free to contact us again. Thanks.
 
 Regards,
@@ -157,6 +186,7 @@ Trendy Sweet Shop"""
             email_template = template_std_3
         else:
             email_template = template_std_1  # fallback
+
     elif template_choice == "Partial Out of Stock":
         if solution_choice == "Refund or exchange":
             email_template = template_partial_1
@@ -166,8 +196,16 @@ Trendy Sweet Shop"""
             email_template = template_partial_3
         else:
             email_template = template_partial_1  # fallback
+
     elif template_choice == "Order Status":
-        email_template = template_status
+        if status_choice == "Sent with tracking provided":
+            email_template = template_status_1
+        elif status_choice == "Sent without tracking provided":
+            email_template = template_status_2
+        elif status_choice == "To be sent":
+            email_template = template_status_3
+        else:
+            email_template = template_status_1  # fallback
 
     # ===== REPLACE PLACEHOLDERS =====
     email_template = email_template.replace(
@@ -203,7 +241,7 @@ def update_visibility(template_choice):
       Customer Name, Out of Stock Product, Dispatched Product, Available Date, Solutions dropdown
 
     - Order Status:
-      Customer Name, Tracking Number, Tracking Link
+      Customer Name, Tracking Number, Tracking Link, Status dropdown
     """
     if template_choice == "Standard out of stock":
         return (
@@ -214,6 +252,7 @@ def update_visibility(template_choice):
             gr.update(visible=False),  # tracking_link_input
             gr.update(visible=True),   # available_date_input
             gr.update(visible=True),   # solutions_dropdown
+            gr.update(visible=False),  # status_dropdown
         )
     elif template_choice == "Partial Out of Stock":
         return (
@@ -224,6 +263,7 @@ def update_visibility(template_choice):
             gr.update(visible=False),  # tracking_link_input
             gr.update(visible=True),   # available_date_input
             gr.update(visible=True),   # solutions_dropdown
+            gr.update(visible=False),  # status_dropdown
         )
     elif template_choice == "Order Status":
         return (
@@ -234,6 +274,7 @@ def update_visibility(template_choice):
             gr.update(visible=True),   # tracking_link_input
             gr.update(visible=False),  # available_date_input
             gr.update(visible=False),  # solutions_dropdown
+            gr.update(visible=True),   # status_dropdown
         )
     else:
         # Blank / default: hide everything
@@ -245,6 +286,7 @@ def update_visibility(template_choice):
             gr.update(visible=False),
             gr.update(visible=False),
             gr.update(visible=False),
+            gr.update(visible=False),
         )
 
 
@@ -313,6 +355,18 @@ with gr.Blocks(title="Email Generator") as demo:
         visible=False,
     )
 
+    # Status dropdown (Order Status)
+    status_dropdown = gr.Dropdown(
+        choices=[
+            "Sent with tracking provided",
+            "Sent without tracking provided",
+            "To be sent",
+        ],
+        value="Sent with tracking provided",
+        label="Status",
+        visible=False,
+    )
+
     generate_button = gr.Button("Generate Email")
     email_output = gr.Textbox(label="Generated Email", lines=15)
 
@@ -328,6 +382,7 @@ with gr.Blocks(title="Email Generator") as demo:
             tracking_link_input,
             available_date_input,
             solutions_dropdown,
+            status_dropdown,
         ]
     )
 
@@ -343,6 +398,7 @@ with gr.Blocks(title="Email Generator") as demo:
             template_choice,
             available_date_input,
             solutions_dropdown,
+            status_dropdown,
         ],
         outputs=email_output
     )