Making public version

backend/src/models/sign.py

@@ -1,7 +1,9 @@
 from typing import List
 
+import numpy as np
 import requests
 from pydantic import BaseModel
+from sqlalchemy.ext.asyncio import AsyncSession
 from sqlmodel import Field, Relationship, SQLModel
 
 from src.exceptions.base_exception import BaseException
@@ -51,6 +53,26 @@ class Sign(SQLModelExtended, table=True):
         if self.video_url is None:
             raise BaseException(404, "Video url not found")
 
+    @classmethod
+    async def get_random(self, session: AsyncSession):
+        signs = await self.get_all(select_in_load=[Sign.sign_videos],session=session)
+
+        sign_videos = [len(s.sign_videos) for s in signs]
+
+        # get probability based on number of videos, lower must be more likely
+        # the sum must be 1
+
+        random_prob = [1 / (x + 1) for x in sign_videos]
+        random_prob = random_prob / np.sum(random_prob)
+
+        # get random sign
+        sign = np.random.choice(signs, p=random_prob)
+
+        return sign
+
+
+
+
 class SignOut(BaseModel):
     id: int
     url: str
@@ -59,3 +81,10 @@ class SignOut(BaseModel):
     video_url: str
 
     sign_videos: List[SignVideo] = []
+
+class SignOutSimple(BaseModel):
+    id: int
+    url: str
+    name: str
+    sign_id: str
+    video_url: str