Update src/main.py

src/main.py

@@ -2,7 +2,6 @@
 import os
 import sys
 from pathlib import Path
-from contextlib import asynccontextmanager
 from datetime import datetime
 from typing import Optional
 
@@ -28,27 +27,21 @@ from api.chat_router import router as chat_router
 from api.task_router import router as task_router
 from middleware.auth import get_current_user
 
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    # Startup logic
-    Base.metadata.create_all(bind=engine)
-    yield
-    # Shutdown logic (if any) can go here
-
 app = FastAPI(
     title="Todo API",
     description="REST API for managing todo items with extended features",
-    version="2.0.0",
-    lifespan=lifespan
+    version="2.0.0"
 )
 
-# FIXED CORS: Added GitHub Pages support
+# FIXED STARTUP: Using on_event to avoid the "async iterator" TypeError
+@app.on_event("startup")
+def startup_event():
+    Base.metadata.create_all(bind=engine)
+
+# FIXED CORS: Using "*" allows your GitHub Pages site to connect without being blocked
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=[
-        "http://localhost:3000",
-        "https://shaheryarshah.github.io", # <--- CHANGE THIS to your GitHub Pages URL
-    ],
+    allow_origins=["*"], 
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -60,26 +53,27 @@ app.include_router(chat_router)
 app.include_router(task_router, prefix="/api/v1")
 
 def _calculate_overdue(todo: Todo) -> bool:
+    """Calculate if a todo is overdue."""
     if todo.due_date is None or todo.completed:
         return False
     return todo.due_date < datetime.utcnow()
 
 def _todo_to_response(todo: Todo) -> TodoResponse:
-    response_data = {
-        "id": todo.id,
-        "user_id": todo.user_id,
-        "title": todo.title,
-        "description": todo.description,
-        "completed": todo.completed,
-        "priority": todo.priority,
-        "tags": todo.tags,
-        "due_date": todo.due_date,
-        "recurrence": todo.recurrence,
-        "created_at": todo.created_at,
-        "updated_at": todo.updated_at,
-        "overdue": _calculate_overdue(todo)
-    }
-    return TodoResponse(**response_data)
+    """Convert Todo model to TodoResponse schema."""
+    return TodoResponse(
+        id=todo.id,
+        user_id=todo.user_id,
+        title=todo.title,
+        description=todo.description,
+        completed=todo.completed,
+        priority=todo.priority,
+        tags=todo.tags,
+        due_date=todo.due_date,
+        recurrence=todo.recurrence,
+        created_at=todo.created_at,
+        updated_at=todo.updated_at,
+        overdue=_calculate_overdue(todo)
+    )
 
 @app.get("/api/v1/todos", response_model=TodoListResponse)
 def get_todos(
@@ -103,10 +97,36 @@ def get_todos(
     todos_response = [_todo_to_response(todo) for todo in todos]
     return {"todos": todos_response, "count": len(todos_response), "has_more": False}
 
-# ... (Keep other routes as they are, but ensure regular spaces are used)
+@app.post("/api/v1/todos", response_model=TodoResponse, status_code=201)
+def create_todo(todo_data: TodoCreate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
+    service = TodoService(db, user_id=current_user.id)
+    todo = service.create(todo_data)
+    return _todo_to_response(todo)
+
+@app.get("/api/v1/todos/{todo_id}", response_model=TodoResponse)
+def get_todo(todo_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
+    service = TodoService(db, user_id=current_user.id)
+    todo = service.get_by_id(todo_id)
+    if not todo:
+        raise HTTPException(status_code=404, detail="Todo not found")
+    return _todo_to_response(todo)
+
+@app.put("/api/v1/todos/{todo_id}", response_model=TodoResponse)
+def update_todo(todo_id: int, todo_data: TodoUpdate, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
+    service = TodoService(db, user_id=current_user.id)
+    todo = service.update(todo_id, todo_data)
+    if not todo:
+        raise HTTPException(status_code=404, detail="Todo not found")
+    return _todo_to_response(todo)
+
+@app.delete("/api/v1/todos/{todo_id}")
+def delete_todo(todo_id: int, db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
+    service = TodoService(db, user_id=current_user.id)
+    if not service.delete(todo_id):
+        raise HTTPException(status_code=404, detail="Todo not found")
+    return {"message": "Todo deleted successfully"}
 
 if __name__ == "__main__":
     import uvicorn
-    # Hugging Face uses PORT 7860 by default
-    port = int(os.getenv("PORT", 7860))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+    # PORT 7860 is required for Hugging Face Spaces
+    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))