From 22e5452003ce0fe45e4f67ae6ddd295f1b098224 Mon Sep 17 00:00:00 2001
From: Arjun <6592213+arkml@users.noreply.github.com>
Date: Wed, 13 May 2026 23:27:38 +0530
Subject: [PATCH] refine draft

---
 apps/x/apps/renderer/src/App.css              | 34 +++++++++-
 .../renderer/src/components/email-view.tsx    | 64 +++++++++++++++----
 2 files changed, 86 insertions(+), 12 deletions(-)

diff --git a/apps/x/apps/renderer/src/App.css b/apps/x/apps/renderer/src/App.css
index b5df52bf..a7b35992 100644
--- a/apps/x/apps/renderer/src/App.css
+++ b/apps/x/apps/renderer/src/App.css
@@ -793,13 +793,45 @@
   border-top: 1px solid var(--gm-border);
 }
 
+.gmail-compose-actions-primary {
+  display: flex;
+  align-items: center;
+  gap: 8px;
+  flex-shrink: 0;
+}
+
+.gmail-refine-button {
+  display: inline-flex;
+  align-items: center;
+  gap: 8px;
+  box-sizing: border-box;
+  height: 30px;
+  padding: 0 14px;
+  border: 1px solid var(--gm-border-strong);
+  border-radius: 6px;
+  background: var(--gm-bg-pill);
+  color: var(--gm-text);
+  font: inherit;
+  font-size: 12px;
+  font-weight: 500;
+  cursor: pointer;
+  transition: background 120ms ease, border-color 120ms ease, color 120ms ease;
+}
+
+.gmail-refine-button:hover {
+  background: var(--gm-bg-pill-hover);
+  border-color: var(--gm-accent);
+  color: var(--gm-accent);
+}
+
 .gmail-send-button {
   display: inline-flex;
   align-items: center;
   gap: 8px;
+  box-sizing: border-box;
   height: 30px;
   padding: 0 14px;
-  border: none;
+  border: 1px solid transparent;
   border-radius: 6px;
   background: var(--gm-accent);
   color: var(--gm-accent-fg);
diff --git a/apps/x/apps/renderer/src/components/email-view.tsx b/apps/x/apps/renderer/src/components/email-view.tsx
index 8894dedd..93f18fc6 100644
--- a/apps/x/apps/renderer/src/components/email-view.tsx
+++ b/apps/x/apps/renderer/src/components/email-view.tsx
@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
-import { Bold, Forward, Italic, Link as LinkIcon, List, ListOrdered, LoaderIcon, Quote, RefreshCw, Reply, Search, Send, Strikethrough } from 'lucide-react'
+import { Bold, Forward, Italic, Link as LinkIcon, List, ListOrdered, LoaderIcon, Quote, RefreshCw, Reply, Search, Send, Sparkles, Strikethrough } from 'lucide-react'
 import { useEditor, EditorContent, type Editor } from '@tiptap/react'
 import StarterKit from '@tiptap/starter-kit'
 import Link from '@tiptap/extension-link'
@@ -463,6 +463,37 @@ function ComposeBox({
     if (editor && sel) editor.chain().focus().setTextSelection(sel).run()
   }
 
+  const refineWithCopilot = () => {
+    if (!editor) return
+    const currentDraft = editor.getText().trim()
+    const subject = thread.subject || '(No subject)'
+
+    const lines: string[] = []
+    lines.push(`Help me refine this draft email response. **Please ask me how I want to refine it before making any changes** — wait for my answer, then apply the edits.`)
+    lines.push('')
+    lines.push(`**Mode:** ${mode === 'reply' ? 'Reply' : 'Forward'}`)
+    lines.push(`**Subject:** ${subject}`)
+    lines.push('')
+    lines.push(`## Thread (${thread.messages.length} message${thread.messages.length === 1 ? '' : 's'})`)
+    lines.push('')
+    thread.messages.forEach((message, index) => {
+      lines.push(`### Message ${index + 1}`)
+      if (message.from) lines.push(`**From:** ${message.from}`)
+      if (message.to) lines.push(`**To:** ${message.to}`)
+      if (message.date) lines.push(`**Date:** ${message.date}`)
+      lines.push('')
+      lines.push((message.body || '(empty)').trim())
+      lines.push('')
+    })
+
+    lines.push(`## Current draft`)
+    lines.push('')
+    lines.push(currentDraft || '(empty — no draft yet)')
+
+    window.__pendingEmailDraft = { prompt: lines.join('\n') }
+    window.dispatchEvent(new Event('email-block:draft-with-assistant'))
+  }
+
   return (
     <div className="gmail-compose-card">
       <div className="gmail-compose-header">
@@ -502,16 +533,27 @@ function ComposeBox({
         </div>
       )}
       <div className="gmail-compose-actions">
-        <button
-          type="button"
-          className="gmail-send-button"
-          onClick={() => {
-            toast('Sending from this view needs Gmail send scope. Draft UI is ready.', 'info')
-          }}
-        >
-          <Send size={15} />
-          Send
-        </button>
+        <div className="gmail-compose-actions-primary">
+          <button
+            type="button"
+            className="gmail-send-button"
+            onClick={() => {
+              toast('Sending from this view needs Gmail send scope. Draft UI is ready.', 'info')
+            }}
+          >
+            <Send size={15} />
+            Send
+          </button>
+          <button
+            type="button"
+            className="gmail-refine-button"
+            onClick={refineWithCopilot}
+            title="Refine this draft with Copilot"
+          >
+            <Sparkles size={15} />
+            Refine
+          </button>
+        </div>
         {editor && <ComposeToolbar editor={editor} onOpenLink={openLink} />}
         <button type="button" className="gmail-compose-link" onClick={onClose}>Discard</button>
       </div>