Build an AI Email Summarizer Bot with Python and Gemini 🤖

import imaplib
import email
import google.generativeai as genai
import smtplib
from email.mime.text import MIMEText

# --- Configuration ---
# Your Gmail account credentials and server details
GMAIL_USER = "your_email@gmail.com"
GMAIL_APP_PASSWORD = "your_app_password"  # Use the app password you generated
IMAP_SERVER = "imap.gmail.com"
SMTP_SERVER = "smtp.gmail.com"
PORT = 993  # IMAP SSL port
SMTP_PORT = 587 # SMTP TLS port

# Your Google AI Studio API Key
GOOGLE_API_KEY = "your_google_api_key"

# The email address to which summaries will be sent
TARGET_EMAIL = "your_personal_email@example.com"

# --- Function to get emails ---
def get_unread_emails():
    """Connects to Gmail and fetches all unread emails."""
    try:
        mail = imaplib.IMAP4_SSL(IMAP_SERVER, PORT)
        mail.login(GMAIL_USER, GMAIL_APP_PASSWORD)
        mail.select("inbox")
        
        # Search for all unread emails
        status, messages = mail.search(None, "(UNSEEN)")
        
        email_messages = []
        for num in messages[0].split():
            status, data = mail.fetch(num, "(RFC822)")
            email_messages.append(data[0][1].decode("utf-8"))
            
        mail.close()
        mail.logout()
        return email_messages
    except Exception as e:
        print(f"Error connecting to Gmail: {e}")
        return []

# --- Function to summarize email with Gemini Pro ---
def summarize_with_gemini(email_content):
    """Uses the Gemini Pro model to summarize the email content."""
    genai.configure(api_key=GOOGLE_API_KEY)
    model = genai.GenerativeModel('gemini-pro')

    prompt = (
        f"You are a helpful assistant that summarizes emails. "
        f"Provide a concise summary of the following email, focusing on the main points and any action items:\n\n"
        f"--- Email Content ---\n"
        f"{email_content}"
    )

    try:
        response = model.generate_content(prompt)
        return response.text
    except Exception as e:
        print(f"Error calling Gemini API: {e}")
        return "Could not generate a summary."

# --- Function to send an email ---
def send_summary_email(subject, body, to_address):
    """Sends an email using SMTP."""
    msg = MIMEText(body)
    msg['Subject'] = subject
    msg['From'] = GMAIL_USER
    msg['To'] = to_address
    
    try:
        with smtplib.SMTP(SMTP_SERVER, SMTP_PORT) as server:
            server.starttls()  # Secure the connection
            server.login(GMAIL_USER, GMAIL_APP_PASSWORD)
            server.sendmail(GMAIL_USER, to_address, msg.as_string())
        print(f"Successfully sent summary email to {to_address}")
    except Exception as e:
        print(f"Error sending email: {e}")

# --- Main Bot Logic ---
if __name__ == "__main__":
    print("Fetching unread emails...")
    unread_emails_raw = get_unread_emails()
    
    if not unread_emails_raw:
        print("No new unread emails found.")
    else:
        print(f"Found {len(unread_emails_raw)} unread email(s).")
        
        for email_raw in unread_emails_raw:
            msg = email.message_from_string(email_raw)
            subject = msg.get("Subject", "No Subject")
            
            # Find the plain text part of the email body
            email_body = ""
            if msg.is_multipart():
                for part in msg.walk():
                    content_type = part.get_content_type()
                    if content_type == "text/plain":
                        email_body = part.get_payload(decode=True).decode()
                        break
            else:
                email_body = msg.get_payload(decode=True).decode()

            print(f"\nProcessing email with subject: '{subject}'")
            
            # Get summary from the AI
            summary = summarize_with_gemini(email_body)
            
            # Send the summary to your personal email
            summary_subject = f"AI Summary: {subject}"
            summary_body = f"Original Subject: {subject}\n\nSummary:\n{summary}"
            
            send_summary_email(summary_subject, summary_body, TARGET_EMAIL)
            
            # NOTE: The bot does not mark emails as "read" to allow for multiple runs
            # without manual intervention, but you could add this feature if desired.