Update utils_hf.py

utils_hf.py

@@ -3,6 +3,7 @@ import streamlit as st
 import yt_dlp
 import numpy as np
 import soundfile as sf
+import pyannote.audio
 from moviepy.editor import AudioFileClip
 from transformers import pipeline
 from pyannote.audio.pipelines.utils.hook import ProgressHook
@@ -13,6 +14,7 @@ whisper_pipeline = pipeline(
     generate_kwargs={"language": "<|es|>", "task": "transcribe"}
 )
 
+
 def diarize_short_segment(audio_path, token, start_time=0, end_time=600):
     """
     Realiza la diarización solo en un segmento corto del archivo de audio.
@@ -44,7 +46,7 @@ def diarize_short_segment(audio_path, token, start_time=0, end_time=600):
             diarization_text += f"{segment.start:.2f} - {segment.end:.2f}: {speaker}\n"
             speakers.add(speaker)
 
-        return diarization_text, len(speakers)
+        return diarization_segments, diarization_text, speakers
     except Exception as e:
         st.error(f"Error al realizar la diarización: {e}")
         return None, 0