Update functions.py

functions.py

@@ -114,12 +114,15 @@ def get_yt_audio(url):
  }
 
  with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+ 
+ info = ydl.extract_info(url, download=False)
+ title = info.get('title', None)
  ydl.download([url])
 
  #with open(temp_audio_file+'.mp3', 'rb') as file:
  audio_file = os.path.join('output', 'audio.mp3')
 
- return audio_file
+ return audio_file, title
 
 
 @st.cache_resource
@@ -289,14 +292,14 @@ def get_spacy():
  return nlp
  
 @st.cache_data
-def inference(link, upload, _asr_model):
+def inference(link, title, upload, _asr_model):
  '''Convert Youtube video or Audio upload to text'''
  
  try:
  
  if validators.url(link):
  
- audio_file = get_yt_audio(link)
+ audio_file, title = get_yt_audio(link)
  # title = yt.title
  
  if 'audio' not in st.session_state:
@@ -330,7 +333,7 @@ def inference(link, upload, _asr_model):
  
  results = ','.join(transcriptions)
 
- return results, None
+ return results, title
 
  elif upload: