Update functions.py

functions.py

@@ -225,7 +225,7 @@ def get_spacy():
 @st.experimental_memo(suppress_st_warning=True)
 def inference(link, upload, _asr_model):
     '''Convert Youtube video or Audio upload to text'''
-
+    
     try:
         
         if validators.url(link):
@@ -235,6 +235,9 @@ def inference(link, upload, _asr_model):
     
             #Get audio file from YT
             audio_file = yt.streams.filter(only_audio=True)[0].download(filename="audio.mp4")
+
+            if 'audio' not in st.session_state:
+                st.session_state['audio'] = audio_file
     
             #Get size of audio file
             audio_size = round(os.path.getsize(audio_file)/(1024*1024),1)
@@ -299,11 +302,12 @@ def inference(link, upload, _asr_model):
 
             return results, "Transcribed Earnings Audio"
                 
-    except:
+    except Exception as e:
 
-        st.write('Whisper API Error, using Whisper module from GitHub, might take longer than expected')
+        st.write(f'''Whisper API Error: {e}, 
+                    Using Whisper module from GitHub, might take longer than expected''')
         
-        results = _asr_model.transcribe(audio_file, task='transcribe', language='en')
+        results = _asr_model.transcribe(st.session_state['audio'], task='transcribe', language='en')
       
         return results['text'], yt.title