Update app.py

app.py

@@ -1,4 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+import os
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
 import datetime
 import requests
 import pytz
@@ -18,6 +21,75 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+def get_gmail_service():
+    creds = Credentials(
+        token=None,
+        refresh_token=os.environ["GMAIL_REFRESH_TOKEN"],
+        client_id=os.environ["GMAIL_CLIENT_ID"],
+        client_secret=os.environ["GMAIL_CLIENT_SECRET"],
+        token_uri="https://oauth2.googleapis.com/token",
+    )
+    return build("gmail", "v1", credentials=creds)
+
+@tool
+def read_latest_emails(max_results: int = 5) -> str:
+    """Reads the latest emails from Gmail inbox.
+    
+    Args:
+        max_results: Number of emails to fetch (default 5)
+    
+    Returns:
+        A summary of recent emails
+    """
+    service = get_gmail_service()
+    results = service.users().messages().list(
+        userId="me", maxResults=max_results, labelIds=["INBOX"]
+    ).execute()
+    
+    messages = results.get("messages", [])
+    summaries = []
+    
+    for msg in messages:
+        detail = service.users().messages().get(
+            userId="me", id=msg["id"], format="metadata",
+            metadataHeaders=["From", "Subject", "Date"]
+        ).execute()
+        headers = {h["name"]: h["value"] for h in detail["payload"]["headers"]}
+        summaries.append(
+            f"From: {headers.get('From')}\n"
+            f"Subject: {headers.get('Subject')}\n"
+            f"Date: {headers.get('Date')}"
+        )
+    
+    return "\n\n".join(summaries)  
+
+@tool
+def send_email(to: str, subject: str, body: str) -> str:
+    """Sends an email via Gmail.
+    
+    Args:
+        to: Recipient email address
+        subject: Email subject
+        body: Email body text
+    
+    Returns:
+        Success or error message
+    """
+    import base64
+    from email.mime.text import MIMEText
+    
+    service = get_gmail_service()
+    message = MIMEText(body)
+    message["to"] = to
+    message["subject"] = subject
+    
+    raw = base64.urlsafe_b64encode(message.as_bytes()).decode()
+    service.users().messages().send(
+        userId="me", body={"raw": raw}
+    ).execute()
+    
+    return f"Email sent to {to} successfully!"
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -56,7 +128,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer],[read_latest_emails,send_email], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,