Update diarization.py

diarization.py

@@ -73,24 +73,28 @@ def diarize_audio(audio_path: str, pipeline, num_speakers: int = None) -> List[T
         else:
             diarization = pipeline(audio_path)
         
-        # Extract segments - handle both old and new pyannote API
+        # Debug: print what we got
+        print(f"🔍 Diarization result type: {type(diarization)}")
+        
+        # Extract segments from Annotation object
         segments = []
         
-        # Try new API first (pyannote 3.x)
-        if hasattr(diarization, 'itertracks'):
-            for turn, _, speaker in diarization.itertracks(yield_label=True):
-                segments.append((turn.start, turn.end, speaker))
-        # Fallback for different output format
-        elif hasattr(diarization, 'get_timeline'):
-            for segment, _, speaker in diarization.get_timeline().itertracks(yield_label=True):
+        # The result should be an Annotation object with itertracks method
+        # Access it properly - the pipeline returns an Annotation directly
+        try:
+            for segment, track, speaker in diarization.itertracks(yield_label=True):
                 segments.append((segment.start, segment.end, speaker))
-        # Direct iteration if it's a list-like object
-        else:
-            # Try to iterate over the diarization result
-            for item in diarization:
-                if hasattr(item, 'start') and hasattr(item, 'end'):
-                    speaker = getattr(item, 'speaker', 'SPEAKER_00')
-                    segments.append((item.start, item.end, speaker))
+        except AttributeError:
+            # If itertracks doesn't exist, try accessing _tracks or similar
+            print(f"🔍 Available attributes: {dir(diarization)}")
+            
+            # Try to access the annotation if wrapped
+            if hasattr(diarization, 'annotation'):
+                for segment, track, speaker in diarization.annotation.itertracks(yield_label=True):
+                    segments.append((segment.start, segment.end, speaker))
+            elif hasattr(diarization, '_annotation'):
+                for segment, track, speaker in diarization._annotation.itertracks(yield_label=True):
+                    segments.append((segment.start, segment.end, speaker))
         
         print(f"✅ Diarization tamamlandı: {len(segments)} segment bulundu")
         return segments