Function create_collection

backend/src/flashcards/api.py:45–78

(
    session: SessionDep,
    current_user: CurrentUser,
    collection_in: CollectionCreate,
    provider: GeminiProviderDep,
)

@router.post("/collections/", response_model=Collection)
async def create_collection(
    session: SessionDep,
    current_user: CurrentUser,
    collection_in: CollectionCreate,
    provider: GeminiProviderDep,
) -> Any:
    name = collection_in.name
    cards = None

    if collection_in.prompt:
        try:
            if not await asyncio.to_thread(
                lambda: check_and_increment_ai_usage_quota(session, current_user)
            ):
                raise HTTPException(
                    status_code=429, detail="Quota for AI usage is reached."
                )
            flashcard_collection = await services.generate_ai_collection(
                provider, collection_in.prompt
            )
            name = flashcard_collection.name
            cards = flashcard_collection.cards
        except EmptyCollectionError:
            raise HTTPException(
                status_code=400, detail="Failed to generate flashcards from the prompt"
            )
        except AIGenerationError as e:
            raise HTTPException(status_code=500, detail=str(e))

    return await asyncio.to_thread(
        lambda: services.create_collection(
            session=session, user_id=current_user.id, name=name, cards=cards
        )
    )


@router.get("/collections/{collection_id}", response_model=Collection)