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: