Add todo management tools and refactor app configuration

app.py

@@ -1,65 +1,62 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
-import datetime
-import requests
-import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
+from smolagents import (
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    HfApiModel,
+    OpenAIServerModel,
+    load_tool,
+    tool,
+)
 
 from Gradio_UI import GradioUI
-
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def my_cutom_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)}"
-
+from todo_agents import (
+    add_task,
+    get_current_time_in_timezone,
+    get_todays_tasks,
+    update_task_status,
+)
+from tools.final_answer import FinalAnswerTool
 
 final_answer = FinalAnswerTool()
+
+# Remote LLM
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',# it is possible that this model may be overloaded
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id="https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud",  # it is possible that this model may be overloaded
+    custom_role_conversions=None,
 )
 
+# # Local LLM
+# model = OpenAIServerModel(
+#     model_id="Qwen/Qwen2.5-Coder-14B-Instruct-GGUF",
+#     api_base="http://100.81.11.125:1234/v1",
+#     api_key="lm-studio",
+# )
 
 # 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:
+with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
-    
+
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[
+        final_answer,
+        get_current_time_in_timezone,
+        get_todays_tasks,
+        add_task,
+        update_task_status,
+    ],  ## 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
+    prompt_templates=prompt_templates,
 )
 
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()

todo.jsonl

@@ -0,0 +1,16 @@
+{"id": "46aa6a55-573c-4325-87ab-5785bd2b72ad", "completed": true, "description": "Buy groceries (milk, bread, eggs)", "last_updated": "2024-03-21"}
+{"id": "d6a05c8d-0fb7-4a68-be17-5b94e07bfc1c", "completed": false, "description": "Water the plants", "last_updated": "2025-02-11"}
+{"id": "7bf3fef5-ed29-483a-81ef-a3c49dd00fc0", "completed": true, "description": "Take out trash", "last_updated": "2024-03-21"}
+{"id": "b5936a2c-df6b-428c-a074-685e0b5c2bba", "completed": true, "description": "Call mom", "last_updated": "2024-03-21"}
+{"id": "eedf7562-6c34-467e-be85-47bad8972005", "completed": false, "description": "Pay electricity bill", "last_updated": "2024-03-21"}
+{"id": "841cff36-5319-4745-b3f5-8b90c3427d4f", "completed": true, "description": "Cut grass in backyard", "last_updated": "2024-03-21"}
+{"id": "55423c2c-bd80-4221-a191-d9ec664dbcbe", "completed": true, "description": "Do laundry", "last_updated": "2025-02-11"}
+{"id": "99fd044e-24fe-4190-9c49-84876fd24528", "completed": true, "description": "Clean bathroom", "last_updated": "2024-03-21"}
+{"id": "d183d6ed-cd5a-4551-9520-e28e610a0f1c", "completed": true, "description": "Buy birthday gift for John", "last_updated": "2024-03-21"}
+{"id": "3cc00430-07be-407f-91bb-3f61e41a6861", "completed": true, "description": "Schedule dentist appointment", "last_updated": "2025-02-11"}
+{"id": "d8c9a03a-f932-49e6-aa15-b49623c2d681", "completed": true, "description": "Vacuum living room", "last_updated": "2024-03-21"}
+{"id": "499dc82a-9e45-48eb-8847-e60c9195e3cd", "completed": true, "description": "Feed the cat", "last_updated": "2024-03-21"}
+{"id": "0527b742-d522-47da-9c05-3f5d8da7cb9a", "completed": false, "description": "Pick up dry cleaning", "last_updated": "2024-03-21"}
+{"id": "803c8862-f346-4024-81d8-76192ad49599", "completed": true, "description": "Buy new light bulbs", "last_updated": "2024-03-21"}
+{"id": "65118069-b2a4-4ea3-b8e9-d7921381cbfc", "completed": true, "description": "Repair my computer", "last_updated": "2025-02-11"}
+{"id": "723323ff-da01-47e0-8837-8feca576eb4e", "completed": false, "description": "Call my mom again", "last_updated": "2025-02-11"}

todo_agents.py

