concurrent

app.py

@@ -12,8 +12,7 @@ import os
 import requests
 from modules.audio import convert, get_audio_duration
 from modules.r2 import upload_to_s3
-from wakaq import Queue, Worker
-import threading
+from concurrent.futures import ThreadPoolExecutor
 
 vpv_webhook = os.environ.get("VPV_WEBHOOK")
 
@@ -71,7 +70,8 @@ class ProcessRequest(BaseModel):
     format: str = "wav"
     speed: float = 0.8
     crossfade: float = 0.1
-
+    
+executor = ThreadPoolExecutor(max_workers=8)
 
 def process_queue(item):
     key, censor, offset, text, format, speed, crossfade, id, receiver, webhook = item
@@ -92,16 +92,7 @@ def process_queue(item):
     }
             
     requests.post(webhook, json=payload)
-
-queue = Queue()
-worker = Worker(queue, process_queue, concurrency=8)
-def start_worker():
-    worker.run()
     
-thread = threading.Thread(target=start_worker)
-thread.daemon = True 
-thread.start()
-
 @app.post("/process")
 def process_audio(payload: ProcessRequest):
     key = payload.key
@@ -119,7 +110,7 @@ def process_audio(payload: ProcessRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
     try:
-        q.put((key, censor, offset, text, format, speed, crossfade, id, receiver, webhook))
+        executor.submit(process_queue, (key, censor, offset, text, format, speed, crossfade, id, receiver, webhook))
         return {"success": True, "err": ""}
         
     except ValueError as e: