Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import logging
+from starlette.background import BackgroundTask # <-- IMPORT THE FIX
 
 # --- FIX FOR ALL PERMISSION ERRORS ---
 # Set environment variables BEFORE importing torch or diffusers.
@@ -92,18 +93,22 @@ async def generate_audio_endpoint(request: AudioRequest):
             audio_int16 = (audio * 32767).astype(np.int16)
             write_wav(temp_file_path, sample_rate, audio_int16)
             logger.info(f"Audio saved to temporary file: {temp_file_path}")
+        
+        # ### THIS IS THE FIX ###
+        # Create a background task to delete the file AFTER the response is sent.
+        cleanup_task = BackgroundTask(os.remove, temp_file_path)
 
         return FileResponse(
             path=temp_file_path, 
             media_type='audio/wav', 
             filename=f"{prompt[:50].replace(' ', '_')}.wav",
-            background=os.remove(temp_file_path)
+            background=cleanup_task # Use the background task here
         )
 
     except Exception as e:
         logger.error(f"Error during audio generation for prompt '{prompt}': {e}", exc_info=True)
         if temp_file_path and os.path.exists(temp_file_path):
-            os.remove(temp_file_path)
+            os.remove(temp_file_path) # Clean up if something else goes wrong
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/")