Update app.py

app.py

@@ -273,18 +273,38 @@ def process_audio(model_choice, mode, voice_print_path, audio_path, speaker_name
         if diarization_pipeline is None:
             return "Pyannote diarization is not available.", None, None
 
+
         diarization = diarization_pipeline({"waveform": waveform, "sample_rate": sample_rate})
-        data = []
 
-        # Handle both pyannote APIs
-        if hasattr(diarization, "itertracks"):
-            for turn, _, speaker in diarization.itertracks(yield_label=True):
-                data.append({"start": turn.start, "end": turn.end, "speaker": speaker})
+        # Run diarization - pass audio file path directly for better compatibility
+        #diarization = diarization_pipeline(audio_path)
+
+        # Correct API for pyannote 3.1+ with DiarizeOutput
+        data = []
+        
+        # Check if we have the new API (DiarizeOutput with speaker_diarization attribute)
+        if hasattr(diarization, 'speaker_diarization'):
+            # New API (pyannote 3.1+) - iterate over speaker_diarization
+            for turn, speaker in diarization.speaker_diarization:
+                data.append({
+                    "start": turn.start,
+                    "end": turn.end,
+                    "speaker": speaker
+                })
+        elif hasattr(diarization, 'itertracks'):
+            # Old API (pyannote < 3.1) - Annotation object
+            for segment, track, speaker in diarization.itertracks(yield_label=True):
+                data.append({
+                    "start": segment.start,
+                    "end": segment.end,
+                    "speaker": speaker
+                })
         else:
-            for segment in diarization.iter_segments():
-                speaker = diarization.label(segment)
-                data.append({"start": segment.start, "end": segment.end, "speaker": speaker})
+            return "Unsupported pyannote.audio version. Please check the diarization output format.", None, None
 
+        if not data:
+            return "No speaker segments found in diarization.", None, None
+                
         diarization_df = pd.DataFrame(data)
 
         # Identify target speaker