Update app.py

app.py

@@ -81,4 +81,44 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
 
     with gr.Column():
         gr.Markdown("### ✅ Classification Results")
-        category = gr.Textbox(label="Category", elem_id='category
+        category = gr.Textbox(label="Category", elem_id='category')
+        priority = gr.Textbox(label="Priority", elem_id='priority')
+        recipient = gr.Textbox(label="Suggested Recipient", elem_id='recipient')
+
+    with gr.Column():
+        gr.Markdown("### ✨ Generated Response Draft")
+        draft_response = gr.Textbox(lines=8, label="Generated Draft", elem_id='draft_response')
+
+    with gr.Column():
+        gr.Markdown("#### Summary")
+        summary = gr.Textbox(label="Summary", elem_id='summary')
+
+    with gr.Column():
+        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", "")
+        pri = result.get("Priority", "")
+        rec = result.get("Suggested Recipient", "")
+        dra = result.get("Draft Response", "")
+        summ = result.get("Summary", "")
+        acts = result.get("Action Items", [])
+        acts = "\n".join(acts) if isinstance(acts, list) else acts
+        return cat, pri, rec, dra, summ, acts
+
+    classify_btn.click(
+        classify_and_render,
+        inputs=[email_box],
+        outputs=[category, priority, recipient, draft_response, summary, action_items]
+    )
+
+if __name__ == "__main__":
+    demo.launch()