testing

App/Transcription/TranscriptionRoutes.py

@@ -12,12 +12,12 @@ import aiofiles, os, re
 import uuid
 import tempfile
 from celery.result import AsyncResult
-from App.Worker import transcription_task, downloadfile
+from App.Worker import transcription_task, downloadfile,downloadUrl
 from App.Users.Model import User
 from App.Users.UserRoutes import get_token_owner
 from App.Users.Schemas import UserSchema
 from .Model import Transcriptions
-from .Utils.fastapi_tasks import perform_background_task,download_with_wget
+from .Utils.fastapi_tasks import perform_background_task
 import yt_dlp
 from fastapi_jwt_auth import AuthJWT
 
@@ -72,7 +72,7 @@ async def download_audio(
         r"(?u)[^-\w.]", "", short_uuid
     )  # Ensure the title is file-friendly
     filename = f"{sanitized_title}.mp3"
-    file_path = os.path.join("/tmp/", "Downloads", filename)
+    file_path = os.path.join("./", "Downloads", filename)
 
     ydl_opts = {
         "format": "bestaudio/best",
@@ -135,10 +135,9 @@ async def create_file(
 ):
     extension = 'wav'
     file_name = f"{genUUID()}.{extension}"
-    file_path = os.path.join("./", "Downloads", file_name)
-    download_with_wget(link=url,download_dir='./Downloads',filename=file_name)
+    file_dir = os.path.join("./", "Downloads")
     # celery task
-    task = transcription_task.delay(file_path, model)
+    task = downloadUrl.delay(link=url, download_dir=file_dir, filename=file_name, model_size=model)
 
     # create a transcription entry
     transcription_enrty = await Transcriptions.objects.create(

App/Worker.py

@@ -1,6 +1,6 @@
 from celery import Celery, chain
 import os
-import time
+import time,subprocess
 import cgi
 from App import celery_config
 import yt_dlp
@@ -22,6 +22,19 @@ def generate_store(self, data, task_id):
     encode(chunks)
     print("hellooo")
 
+def download_with_wget(link, download_dir, filename):
+    subprocess.run(["aria2c", link, "-d", download_dir, "-o", filename])
+
+@celery.task(name="download", bind=True)
+def downloadUrl(self, link, download_dir, filename, model_size="base"):
+    file_path=os.path.join(download_dir,filename)
+    download_with_wget(link=link,download_dir=download_dir,filename=filename)
+
+    data = transcribe_file(state=self, file_path=file_path, model_size=model_size)
+    generate_store.delay(data["content"], self.request.id)
+    return data
+
+
 
 @celery.task(name="transcription", bind=True)
 def transcription_task(self, file_path, model_size="tiny"):