email_draft_manager.py

import json
import requests
import re
from datetime import datetime, timedelta

def extract_sender_info(from_header):
   name = re.match(r'^([^<]+)', from_header)
   email = re.findall(r'<(.+?)>', from_header)
   return {
       'name': name.group(1).strip() if name else '',
       'email': email[0] if email else from_header.strip()
   }

def batch_classify_emails(emails):
   messages = [
       {"role": "system", "content": "You are an email assistant that determines if emails need replies."},
       {"role": "user", "content": f"""For each email, determine if it needs a reply. Reply with a JSON array of true/false.

Emails:
{[{
   'subject': e['subject'],
   'from': extract_sender_info(e['from'])['email'],
} for e in emails]}

Rules:
- true: Personal emails, business communications needing action, direct questions
- false: Marketing, newsletters, broadcasts, automated notifications, no-reply addresses"""}
   ]

   response = requests.post(
       "https://scripty.me/api/assistant/call",
       headers={"Authorization": f"Bearer {authtoken}"},
       json={"model": "mixtral-8x7b-32768", "messages": messages}
   ).json()

   try:
       return json.loads(response.get('content', '[]'))
   except:
       return [False] * len(emails)

def generate_draft(subject, body, from_header, to_addr):
   sender = extract_sender_info(from_header)
   
   messages = [
       {"role": "system", "content": f"You are an email assistant helping {to_addr} write replies. Do not include Subject or email headers in your response."},
       {"role": "user", "content": f"""Write a reply from {to_addr} to {sender['name']}.

Original Email:
Body: {body}

Guidelines:
1. Start with greeting
2. Match original tone
3. Address all points
4. Be concise"""}
   ]

   response = requests.post(
       "https://scripty.me/api/assistant/call",
       headers={"Authorization": f"Bearer {authtoken}"},
       json={
           "model": "mixtral-8x7b-32768",
           "messages": messages
       }
   ).json()

   draft = response.get('content', "Thank you for your email. I will respond shortly.")
   
   return {
       "content": draft,
       "reply_to": sender['email'],
       "reply_subject": f"Re: {re.sub('^(Re: )*', '', subject)}"
   }

def process_unread_emails(days=1):
   try:
       print(f"Checking emails from past {days} days")
       response = requests.get(
           "https://scripty.me/api/assistant/email/unread",
           headers={"Authorization": f"Bearer {authtoken}"},
           params={"days": days}
       )
       emails = response.json()
       print(f"Found {len(emails) if isinstance(emails, list) else 'invalid'} emails")
       if not isinstance(emails, list):
           return {"success": False, "error": "Invalid response from email API"}
       
       results = []
       needs_reply = batch_classify_emails(emails)
       
       for i, email_data in enumerate(emails):
           if needs_reply[i]:
               draft = generate_draft(
                   email_data["subject"],
                   email_data["body"], 
                   email_data["from"],
                   email_data["to"]
               )
               
               if draft:
                   formatted_draft = {
                       "original_message_id": email_data["message_id"],
                       "content": draft["content"],
                       "subject": draft["reply_subject"],  
                       "to": draft["reply_to"]
                   }
                   
                   requests.post(
                       "https://scripty.me/api/assistant/email/draft",
                       headers={"Authorization": f"Bearer {authtoken}"},
                       json=formatted_draft
                   )
                   results.append({
                       "subject": email_data["subject"],
                       "draft_created": True
                   })

       return {"success": True, "results": results}
   except Exception as e:
       return {"success": False, "error": str(e)}

async def func(args):
   try:
       days = int(args.get("days", 1))
       return json.dumps(process_unread_emails(days))
   except Exception as e:
       return json.dumps({"success": False, "error": str(e)})

object = {
   "name": "email_draft_manager",
   "description": "Process unread emails and generate AI-powered draft replies",
   "parameters": {
       "type": "object",
       "properties": {
           "days": {
               "type": "number",
               "description": "Number of days back to process emails",
               "default": 1
           }
       }
   }
}