fixes

api/endpoints/generation_router.py

@@ -66,6 +66,7 @@ async def get_generations(
         character_id: Optional[str] = None, 
         limit: int = 10, 
         offset: int = 0,
+        only_liked: bool = False,
         generation_service: GenerationService = Depends(get_generation_service),
         current_user: dict = Depends(get_current_user),
         project_id: Optional[str] = Depends(get_project_id),
@@ -75,13 +76,16 @@ async def get_generations(
     
     # If project_id is set, we don't filter by user to show all project-wide generations
     created_by_filter = None if project_id else str(current_user["_id"])
+    only_liked_by = str(current_user["_id"]) if only_liked else None
 
     return await generation_service.get_generations(
         character_id=character_id, 
         limit=limit, 
         offset=offset, 
         created_by=created_by_filter, 
-        project_id=project_id
+        project_id=project_id,
+        only_liked_by=only_liked_by,
+        current_user_id=str(current_user["_id"])
     )
 
 
@@ -150,7 +154,7 @@ async def get_generation_group(
         generation_service: GenerationService = Depends(get_generation_service),
         current_user: dict = Depends(get_current_user)
 ):
-    return await generation_service.get_generations_by_group(group_id)
+    return await generation_service.get_generations_by_group(group_id, current_user_id=str(current_user["_id"]))
 
 
 @router.get("/{generation_id}", response_model=GenerationResponse)
@@ -159,7 +163,7 @@ async def get_generation(
         generation_service: GenerationService = Depends(get_generation_service),
         current_user: dict = Depends(get_current_user)
 ) -> GenerationResponse:
-    gen = await generation_service.get_generation(generation_id)
+    gen = await generation_service.get_generation(generation_id, current_user_id=str(current_user["_id"]))
     if not gen:
         raise HTTPException(status_code=404, detail="Generation not found")
         
@@ -176,6 +180,18 @@ async def get_generation(
     return gen
 
 
+@router.post("/{generation_id}/like", response_model=dict)
+async def toggle_like(
+        generation_id: str,
+        generation_service: GenerationService = Depends(get_generation_service),
+        current_user: dict = Depends(get_current_user)
+):
+    is_liked = await generation_service.toggle_like(generation_id, str(current_user["_id"]))
+    if is_liked is None:
+        raise HTTPException(status_code=404, detail="Generation not found")
+    return {"is_liked": is_liked}
+
+
 @router.post("/import", response_model=GenerationResponse)
 async def import_external_generation(
     request: Request,

api/endpoints/idea_router.py

@@ -68,18 +68,10 @@ async def get_idea_generations(
     idea_id: str,
     limit: int = 50,
     offset: int = 0,
-    generation_service: GenerationService = Depends(get_generation_service)
+    generation_service: GenerationService = Depends(get_generation_service),
+    current_user: dict = Depends(get_current_user)
 ):
-    # Depending on how generation service implements filtering by idea_id.
-    # We might need to update generation_service to support getting by idea_id directly
-    # or ensure generic get_generations supports it.
-    # Looking at generation_router.py, get_generations doesn't have idea_id arg?
-    # Let's check generation_service.get_generations signature again.
-    # It has: (character_id, limit, offset, user_id, project_id). NO IDEA_ID.
-    # I need to update GenerationService.get_generations too!
-    
-    # For now, let's assume I will update it.
-    return await generation_service.get_generations(idea_id=idea_id, limit=limit, offset=offset)
+    return await generation_service.get_generations(idea_id=idea_id, limit=limit, offset=offset, current_user_id=str(current_user["_id"]))
 
 @router.post("/{idea_id}/generations/{generation_id}")
 async def add_generation_to_idea(