Update main.py

main.py

@@ -18,6 +18,7 @@ import json
 import time
 import tempfile
 import logging
+import shutil
 import requests
 import cloudinary
 import cloudinary.uploader
@@ -53,6 +54,17 @@ app.add_middleware(
 )
 
 
+def _safe_cleanup(out_dir):
+    """Remove temp directory safely — never raises."""
+    try:
+        if hasattr(denoiser, 'cleanup_temp_files'):
+            denoiser.cleanup_temp_files(out_dir)
+        elif out_dir and os.path.exists(out_dir):
+            shutil.rmtree(out_dir, ignore_errors=True)
+    except Exception as e:
+        logger.warning(f"Cleanup warning (non-fatal): {e}")
+
+
 def run_pipeline(audio_path, src_lang="auto", tgt_lang="te",
                  opt_fillers=True, opt_stutters=True, opt_silences=True):
     """
@@ -72,7 +84,6 @@ def run_pipeline(audio_path, src_lang="auto", tgt_lang="te",
             fillers=opt_fillers,
             stutters=opt_stutters,
             long_silences=opt_silences,
-            # ✅ breaths/mouth_sounds removed — handled by studio_sound=True
         )
         clean_audio = result["audio_path"]
         stats = {
@@ -116,7 +127,7 @@ def run_pipeline(audio_path, src_lang="auto", tgt_lang="te",
         try:
             upload_result = cloudinary.uploader.upload(
                 clean_audio,
-                resource_type="video",  # audio files use "video"
+                resource_type="video",
                 folder="clearwave_enhanced",
             )
             enhanced_url = upload_result["secure_url"]
@@ -124,6 +135,7 @@ def run_pipeline(audio_path, src_lang="auto", tgt_lang="te",
         except Exception as e:
             logger.error(f"Cloudinary failed: {e}")
 
+        # ✅ yield done INSIDE try so cleanup never interrupts it
         yield {
             "status": "done",
             "step": 4,
@@ -146,8 +158,9 @@ def run_pipeline(audio_path, src_lang="auto", tgt_lang="te",
         logger.error(f"Pipeline error: {e}", exc_info=True)
         yield {"status": "error", "message": f"Pipeline failed: {str(e)}"}
 
-    # Cleanup
-    denoiser.cleanup_temp_files(out_dir)
+    finally:
+        # ✅ Cleanup is now in finally — runs after done is yielded, never crashes pipeline
+        _safe_cleanup(out_dir)
 
 
 @app.get("/api/health")
@@ -180,14 +193,13 @@ async def process_url(request: Request):
             resp = requests.get(audio_url, timeout=60, stream=True)
             resp.raise_for_status()
 
-            # Detect format
             lower_url = audio_url.lower().split("?")[0]
             suffix = ".mp3"
-            if ".opus" in lower_url: suffix = ".opus"
-            elif ".ogg" in lower_url: suffix = ".ogg"
-            elif ".aac" in lower_url: suffix = ".aac"
-            elif ".m4a" in lower_url: suffix = ".m4a"
-            elif ".wav" in lower_url: suffix = ".wav"
+            if ".opus" in lower_url:   suffix = ".opus"
+            elif ".ogg" in lower_url:  suffix = ".ogg"
+            elif ".aac" in lower_url:  suffix = ".aac"
+            elif ".m4a" in lower_url:  suffix = ".m4a"
+            elif ".wav" in lower_url:  suffix = ".wav"
 
             tmp = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
             for chunk in resp.iter_content(chunk_size=65536):
@@ -209,7 +221,7 @@ async def process_url(request: Request):
 
         try:
             os.unlink(tmp.name)
-        except:
+        except Exception:
             pass
 
     return StreamingResponse(