upload to telegram

App/Transcription/TranscriptionRoutes.py

@@ -9,6 +9,7 @@ from .Schemas import (
 )
 from App import bot
 import aiofiles, os, re
+import uuid
 import tempfile
 from celery.result import AsyncResult
 from App.Worker import transcription_task, downloadfile
@@ -27,6 +28,10 @@ from App.Embedding.utils.Initialize import delete_documents
 
 transcription_router = APIRouter(tags=["Transcription"])
 
+def genUUID():
+    uuid_value = uuid.uuid4()
+    short_uuid = str(uuid_value)[:6]
+    return short_uuid
 
 @transcription_router.get("/download-audio")
 async def download_audio(
@@ -38,14 +43,17 @@ async def download_audio(
     ),
     user: UserSchema = Depends(get_token_owner),
 ):
-    youtube_url= url
+
+    youtube_url = url
     parsed_url = urlparse(youtube_url)
 
     # Get the query parameters
     query_parameters = parse_qs(parsed_url.query)
 
     # Get the value of the 'v' parameter
-    v_param_value = query_parameters.get('v', [])[0] if 'v' in query_parameters else None
+    v_param_value = (
+        query_parameters.get("v", [])[0] if "v" in query_parameters else None
+    )
     url = f"https://www.youtube.com/watch?v={v_param_value}"
     if user == None:
         return {"code": 400, "message": "doesn't exist", "payload": None}
@@ -57,9 +65,9 @@ async def download_audio(
     with yt_dlp.YoutubeDL(ydl_opts_info) as ydl:
         info_dict = ydl.extract_info(url, download=False)
         video_title = info_dict.get("title", None)
-
+    short_uuid = genUUID()
     sanitized_title = re.sub(
-        r"(?u)[^-\w.]", "", video_title
+        r"(?u)[^-\w.]", "", short_uuid
     )  # Ensure the title is file-friendly
     filename = f"{sanitized_title}.mp3"
     file_path = os.path.join("./", "Downloads", filename)
@@ -70,7 +78,7 @@ async def download_audio(
     }
 
     task = downloadfile.delay(url=url, ydl_opts=ydl_opts, model_size=model)
-    response = {"task_id": task.id, "file_name": f"{video_title}.mp3"}
+    response = {"task_id": task.id, "file_name": video_title }
     transcription_enrty = await Transcriptions.objects.create(
         user=user, youtubeLink=url, **response
     )
@@ -124,10 +132,12 @@ async def create_file(
     ),
     user: UserSchema = Depends(get_token_owner),
 ):
+    extension = file.filename.split('.')[-1]
+    file_name = f'{genUUID()}.{extension}'
     # Write the file to disk asynchronously
     Upload_dir = ""
     try:
-        async with aiofiles.open(file.filename, "wb") as f:
+        async with aiofiles.open(file_name, "wb") as f:
             while contents := await file.read(1024 * 1):
                 await f.write(contents)
 
@@ -139,11 +149,11 @@ async def create_file(
         await file.close()
 
     # celery task
-    task = transcription_task.delay(file.filename, model)
+    task = transcription_task.delay(file_name, model)
 
     # create a transcription entry
     transcription_enrty = await Transcriptions.objects.create(
-        task_id=task.id, user=user, file_name=file.filename
+        task_id=task.id, user=user, file_name=file_name
     )
     background_tasks.add_task(perform_background_task, file=file, task_id=task.id)
     return {
@@ -162,7 +172,7 @@ async def get_status(task_id):
     result = BaseTranscription(**entry.__dict__)
 
     if result.status == "SUCCESS":
-        result.percentage='100'
+        result.percentage = "100"
         return result
     task_result = AsyncResult(task_id)
     # print(task_result.result)
@@ -175,17 +185,17 @@ async def get_status(task_id):
 
     if task_result.status == "SUCCESS":
         trans = TranscriptionMetadata(**task_result.result)
-        percentage='100'
+        percentage = "100"
         await entry.update(**trans.dict())
     else:
         try:
             _trans = TranscriptionMetadata(**task_result.result)
-            percentage=_trans.percentage
+            percentage = _trans.percentage
         except Exception as e:
             print(e)
             return {"payload": None, "message": "Nothing found", "code": 200}
         await entry.update(**_trans.dict())
 
     result = BaseTranscription(**entry.__dict__)
-    result.percentage=percentage
-    return result
+    result.percentage = percentage
+    return result

App/app.py

@@ -47,7 +47,7 @@ def authjwt_exception_handler(request: Request, exc: AuthJWTException):
 
 @app.on_event("startup")
 async def startup_event():
-    # await bot.start(bot_token="6183919505:AAEhHFt4mI18bQeAf2Lj7AePXFRPVLrOFM8")
+    await bot.start(bot_token="6183919505:AAEhHFt4mI18bQeAf2Lj7AePXFRPVLrOFM8")
     # await upload_bot.start()
     # await models.create_all()
     # models.metadata.create_all()