Update app.py

app.py

@@ -78,28 +78,33 @@ if uploaded_video is not None:
             # Show the transcription
             st.text_area("Transcription", transcription, height=300)
 
-            # Provide the audio file to the user for download
-            st.audio(wav_audio_file, format="audio/wav")
-
-            # Add download buttons for the transcription and audio
-            # Downloadable transcription file
-            st.download_button(
-                label="Download Transcription",
-                data=transcription,
-                file_name="transcription.txt",
-                mime="text/plain"
-            )
-            
-            # Downloadable audio file
-            with open(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"
-                )
+            # Store transcription and audio file in session state
+            st.session_state.transcription = transcription
+            st.session_state.wav_audio_file = wav_audio_file
 
             # Cleanup temporary files
             os.remove(tmp_video_path)
             os.remove(audio_file)
-            os.remove(wav_audio_file)
+
+# 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
+    st.audio(st.session_state.wav_audio_file, format='audio/wav')
+    
+    # Add download buttons for the transcription and audio
+    # Downloadable transcription file
+    st.download_button(
+        label="Download Transcription",
+        data=st.session_state.transcription,
+        file_name="transcription.txt",
+        mime="text/plain"
+    )
+    
+    # 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"
+        )