Add display_number to content creation logic

Author: lickem22

core_backend/app/contents/models.py

@@ -141,12 +141,16 @@ async def save_content_to_db(
     content_embedding = await _get_content_embeddings(
         content=content, metadata=metadata
     )
+    latest_display_number = await get_latest_display_number(
+        asession=asession, workspace_id=workspace_id
+    )
     content_db = ContentDB(
         content_embedding=content_embedding,
         content_metadata=content.content_metadata,
         content_tags=content.content_tags,
         content_text=content.content_text,
         content_title=content.content_title,
+        display_number=latest_display_number + 1,
         created_datetime_utc=datetime.now(timezone.utc),
         updated_datetime_utc=datetime.now(timezone.utc),
         workspace_id=workspace_id,
@@ -563,3 +567,32 @@ async def update_votes_in_db(
     content_db = await asession.merge(content_db)
     await asession.commit()
     return content_db
+
+
+async def get_latest_display_number(
+    *, asession: AsyncSession, workspace_id: int
+) -> int:
+    """Get the latest display number from the database.
+
+    Parameters
+    ----------
+    asession
+        The SQLAlchemy async session to use for all database connections.
+    workspace_id
+        The ID of the workspace to get the latest display number from.
+
+    Returns
+    -------
+    int
+        The latest display number if it exists, otherwise 0.
+    """
+
+    stmt = (
+        select(ContentDB.display_number)
+        .where(ContentDB.workspace_id == workspace_id)
+        .order_by(ContentDB.display_number.desc())
+        .limit(1)
+    )
+    result = await asession.execute(stmt)
+    latest_display_number = result.scalar_one_or_none()
+    return latest_display_number or 0