+ ideas

api/dependency.py

@@ -45,6 +45,11 @@ def get_generation_service(
 ) -> GenerationService:
     return GenerationService(dao, gemini, s3_adapter, bot)
 
+from api.service.idea_service import IdeaService
+
+def get_idea_service(dao: DAO = Depends(get_dao)) -> IdeaService:
+    return IdeaService(dao)
+
 from fastapi import Header
 
 async def get_project_id(x_project_id: Optional[str] = Header(None, alias="X-Project-ID")) -> Optional[str]:

api/endpoints/idea_router.py

@@ -0,0 +1,61 @@
+from typing import List, Optional
+from fastapi import APIRouter, Depends, HTTPException, Query
+from api.dependency import get_idea_service, get_current_user, get_project_id, get_generation_service
+from api.service.idea_service import IdeaService
+from api.service.generation_service import GenerationService
+from models.Idea import Idea
+from api.models.GenerationRequest import GenerationResponse
+
+router = APIRouter(prefix="/ideas", tags=["ideas"])
+
+@router.post("", response_model=Idea)
+async def create_idea(
+    name: str,
+    project_id: str = Depends(get_project_id),
+    current_user: dict = Depends(get_current_user),
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    if not project_id:
+        raise HTTPException(status_code=400, detail="Project ID header is required")
+    
+    return await idea_service.create_idea(name, project_id, str(current_user["_id"]))
+
+@router.get("", response_model=List[Idea])
+async def get_ideas(
+    project_id: str = Depends(get_project_id),
+    limit: int = 20,
+    offset: int = 0,
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    if not project_id:
+        raise HTTPException(status_code=400, detail="Project ID header is required")
+    return await idea_service.get_ideas(project_id, limit, offset)
+
+@router.get("/{idea_id}", response_model=Idea)
+async def get_idea(
+    idea_id: str,
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    idea = await idea_service.get_idea(idea_id)
+    if not idea:
+        raise HTTPException(status_code=404, detail="Idea not found")
+    return idea
+
+@router.delete("/{idea_id}")
+async def delete_idea(
+    idea_id: str,
+    idea_service: IdeaService = Depends(get_idea_service)
+):
+    success = await idea_service.delete_idea(idea_id)
+    if not success:
+        raise HTTPException(status_code=404, detail="Idea not found or could not be deleted")
+    return {"status": "success"}
+
+@router.get("/{idea_id}/generations", response_model=List[GenerationResponse])
+async def get_idea_generations(
+    idea_id: str,
+    limit: int = 50,
+    offset: int = 0,
+    generation_service: GenerationService = Depends(get_generation_service)
+):
+    return await generation_service.get_generations(idea_id=idea_id, limit=limit, offset=offset)

api/models/GenerationRequest.py

@@ -17,6 +17,7 @@ class GenerationRequest(BaseModel):
     use_profile_image: bool = True
     assets_list: List[str]
     project_id: Optional[str] = None
+    idea_id: Optional[str] = None
     count: int = Field(default=1, ge=1, le=10)
 
 
@@ -47,6 +48,7 @@ class GenerationResponse(BaseModel):
     cost: Optional[float] = None
     created_by: Optional[str] = None
     generation_group_id: Optional[str] = None
+    idea_id: Optional[str] = None
     created_at: datetime = datetime.now(UTC)
     updated_at: datetime = datetime.now(UTC)
 

api/service/generation_service.py

@@ -137,6 +137,10 @@ class GenerationService:
             if generation_group_id:
                 generation_model.generation_group_id = generation_group_id
             
+            # Explicitly set idea_id from request if present (already in model_dump, but ensuring clarity)
+            if generation_request.idea_id:
+                generation_model.idea_id = generation_request.idea_id
+            
             gen_id = await self.dao.generations.create_generation(generation_model)
             generation_model.id = gen_id
 

api/service/idea_service.py

@@ -0,0 +1,22 @@
+from typing import List, Optional
+from repos.dao import DAO
+from models.Idea import Idea
+
+class IdeaService:
+    def __init__(self, dao: DAO):
+        self.dao = dao
+
+    async def create_idea(self, name: str, project_id: str, user_id: str) -> Idea:
+        idea = Idea(name=name, project_id=project_id, created_by=user_id)
+        idea_id = await self.dao.ideas.create_idea(idea)
+        idea.id = idea_id
+        return idea
+
+    async def get_ideas(self, project_id: str, limit: int = 20, offset: int = 0) -> List[Idea]:
+        return await self.dao.ideas.get_ideas(project_id, limit, offset)
+
+    async def get_idea(self, idea_id: str) -> Optional[Idea]:
+        return await self.dao.ideas.get_idea(idea_id)
+
+    async def delete_idea(self, idea_id: str) -> bool:
+        return await self.dao.ideas.delete_idea(idea_id)