Update app.py

app.py

@@ -3,9 +3,14 @@ import datetime
 import requests
 import pytz
 import yaml
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# -- Your Custom Tools --
+
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """A tool that does nothing yet 
@@ -28,20 +33,64 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Modified Final Answer with name prefix
+@tool
+def send_html_email(
+    smtp_server: str,
+    smtp_port: int,
+    sender_email: str,
+    sender_password: str,
+    recipient_email: str,
+    subject: str,
+    html_content: str
+) -> str:
+    """
+    A tool to send an HTML email.
+    Args:
+        smtp_server: The SMTP server address (e.g., 'smtp.gmail.com')
+        smtp_port: The SMTP server port (e.g., 587 for TLS)
+        sender_email: The sender's email address
+        sender_password: The sender's email password or app-specific password
+        recipient_email: The recipient's email address
+        subject: The subject of the email
+        html_content: The HTML content of the email
+    Returns:
+        A status message indicating success or failure.
+    """
+    try:
+        message = MIMEMultipart()
+        message['From'] = sender_email
+        message['To'] = recipient_email
+        message['Subject'] = subject
+        message.attach(MIMEText(html_content, 'html'))
+
+        session = smtplib.SMTP(smtp_server, smtp_port)
+        session.starttls()
+        session.login(sender_email, sender_password)
+        session.sendmail(sender_email, recipient_email, message.as_string())
+        session.quit()
+
+        return "Email sent successfully!"
+    except Exception as e:
+        return f"Failed to send email: {str(e)}"
+
+# -- Custom Final Answer Tool --
+
 class CustomFinalAnswer(FinalAnswerTool):
     def __call__(self, answer: str) -> str:
         return f"Phoom Said: {super().__call__(answer)}"
 
 final_answer = CustomFinalAnswer()
 
-# Load and modify prompts
+# -- Load Prompts --
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 prompt_templates['system'] = prompt_templates.get('system', '') + \
     "\n\nIMPORTANT: Always begin your final answer with 'Phoom Said:' followed by the response."
 
+# -- Load Model and Tools --
+
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -53,7 +102,13 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, my_custom_tool],
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        image_generation_tool,
+        my_custom_tool,
+        send_html_email  # <<<<< ADDED EMAIL TOOL
+    ],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -63,4 +118,6 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+# -- Launch UI --
+
+GradioUI(agent).launch()