Create worker.py

worker.py

@@ -0,0 +1,32 @@
+from threading import Thread
+from queue import Queue
+from generate import generate_music
+from cleanup import cleanup
+
+job_queue = Queue()
+jobs = {}
+
+
+def worker():
+
+    while True:
+        job_id, prompt, duration = job_queue.get()
+
+        try:
+            jobs[job_id]["status"] = "processing"
+
+            file_path = generate_music(prompt, duration)
+
+            jobs[job_id]["status"] = "done"
+            jobs[job_id]["file"] = file_path
+
+            cleanup()
+
+        except Exception as e:
+            jobs[job_id]["status"] = "error"
+            jobs[job_id]["error"] = str(e)
+
+        job_queue.task_done()
+
+
+Thread(target=worker, daemon=True).start()