Update app.py

app.py

@@ -3,6 +3,7 @@ import tempfile
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.middleware.cors import CORSMiddleware
 from faster_whisper import WhisperModel
+from df.enhance import enhance, init_df, load_audio, save_audio
 from pipeline.postprocess import process
 
 app = FastAPI()
@@ -19,6 +20,10 @@ print("Loading Whisper model...", flush=True)
 model = WhisperModel("large-v3", device="cpu", compute_type="int8")
 print("Whisper model ready!")
 
+print("Loading DeepFilterNet model...", flush=True)
+df_model, df_state, _ = init_df()
+print("DeepFilterNet model ready!")
+
 @app.get("/health")
 async def health():
     return {"status": "ok", "message": "SubFlow backend is running"}
@@ -27,6 +32,7 @@ async def health():
 async def transcribe(websocket: WebSocket):
     await websocket.accept()
     tmp_path = None
+    enhanced_path = None
 
     try:
         print("Step 1: waiting for bytes")
@@ -39,11 +45,20 @@ async def transcribe(websocket: WebSocket):
             tmp_path = tmp.name
         print(f"Step 3: saved to {tmp_path}")
 
+        await websocket.send_json({"status": "cleaning"})
+        print("Step 3.5: cleaning audio with DeepFilterNet")
+
+        audio_data, _ = load_audio(tmp_path, sr=df_state.sr())
+        enhanced_audio = enhance(df_model, df_state, audio_data)
+        enhanced_path = tmp_path.replace(".wav", "_enhanced.wav")
+        save_audio(enhanced_path, enhanced_audio, df_state.sr())
+        print(f"Step 3.6: cleaned audio saved to {enhanced_path}")
+
         await websocket.send_json({"status": "transcribing"})
         print("Step 4: starting transcription")
 
         segments_raw, info = model.transcribe(
-            tmp_path,
+            enhanced_path,
             language="en",
             word_timestamps=True,
             vad_filter=True,
@@ -97,6 +112,8 @@ async def transcribe(websocket: WebSocket):
     finally:
         if tmp_path and os.path.exists(tmp_path):
             os.remove(tmp_path)
+        if enhanced_path and os.path.exists(enhanced_path):
+            os.remove(enhanced_path)
         try:
             await websocket.close()
         except Exception: