Merge pull request #2 from DemocracyLab/sig

Sig

Author: ddfridley

Sample email for Generative AI.md

@@ -1,4 +1,4 @@
-# Sample email for Generative AI 
+# Sample email for Generative AI
 Please write unique subject lines and emails using the sample below. Please follow the following rules:
 
 - Rewrite the paragraphs, keeping the order and number of paragraphs from the sample
@@ -16,7 +16,7 @@ Please write unique subject lines and emails using the sample below. Please foll
 - For the two issues that are listed, don't put a space at the beginning of each line
 - Each of the two issues must be on a separate line.
 - Don't label the subject line and email body, just bold the subject line
-- Always end with, "Best,"
+- Always end with, "Best," the signature will come from the gmail sig
 - Every sentence other than links should be unique, don't repeat across emails unless otherwise instructed above
 - Write in a professional and friendly tone
 - The subject line must be text only, not Markdown or HTML
@@ -43,5 +43,3 @@ Thank you for considering this opportunity. We look forward to connecting with y
 
 Best,
 
-
-David

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-mail",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "AI-powered email personalization tool with Gmail and Google Sheets integration",
   "main": "dist-electron/index.js",
   "type": "module",

package.json

@@ -103,7 +103,8 @@ export function setupGmailHandlers(ipcMain: IpcMain, store: Store<AppConfig>, ma
         scope: [
           'https://www.googleapis.com/auth/gmail.send',
           'https://www.googleapis.com/auth/gmail.readonly',
-          'https://www.googleapis.com/auth/spreadsheets'
+          'https://www.googleapis.com/auth/spreadsheets',
+          'https://www.googleapis.com/auth/gmail.settings.basic'
         ],
         prompt: 'consent'
       });
@@ -208,14 +209,30 @@ export function setupGmailHandlers(ipcMain: IpcMain, store: Store<AppConfig>, ma
         refresh_token: config.google.refreshToken
       });
 
+      // Fetch user's signature
+      let signature = '';
+      try {
+        const sendAsResponse = await gmail.users.settings.sendAs.list({
+          auth: client,
+          userId: 'me'
+        });
+        const aliases = sendAsResponse.data.sendAs || [];
+        const alias = aliases.find(a => a.sendAsEmail === config.user.email) || aliases.find(a => a.isPrimary);
+        if (alias && alias.signature) {
+          signature = `<br><br>${alias.signature}`;
+        }
+      } catch (sigError) {
+        console.warn('Failed to fetch signature:', sigError);
+      }
+
       // Create email
       const message = [
         `From: ${config.user.name} <${config.user.email}>`,
         `To: ${emailData.to}`,
         `Subject: ${emailData.subject}`,
         'Content-Type: text/html; charset=utf-8',
         '',
-        emailData.body
+        emailData.body + signature
       ].join('\n');
 
       const encodedMessage = Buffer.from(message)

src/main/gmail.ts

@@ -1,4 +1,4 @@
-import { useState, useEffect } from 'react';
+import { useState, useEffect, useRef } from 'react';
 import { AppProvider, useApp } from './context/AppContext';
 import ConfigTab from './components/ConfigTab';
 import ContextTab from './components/ContextTab';
@@ -10,10 +10,11 @@ type Tab = 'config' | 'context' | 'email';
 function AppContent() {
   const { config } = useApp();
   const [activeTab, setActiveTab] = useState<Tab>('config');
+  const initialTabSet = useRef(false);
 
   // Determine initial tab based on configuration completeness
   useEffect(() => {
-    if (!config) return;
+    if (!config || initialTabSet.current) return;
 
     // Check if configuration is complete
     const hasUserInfo = config.user.name && config.user.email;
@@ -33,6 +34,8 @@ function AppContent() {
     } else {
       setActiveTab('email');
     }
+    
+    initialTabSet.current = true;
   }, [config]);
 
   return (