19 lines
425 B
Python
19 lines
425 B
Python
import redis, json, os
|
|
|
|
REDIS_HOST=os.environ.get("REDIS_HOST","redis")
|
|
r=redis.Redis(host=REDIS_HOST, port=6379, db=0)
|
|
|
|
QUEUE="tts_queue"
|
|
RESULT="tts_result"
|
|
|
|
def fetch_job():
|
|
data=r.rpop(QUEUE)
|
|
if not data:
|
|
return None
|
|
return json.loads(data)
|
|
|
|
def store_result(job_id:str, audio:bytes, mime:str):
|
|
key=f"{RESULT}:{job_id}"
|
|
obj={"audio": audio.hex(), "mime": mime}
|
|
r.set(key, json.dumps(obj))
|