from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
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 = "shantanudave100@gmail.com"  # 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": "shantanu.dave@canda.com",
        "shan2": "daveshantanu1@gmail.com",
        "shan3": "shantanu.dave01@gmail.com"
    }

    # 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
    #Keep this format for the description / args / args description but feel free to modify the tool
    """A tool that does nothing yet 
    Args:
        arg1: the first argument
        arg2: the second argument
    """
    return "What magic will you build ?"

@tool
def get_current_time_in_timezone(timezone: str) -> str:
    """A tool that fetches the current local time in a specified timezone.
    Args:
        timezone: A string representing a valid timezone (e.g., 'America/New_York').
    """
    try:
        # Create timezone object
        tz = pytz.timezone(timezone)
        # Get current time in that timezone
        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
        return f"The current local time in {timezone} is: {local_time}"
    except Exception as e:
        return f"Error fetching time for timezone '{timezone}': {str(e)}"


final_answer = FinalAnswerTool()

# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)


# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[send_email_tool,image_generation_tool,get_current_time_in_timezone,final_answer,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()