Update app.py

app.py

@@ -50,20 +50,26 @@ class GenerationTimeout(Exception):
 def timeout_handler(signum, frame):
     raise GenerationTimeout()
 
+MAX_RUNTIME = 600  # 10 minutes (CPU-safe)
+
 def worker():
     while True:
         job_id = job_queue.get()
         job = jobs.get(job_id)
 
-        try:
-            jobs[job_id]["status"] = JobStatus.PROCESSING
+        if not job:
+            job_queue.task_done()
+            continue
 
-            signal.signal(signal.SIGALRM, timeout_handler)
-            signal.alarm(420)  # 7 minutes max
+        start_time = time.time()
+        jobs[job_id]["status"] = JobStatus.PROCESSING
 
+        try:
             wav = model.generate([job["prompt"]])[0]
 
-            signal.alarm(0)
+            elapsed = time.time() - start_time
+            if elapsed > MAX_RUNTIME:
+                raise TimeoutError("Generation exceeded max runtime")
 
             filename = f"{job_id}.wav"
             path = os.path.join(OUTPUT_DIR, filename)
@@ -81,12 +87,6 @@ def worker():
                 "file_path": path
             })
 
-        except GenerationTimeout:
-            jobs[job_id].update({
-                "status": JobStatus.FAILED,
-                "error": "Generation timed out (CPU limit)"
-            })
-
         except Exception as e:
             jobs[job_id].update({
                 "status": JobStatus.FAILED,