Update app.py

app.py

@@ -70,13 +70,12 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         ex3 = "From: accounts@supplier.com\nSubject: Invoice #2024-00123 Due Date Reminder\nDear Valued Client, This is a reminder invoice #2024-00123 for $1,500.00 is due..."
         ex4 = "From: it@company.com\nSubject: Server Down Alert\nURGENT: Database server unreachable since 3AM. Needs escalation."
         ex5 = "From: spammer@promo.com\nSubject: BIG SALE!!!\nDon't miss our special offers!"
-        gr.Button("Support Request").click(lambda: ex1, None, 'email_box')
-        gr.Button("Sales Inquiry").click(lambda: ex2, None, 'email_box')
-        gr.Button("Finance / Billing").click(lambda: ex3, None, 'email_box')
-        gr.Button("Urgent Incident").click(lambda: ex4, None, 'email_box')
-        gr.Button("Spam / Marketing").click(lambda: ex5, None, 'email_box')
-
-    gr.Markdown("#### 📋 Paste or Load a Full Email:")
+        support_button = gr.Button("Support Request")
+        sales_button = gr.Button("Sales Inquiry")
+        finance_button = gr.Button("Finance / Billing")
+        urgent_button = gr.Button("Urgent Incident")
+        spam_button = gr.Button("Spam / Marketing")
+    
     email_box = gr.Textbox(lines=8, label="Paste the entire raw email (including lines like 'From:', 'Subject:', etc.)", elem_id='email_box')
     classify_btn = gr.Button("Classify Email", elem_id='classify_btn', variant="primary")
 
@@ -98,6 +97,12 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         gr.Markdown("#### 🟣 Extracted Action Items")
         action_items = gr.Textbox(lines=4, label="Extracted Action Items", elem_id='actions')
 
+    support_button.click(lambda: ex1, None, email_box)
+    sales_button.click(lambda: ex2, None, email_box)
+    finance_button.click(lambda: ex3, None, email_box)
+    urgent_button.click(lambda: ex4, None, email_box)
+    spam_button.click(lambda: ex5, None, email_box)
+
     def classify_and_render(email_text):
         result = email_classifier_router(email_text)
         cat = result.get("Category", "")