Update functions.py

functions.py

@@ -31,15 +31,15 @@ def load_whisper_api(audio):
     return transcript
 
 @st.cache_data
-def get_transcribe_podcast(rss_url, local_path='podcast/'):
+def get_transcribe_podcast(rss_url, local_path='/data/'):
     
     st.info("Starting Podcast Transcription Function...")
     print("Feed URL: ", rss_url)
-    # print("Local Path:", local_path)  
+    print("Local Path:", local_path)  
 
     
     # Download the podcast episode by parsing the RSS feed
-    # p = Path(local_path)
+    p = Path(local_path)
     # p.mkdir(exist_ok=True)
 
     st.info("Downloading the podcast episode...")
@@ -48,10 +48,10 @@ def get_transcribe_podcast(rss_url, local_path='podcast/'):
   
     with requests.get(rss_url, stream=True) as r:
         r.raise_for_status()
-        # episode_path = p.joinpath(episode_name)
-        # print(f'episode path {episode_path}')
+        episode_path = p.joinpath(episode_name)
+        print(f'episode path {episode_path}')
         
-    with open(episode_name, 'wb') as f:
+    with open(episode_path, 'wb') as f:
         for chunk in r.iter_content(chunk_size=8192):
             f.write(chunk.getbuffer())
 
@@ -60,7 +60,7 @@ def get_transcribe_podcast(rss_url, local_path='podcast/'):
     # Perform the transcription
     st.info("Starting podcast transcription")
     
-    audio_file = episode_name
+    audio_file = episode_path
     
     
     #Get size of audio file
@@ -89,7 +89,7 @@ def get_transcribe_podcast(rss_url, local_path='podcast/'):
 
         for i, chunk in enumerate(chunks):
             chunk.export(f'chunk_{i}.mp3', format='mp3')
-            transcriptions.append(load_whisper_api(f'chunk_{i}.mp3')['text'])
+            transcriptions.append(load_whisper_api(f'chunk_{i}.mp3'))
 
         results = ','.join(transcriptions)
 
@@ -192,7 +192,7 @@ def get_podcast_highlights(podcast_transcript):
     return podcastHighlights
 
 @st.cache_data
-def process_podcast(url, path='podcast/'):
+def process_podcast(url, path='/data/'):
 
     '''Get podcast transcription into json'''