Update app.py

app.py

@@ -62,26 +62,44 @@ if uploaded_video is not None:
     with tempfile.NamedTemporaryFile(delete=False) as tmp_video:
         tmp_video.write(uploaded_video.read())
         tmp_video_path = tmp_video.name
-    
-    # Convert video to audio
-    st.write("Converting video to audio...")
-    audio_file = video_to_audio(tmp_video_path)
-    
-    # Convert the extracted MP3 audio to WAV
-    st.write("Converting audio to WAV...")
-    wav_audio_file = convert_mp3_to_wav(audio_file)
-    
-    # Provide the audio file to the user for download
-    st.audio(wav_audio_file, format='audio/wav')
-    
-    # Transcribe audio to text
-    st.write("Transcribing audio to text...")
-    transcription = transcribe_audio(wav_audio_file)
-    
-    # Show the transcription
-    st.text_area("Transcription", transcription, height=300)
-    
-    # Cleanup temporary files
-    os.remove(tmp_video_path)
-    os.remove(audio_file)
-    os.remove(wav_audio_file)
+
+    # Add an "Analyze Video" button
+    if st.button("Analyze Video"):
+        with st.spinner("Processing video... Please wait."):
+            # Convert video to audio
+            audio_file = video_to_audio(tmp_video_path)
+            
+            # Convert the extracted MP3 audio to WAV
+            wav_audio_file = convert_mp3_to_wav(audio_file)
+            
+            # Transcribe audio to text
+            transcription = transcribe_audio(wav_audio_file)
+
+            # 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"
+                )
+
+            # Cleanup temporary files
+            os.remove(tmp_video_path)
+            os.remove(audio_file)
+            os.remove(wav_audio_file)