[V0.4] Added TODOs

backend/modules/nlp/api.py

@@ -15,6 +15,10 @@ from modules.nlp.models import ChatMessage # Import ChatMessage model
 from modules.calendar.service import create_calendar_event, get_calendar_events, update_calendar_event, delete_calendar_event
 from modules.calendar.models import CalendarEvent
 from modules.calendar.schemas import CalendarEventCreate, CalendarEventUpdate
+# Import TODO services, schemas, and model
+from modules.todo import service as todo_service
+from modules.todo.models import Todo
+from modules.todo.schemas import TodoCreate, TodoUpdate
 
 router = APIRouter(prefix="/nlp", tags=["nlp"])
 
@@ -31,6 +35,18 @@ def format_calendar_events(events: List[CalendarEvent]) -> List[str]:
         formatted.append(f"- {title} ({start_str}{' - ' + end_str if end_str else ''})")
     return formatted
 
+# Helper to format TODO items (expects list of Todo models)
+def format_todos(todos: List[Todo]) -> List[str]:
+    if not todos:
+        return ["Your TODO list is empty."]
+    formatted = ["Here is your TODO list:"]
+    for todo in todos:
+        status = "[X]" if todo.complete else "[ ]"
+        date_str = f" (Due: {todo.date.strftime('%Y-%m-%d')})" if todo.date else ""
+        remind_str = " (Reminder)" if todo.remind else ""
+        formatted.append(f"- {status} {todo.task}{date_str}{remind_str} (ID: {todo.id})")
+    return formatted
+
 # Update the response model for the endpoint
 @router.post("/process-command", response_model=ProcessCommandResponse)
 def process_command(request_data: ProcessCommandRequest, current_user: User = Depends(get_current_user), db: Session = Depends(get_db)):
@@ -129,6 +145,60 @@ def process_command(request_data: ProcessCommandRequest, current_user: User = De
                 # ---------------------------------
                 return ProcessCommandResponse(responses=responses)
 
+            # --- Add TODO Cases --- 
+            case "get_todos":
+                todos: List[Todo] = todo_service.get_todos(db, user=current_user, **params)
+                formatted_responses = format_todos(todos)
+                responses.extend(formatted_responses)
+                # --- Save Additional AI Responses ---
+                for resp in formatted_responses:
+                    save_chat_message(db, user_id=current_user.id, sender=MessageSender.AI, text=resp)
+                # ----------------------------------
+                return ProcessCommandResponse(responses=responses)
+
+            case "add_todo":
+                todo_data = TodoCreate(**params)
+                created_todo = todo_service.create_todo(db, todo=todo_data, user=current_user)
+                add_response = f"Added TODO: '{created_todo.task}' (ID: {created_todo.id})."
+                responses.append(add_response)
+                # --- Save Additional AI Response ---
+                save_chat_message(db, user_id=current_user.id, sender=MessageSender.AI, text=add_response)
+                # ---------------------------------
+                return ProcessCommandResponse(responses=responses)
+
+            case "update_todo":
+                todo_id = params.pop('todo_id', None)
+                if todo_id is None:
+                    error_msg = "TODO ID is required for update."
+                    save_chat_message(db, user_id=current_user.id, sender=MessageSender.AI, text=error_msg)
+                    raise HTTPException(status_code=400, detail=error_msg)
+                todo_data = TodoUpdate(**params)
+                updated_todo = todo_service.update_todo(db, todo_id=todo_id, todo_update=todo_data, user=current_user)
+                update_response = f"Updated TODO ID {updated_todo.id}: '{updated_todo.task}'."
+                if 'complete' in params:
+                    status = "complete" if params['complete'] else "incomplete"
+                    update_response += f" Marked as {status}."
+                responses.append(update_response)
+                # --- Save Additional AI Response ---
+                save_chat_message(db, user_id=current_user.id, sender=MessageSender.AI, text=update_response)
+                # ---------------------------------
+                return ProcessCommandResponse(responses=responses)
+
+            case "delete_todo":
+                todo_id = params.get('todo_id')
+                if todo_id is None:
+                    error_msg = "TODO ID is required for delete."
+                    save_chat_message(db, user_id=current_user.id, sender=MessageSender.AI, text=error_msg)
+                    raise HTTPException(status_code=400, detail=error_msg)
+                deleted_todo = todo_service.delete_todo(db, todo_id=todo_id, user=current_user)
+                delete_response = f"Deleted TODO ID {deleted_todo.id}: '{deleted_todo.task}'."
+                responses.append(delete_response)
+                # --- Save Additional AI Response ---
+                save_chat_message(db, user_id=current_user.id, sender=MessageSender.AI, text=delete_response)
+                # ---------------------------------
+                return ProcessCommandResponse(responses=responses)
+            # --- End TODO Cases ---
+
             case _: 
                 print(f"Warning: Unhandled intent '{intent}' reached api.py match statement.")
                 # The initial response_text was already saved