Update app.py

app.py

@@ -7,6 +7,64 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from smolagents.tools import tool
+
+# Your Gmail SMTP credentials
+SMTP_SERVER = "smtp.gmail.com"
+SMTP_PORT = 587
+SMTP_USERNAME = "[email protected]"  # Replace with your Gmail
+SMTP_PASSWORD = "wbso priu rshv atep"  # Replace with your actual App Password
+
+@tool
+def send_email_tool(recipient: str, subject: str, body: str, recipient_email: str = None) -> str:
+    """A tool that sends an actual email. If the recipient is not in the predefined list, it asks for an email ID.
+    
+    Args:
+        recipient: A string representing the contact name or any custom name.
+        subject: The subject line of the email.
+        body: The main content of the email.
+        recipient_email: (Optional) The email address if recipient is not predefined.
+    
+    Returns:
+        A success or failure message indicating whether the email was sent.
+    """
+    # Predefined contact list
+    contacts = {
+        "shan1": "[email protected]",
+        "shan2": "[email protected]",
+        "shan3": "[email protected]"
+    }
+
+    # If recipient is in predefined list, use their email
+    if recipient in contacts:
+        recipient_email = contacts[recipient]
+    elif not recipient_email:
+        return f"⚠️ Please provide an email address for '{recipient}'."
+
+    # Create the email
+    msg = MIMEMultipart()
+    msg["From"] = SMTP_USERNAME
+    msg["To"] = recipient_email
+    msg["Subject"] = subject
+    msg.attach(MIMEText(body, "plain"))
+
+    try:
+        # Connect to Gmail SMTP Server
+        server = smtplib.SMTP(SMTP_SERVER, SMTP_PORT)
+        server.starttls()  # Secure the connection
+        server.login(SMTP_USERNAME, SMTP_PASSWORD)
+        server.sendmail(SMTP_USERNAME, recipient_email, msg.as_string())
+        server.quit()
+
+        return f"✅ Email successfully sent to {recipient_email} with subject '{subject}'."
+    except Exception as e:
+        return f"❌ Failed to send email: {str(e)}"
+
+
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -55,7 +113,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=[send_email_tool,image_generation_tool,get_current_time_in_timezone,final_answer,DuckDuckGoSearchTool(),useless_advice], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,