added choices and stuff

App/Transcription/TranscriptionRoutes.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, status, Form, UploadFile, File
+from fastapi import APIRouter, status, Form, UploadFile, File, Query
 from typing_extensions import Annotated
 from .Schemas import UserDetails
 import aiofiles
@@ -13,7 +13,13 @@ transcription_router = APIRouter(tags=["User"])
 
 
 @transcription_router.post("/uploadfile/")
-async def create_file(file: UploadFile, userId: int = 1):
+async def create_file(
+    file: UploadFile,
+    userId: int = 1,
+    model: str = Query(
+        "tiny", choices=("tiny", "small", "medium"), description="Model Sizes"
+    ),
+):
     # Read the file contents
     contents = await file.read()
 
@@ -22,7 +28,7 @@ async def create_file(file: UploadFile, userId: int = 1):
         await f.write(contents)
 
     # celery task
-    task = transcription_task.delay(file.filename)
+    task = transcription_task.delay(file.filename, model)
 
     return {"file_size": file.size, "file_name": file.filename, "task_id": task.id}
 

App/Worker.py

@@ -9,5 +9,5 @@ celery.config_from_object(celery_config)
 
 
 @celery.task(name="create_task", bind=True)
-def transcription_task(self, file_path):
-    return transcribe_file(state=self, file_path=file_path, model_size="tiny")
+def transcription_task(self, file_path, model_size="tiny"):
+    return transcribe_file(state=self, file_path=file_path, model_size=model_size)