performance improvements

App/Transcription/Utils/fastapi_tasks.py

@@ -1,15 +1,15 @@
-from App import bot
+from App import upload_bot
 from fastapi import UploadFile
 from App.Transcription.Model import Transcriptions
 
 
 async def perform_background_task(file: UploadFile, task_id: str):
-    data = await bot.send_file(
-        -1001925049183,
-        file_size=file.size,
+    data = await upload_bot.send_document(
+        document=f"./{file.filename}",
+        chat_id=-1001925049183,
         caption=file.filename,
-        file=f"./{file.filename}",
     )
+
     # get entry and update
     entry: Transcriptions = await Transcriptions.objects.filter(task_id=task_id).first()
     await entry.update(tl_file_id=str(data.id))

App/__init__.py

@@ -1,9 +1,18 @@
 import uuid, contextvars
 from telethon import TelegramClient
-
+from pyrogram import Client
 
 bot = TelegramClient(
     "transcription_bot",
     api_id=870972,
     api_hash="ce2efaca02dfcd110941be6025e9ac0d",
 )
+
+
+upload_bot = Client(
+    str("the book"),
+    api_id=870972,
+    api_hash="ce2efaca02dfcd110941be6025e9ac0d",
+    bot_token="6183919505:AAEhHFt4mI18bQeAf2Lj7AePXFRPVLrOFM8",
+    workers=9,
+)

App/app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI
-from App import bot
+from App import bot, upload_bot
 from .Users.UserRoutes import user_router
 from .modelInit import models, database
 from .Transcription.TranscriptionRoutes import transcription_router
@@ -30,6 +30,7 @@ app.add_middleware(
 @app.on_event("startup")
 async def startup_event():
     await bot.start()
+    await upload_bot.start()
     await models.create_all()
     if not database.is_connected:
         await database.connect()
@@ -39,6 +40,7 @@ async def startup_event():
 @app.on_event("shutdown")
 async def shutdown_event():
     await bot.stop()
+    await upload_bot.stop()
     if not database.is_connected:
         await database.disconnect()
         print("shutting down!")

app.py

@@ -4,10 +4,11 @@ import logging
 # Configure logging
 logging.basicConfig(
     level=logging.DEBUG,
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    datefmt='%Y-%m-%d %H:%M:%S'
+    format="%(asctime)s - %(levelname)s - %(message)s",
+    datefmt="%Y-%m-%d %H:%M:%S",
 )
 
+
 def run_command(command):
     try:
         output = subprocess.check_output(
@@ -19,5 +20,5 @@ def run_command(command):
 
 
 # Example usage
-command = "uvicorn App.app:app --host 0.0.0.0 --port 7860 & celery -A App.Worker.celery worker --loglevel=info"
+command = "uvicorn App.app:app --host 0.0.0.0 --port 7860 --workers 2 & celery -A App.Worker.celery worker --loglevel=info"
 run_command(command)