debugging pyannote

app/services/asr.py

@@ -58,13 +58,13 @@ def transcribe(audio_path: str) -> List[Dict]:
     try:
         diarization = diarization_pipeline(audio_path)
 
-        # --- DEBUGGING LINES ADDED HERE ---
+        # --- DEBUGGING LINES (optional) ---
         print("\n--- Diarization Output ---")
         print(f"Type of diarization object: {type(diarization)}")
         print("Diarization object content:")
         print(diarization)
         print("--- End Diarization Output ---\n")
-        # ------------------------------------
+        # ----------------------------------
 
     except Exception as e:
         print(f"Error during diarization: {e}")
@@ -74,11 +74,14 @@ def transcribe(audio_path: str) -> List[Dict]:
 
     # 3. Assign Speaker to Segments
     out_segments = []
+
+    # For pyannote 4.x, use diarization.speaker_diarization
+    annotation = diarization.speaker_diarization  # This is an Annotation object
+
     for seg in whisper_segments:
-        # Find the speaker for the segment's midpoint
         midpoint = seg["start"] + (seg["end"] - seg["start"]) / 2
         speaker = "UNKNOWN"
-        for turn, _, speaker_label in diarization.itertracks(yield_label=True):
+        for turn, _, speaker_label in annotation.itertracks(yield_label=True):
             if turn.start <= midpoint <= turn.end:
                 speaker = speaker_label
                 break
@@ -94,5 +97,4 @@ def transcribe(audio_path: str) -> List[Dict]:
 
     if not out_segments:
         return [{"start": 0.0, "end": 0.0, "speaker": "A", "text": ""}]
-
     return out_segments