Update app.py

app.py

@@ -4,6 +4,7 @@ import speech_recognition as sr
 from pydub import AudioSegment
 import tempfile
 import os
+import io
 
 # Function to convert video to audio
 def video_to_audio(video_file):
@@ -84,7 +85,11 @@ if tab == "Video":
 
                 # Store transcription and audio file in session state
                 st.session_state.transcription = transcription
-                st.session_state.wav_audio_file = wav_audio_file
+                
+                # Store the audio file as a BytesIO object in memory
+                with open(wav_audio_file, "rb") as f:
+                    audio_data = f.read()
+                    st.session_state.wav_audio_file = io.BytesIO(audio_data)
 
                 # Cleanup temporary files
                 os.remove(tmp_video_path)
@@ -93,8 +98,7 @@ if tab == "Video":
     # Check if transcription and audio file are stored in session state
     if 'transcription' in st.session_state and 'wav_audio_file' in st.session_state:
         # Provide the audio file to the user for download
-        with open(st.session_state.wav_audio_file, "rb") as audio_file_data:
-            st.audio(audio_file_data.read(), format='audio/wav')
+        st.audio(st.session_state.wav_audio_file, format='audio/wav')
         
         # Add download buttons for the transcription and audio
         # Downloadable transcription file
@@ -106,13 +110,12 @@ if tab == "Video":
         )
         
         # Downloadable audio file
-        with open(st.session_state.wav_audio_file, "rb") as audio_file_data:
-            st.download_button(
-                label="Download Audio",
-                data=audio_file_data,
-                file_name="converted_audio.wav",
-                mime="audio/wav"
-            )
+        st.download_button(
+            label="Download Audio",
+            data=st.session_state.wav_audio_file,
+            file_name="converted_audio.wav",
+            mime="audio/wav"
+        )
 
 elif tab == "Audio":
     # File uploader for audio
@@ -141,7 +144,11 @@ elif tab == "Audio":
 
                 # Store transcription in session state
                 st.session_state.transcription_audio = transcription
-                st.session_state.wav_audio_file_audio = wav_audio_file
+                
+                # Store the audio file as a BytesIO object in memory
+                with open(wav_audio_file, "rb") as f:
+                    audio_data = f.read()
+                    st.session_state.wav_audio_file_audio = io.BytesIO(audio_data)
 
                 # Cleanup temporary audio file
                 os.remove(tmp_audio_path)
@@ -149,8 +156,7 @@ elif tab == "Audio":
         # Check if transcription and audio file are stored in session state
         if 'transcription_audio' in st.session_state and 'wav_audio_file_audio' in st.session_state:
             # Provide the audio file to the user for download
-            with open(st.session_state.wav_audio_file_audio, "rb") as audio_file_data:
-                st.audio(audio_file_data.read(), format='audio/wav')
+            st.audio(st.session_state.wav_audio_file_audio, format='audio/wav')
             
             # Add download buttons for the transcription and audio
             # Downloadable transcription file
@@ -162,10 +168,9 @@ elif tab == "Audio":
             )
             
             # Downloadable audio file
-            with open(st.session_state.wav_audio_file_audio, "rb") as audio_file_data:
-                st.download_button(
-                    label="Download Audio",
-                    data=audio_file_data,
-                    file_name="converted_audio_audio.wav",
-                    mime="audio/wav"
-                )
+            st.download_button(
+                label="Download Audio",
+                data=st.session_state.wav_audio_file_audio,
+                file_name="converted_audio_audio.wav",
+                mime="audio/wav"
+            )