from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
import datetime
import requests
import pytz
import yaml
import json
import os
from tools.final_answer import FinalAnswerTool
from Gradio_UI import GradioUI

@tool
def task_manager(action: str, task_data: str = "") -> str:
    """A tool for managing daily tasks.
    
    Args:
        action: The operation to perform ("add", "list", "complete", "delete", or "help")
        task_data: Additional information related to the task (task description, task ID, etc.)
    
    Returns:
        A text containing information about the result of the operation
    """
    import json
    import os
    from datetime import datetime
    
    # File to store task data
    TASKS_FILE = "tasks.json"
    
    # Create an empty task list 
    if not os.path.exists(TASKS_FILE):
        with open(TASKS_FILE, "w") as f:
            json.dump({"tasks": []}, f)
    
    # Load tasks
    with open(TASKS_FILE, "r") as f:
        data = json.load(f)
    
    # Task list
    tasks = data["tasks"]
    
    # Help message
    if action == "help" or action == "":
        return """Task Manager Tool Usage:
        - add: Add a new task (e.g.: task_manager("add", "Buy milk from the store"))
        - list: List all tasks (e.g.: task_manager("list"))
        - complete: Mark a task as completed (e.g.: task_manager("complete", "1"))
        - delete: Delete a task (e.g.: task_manager("delete", "1"))
        - help: Show this help message"""
    
    # Add a new task
    elif action == "add":
        if not task_data:
            return "Error: Task description cannot be empty!"
        
        # Create a new task
        new_task = {
            "id": len(tasks) + 1,
            "description": task_data,
            "created_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
            "completed": False
        }
        
        # Add the task to the list
        tasks.append(new_task)
        
        # Save changes
        with open(TASKS_FILE, "w") as f:
            json.dump(data, f, indent=2)
        
        return f"Task added: {task_data} (ID: {new_task['id']})"
    
    # List tasks
    elif action == "list":
        if not tasks:
            return "Your task list is empty."
        
        result = "YOUR TASKS:\n"
        result += "-" * 50 + "\n"
        
        for task in tasks:
            status = "✓" if task["completed"] else "□"
            result += f"{task['id']}. [{status}] {task['description']} ({task['created_at']})\n"
        
        return result
    
    # Mark task as completed
    elif action == "complete":
        try:
            task_id = int(task_data)
            
            # Find task by ID
            for task in tasks:
                if task["id"] == task_id:
                    task["completed"] = True
                    
                    # Save changes
                    with open(TASKS_FILE, "w") as f:
                        json.dump(data, f, indent=2)
                    
                    return f"Task completed: {task['description']}"
            
            return f"Error: Task with ID {task_id} not found."
            
        except ValueError:
            return "Error: You must specify a valid task ID."
    
    # Delete task
    elif action == "delete":
        try:
            task_id = int(task_data)
            
            # Find task by ID
            for i, task in enumerate(tasks):
                if task["id"] == task_id:
                    # Remove task from the list
                    deleted_task = tasks.pop(i)
                    
                    # Save changes
                    with open(TASKS_FILE, "w") as f:
                        json.dump(data, f, indent=2)
                    
                    return f"Task deleted: {deleted_task['description']}"
            
            return f"Error: Task with ID {task_id} not found."
            
        except ValueError:
            return "Error: You must specify a valid task ID."
    
    else:
        return f"Error: Unknown action '{action}'. Valid actions: add, list, complete, delete, help"

@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=[final_answer, task_manager, get_current_time_in_timezone, image_generation_tool],  # Eklediğimiz task_manager aracını dahil ettik
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates
)

GradioUI(agent).launch()