@@ -0,0 +1,166 @@
+import datetime
+import json
+import uuid
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Dict, List
+
+import pytz
+from smolagents import tool
+
+
+@dataclass
+class Task:
+    id: str = None
+    completed: bool = False
+    description: str = ""
+    last_updated: str = ""
+
+    def __post_init__(self):
+        if self.id is None:
+            self.id = str(uuid.uuid4())
+
+    def to_dict(self) -> Dict:
+        return {
+            "id": self.id,
+            "completed": self.completed,
+            "description": self.description,
+            "last_updated": self.last_updated,
+        }
+
+
+class TodoManager:
+    def __init__(self, filepath: str = "todo.jsonl"):
+        self.filepath = Path(filepath)
+        if not self.filepath.exists():
+            self.filepath.write_text("")
+
+    def _read_tasks(self) -> List[Task]:
+        if not self.filepath.exists():
+            return []
+
+        try:
+            with self.filepath.open("r", encoding="utf-8") as file:
+                data = [json.loads(line) for line in file if line.strip()]
+
+            return [
+                Task(
+                    description=item.get("description", item.get("task", "")),
+                    completed=item.get("completed", item.get("status") == "✅"),
+                    last_updated=item["last_updated"],
+                    id=item.get("id"),
+                )
+                for item in data
+            ]
+        except json.JSONDecodeError:
+            return []
+
+    def _save_tasks(self, tasks: List[Task]):
+        with self.filepath.open("w") as file:
+            for task in tasks:
+                file.write(json.dumps(task.to_dict()) + "\n")
+
+
+@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:
+        tz = pytz.timezone(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)}"
+
+
+@tool
+def get_todays_tasks() -> str:
+    """List all tasks with their status and details.
+    Args:
+        None
+    Returns:
+        A formatted string containing all tasks with their status and IDs.
+    """
+    manager = TodoManager("./todo.jsonl")
+    tasks = manager._read_tasks()
+
+    if not tasks:
+        return "No tasks found"
+
+    pending = [t for t in tasks if not t.completed]
+    completed = [t for t in tasks if t.completed]  # Uncomment this to show all tasks
+
+    output = [f"📅 Tasks Report for {datetime.date.today()}"]
+    output.append("\n🔴 Pending Tasks:")
+    output.extend(f"[{t.id[:8]}] {t.description}" for t in pending)
+    output.append("\n✅ Completed Tasks:")  # Show both pending and completed tasks
+    output.extend(f"[{t.id[:8]}] {t.description}" for t in completed)
+    output.append(f"\nTotal: {len(pending)} pending, {len(completed)} completed")
+
+    return "\n".join(output)
+
+
+@tool
+def update_task_status(task_id: str) -> str:
+    """Toggle a task's completion status between complete and pending.
+    Args:
+        task_id: First 8 characters of the task's UUID (e.g., '65118069' from '65118069-b2a4-4ea3-b8e9-d7921381cbfc').
+                You can find task IDs in square brackets when listing tasks.
+    Returns:
+        Success message with task details if found and updated.
+        Error message if task not found or invalid input.
+    Examples:
+        > update_task_status('65118069')
+        "✅ Task [65118069] marked as complete: Repair my computer"
+        > update_task_status('65118069')
+        "✅ Task [65118069] marked as pending: Repair my computer"
+    """
+    # Validate task_id format
+    if not task_id or not isinstance(task_id, str) or len(task_id) < 1:
+        return "❌ Please provide a valid task ID"
+
+    try:
+        manager = TodoManager("./todo.jsonl")
+        tasks = manager._read_tasks()
+
+        if not tasks:
+            return "❌ No tasks found in the todo list"
+
+        # Find and update the task
+        for task in tasks:
+            if task.id.startswith(task_id):
+                task.completed = not task.completed
+                task.last_updated = datetime.date.today().strftime("%Y-%m-%d")
+                manager._save_tasks(tasks)
+                return f"✅ Task [{task_id}] marked as {'complete' if task.completed else 'pending'}: {task.description}"
+
+        return f"❌ No task found with ID '{task_id}'"
+
+    except (json.JSONDecodeError, IOError) as e:
+        return f"❌ Error accessing todo list: {str(e)}"
+    except Exception as e:
+        return f"❌ Unexpected error: {str(e)}"
+
+
+@tool
+def add_task(description: str) -> str:
+    """Add a new task to the todo list.
+    Args:
+        description: The text description of the task to add.
+    Returns:
+        A confirmation message indicating the task was added.
+    """
+    manager = TodoManager("./todo.jsonl")
+    tasks = manager._read_tasks()
+
+    new_task = Task(
+        description=description,
+        completed=False,
+        last_updated=datetime.date.today().strftime("%Y-%m-%d"),
+    )
+
+    tasks.append(new_task)
+    manager._save_tasks(tasks)
+    return f"✅ Added new task: {description}"