Update app.py

app.py

@@ -71,22 +71,24 @@ def worker():
             if elapsed > MAX_RUNTIME:
                 raise TimeoutError("Generation exceeded max runtime")
 
-            filename = f"{job_id}.wav"
-            path = os.path.join(OUTPUT_DIR, filename)
+            base_path = os.path.join(OUTPUT_DIR, job_id)
 
             audio_write(
-                path,
+                base_path,
                 wav.cpu(),
                 model.sample_rate,
                 strategy="loudness",
                 loudness_compressor=True
             )
-
+            
+            final_path = base_path + ".wav"
+            
             jobs[job_id].update({
                 "status": JobStatus.COMPLETED,
-                "file_path": path
+                "file_path": final_path
             })
 
+
         except Exception as e:
             jobs[job_id].update({
                 "status": JobStatus.FAILED,