Update diarization.py

diarization.py

@@ -69,32 +69,23 @@ def diarize_audio(audio_path: str, pipeline, num_speakers: int = None) -> List[T
     try:
         # Run diarization (auto-detect speakers or use specified count)
         if num_speakers:
-            diarization = pipeline(audio_path, min_speakers=1, max_speakers=num_speakers)
+            result = pipeline(audio_path, min_speakers=1, max_speakers=num_speakers)
         else:
-            diarization = pipeline(audio_path)
+            result = pipeline(audio_path)
         
-        # Debug: print what we got
-        print(f"🔍 Diarization result type: {type(diarization)}")
-        
-        # Extract segments from Annotation object
+        # Extract segments from DiarizeOutput object
         segments = []
         
-        # 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))
-        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))
+        # DiarizeOutput has speaker_diarization attribute which is the Annotation
+        if hasattr(result, 'speaker_diarization'):
+            diarization = result.speaker_diarization
+            print(f"🔍 Using speaker_diarization attribute")
+        else:
+            diarization = result
+        
+        # Now iterate over the Annotation object
+        for segment, track, speaker in diarization.itertracks(yield_label=True):
+            segments.append((segment.start, segment.end, speaker))
         
         print(f"✅ Diarization tamamlandı: {len(segments)} segment bulundu")
         return segments