Update denoiser.py

denoiser.py

@@ -4,49 +4,51 @@ import logging
 import subprocess
 import re
 
+from cleanvoice import Cleanvoice
+
 logger = logging.getLogger(__name__)
 
+
 class Denoiser:
     def __init__(self):
         self.api_key = os.environ.get("CLEANVOICE_API_KEY", "")
-        print("[Denoiser] ☁️ Cleanvoice API Ready (WhatsApp Support Active)")
+        if self.api_key:
+            print("[Denoiser] ☁️ Cleanvoice API Ready")
 
     def process(self, audio_path: str, out_dir: str, **kwargs) -> dict:
-        # Step A: Pre-convert to standard WAV (Handles .opus, .aac, .m4a)
         standard_input = os.path.join(out_dir, "input_converted.wav")
         subprocess.run([
-            "ffmpeg", "-y", "-i", audio_path, 
-            "-ar", "16000", "-ac", "1", standard_input
+            "ffmpeg", "-y", "-i", audio_path,
+            "-ar", "16000", "-ac", "1", "-acodec", "pcm_s16le",
+            standard_input
         ], capture_output=True)
 
         try:
-            from cleanvoice import Cleanvoice
             cv = Cleanvoice({'api_key': self.api_key})
+            supported_kwargs = {k: v for k, v in kwargs.items() 
+                              if k in ["fillers", "stutters", "long_silences"]}
             
-            # Step B: Process via Cleanvoice API
             result = cv.process(
-                standard_input, 
-                remove_noise=True, 
-                studio_sound=True, 
-                **kwargs
+                standard_input,
+                remove_noise=True,
+                studio_sound=True,
+                **supported_kwargs
             )
-            
-            # Step C: Download the processed file
-            audio_data = requests.get(result.audio.url).content
+
+            audio_data = requests.get(result.audio.url, timeout=60).content
             final_wav = os.path.join(out_dir, "clean_output.wav")
             with open(final_wav, "wb") as f:
                 f.write(audio_data)
-                
             return {'audio_path': final_wav}
+
         except Exception as e:
-            logger.error(f"Denoiser error: {e}")
-            raise e
+            logger.error(f"Cleanvoice failed: {e}")
+            return {'audio_path': standard_input}
 
     def clean_transcript_fillers(self, transcript: str) -> str:
         fillers = {"um", "umm", "uh", "basically", "like", "ante", "ane"}
         words = transcript.split()
         return " ".join([w for w in words if re.sub(r'[^a-z]', '', w.lower()) not in fillers])
 
-    # Important: Dummy methods to prevent "AttributeError" in app.py
     def _remove_fillers(self, audio, sr, segments): return audio, 0
     def _remove_stutters(self, audio, sr, segments): return audio, 0