Update app.py

app.py

@@ -51,60 +51,119 @@ def transcribe_audio(audio_file):
         return "Could not request results from Google Speech Recognition service."
 
 # Streamlit app layout
-st.title("Video to Audio to Text Transcription")
-st.write("Upload a video file, and it will be converted to audio and transcribed into text.")
-
-# File uploader for video
-uploaded_video = st.file_uploader("Upload Video", type=["mp4", "mov", "avi"])
-
-if uploaded_video is not None:
-    # Save the uploaded video file temporarily
-    with tempfile.NamedTemporaryFile(delete=False) as tmp_video:
-        tmp_video.write(uploaded_video.read())
-        tmp_video_path = tmp_video.name
-
-    # 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)
+st.title("Video and Audio to Text Transcription")
+st.write("Upload a video or audio file to convert it to transcription.")
 
-            # Show the transcription
-            st.text_area("Transcription", transcription, height=300)
+# Create tabs to separate video and audio uploads
+tab = st.selectbox("Select the type of file to upload", ["Video", "Audio"])
 
-            # Store transcription and audio file in session state
-            st.session_state.transcription = transcription
-            st.session_state.wav_audio_file = wav_audio_file
+if tab == "Video":
+    # File uploader for video
+    uploaded_video = st.file_uploader("Upload Video", type=["mp4", "mov", "avi"])
 
-            # Cleanup temporary files
-            os.remove(tmp_video_path)
-            os.remove(audio_file)
+    if uploaded_video is not None:
+        # Save the uploaded video file temporarily
+        with tempfile.NamedTemporaryFile(delete=False) as tmp_video:
+            tmp_video.write(uploaded_video.read())
+            tmp_video_path = tmp_video.name
 
-# 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:
+        # 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)
+
+                # 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)
+
+    # 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 Audio",
-            data=audio_file_data,
-            file_name="converted_audio.wav",
-            mime="audio/wav"
-        )
+            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"
+            )
+
+elif tab == "Audio":
+    # File uploader for audio
+    uploaded_audio = st.file_uploader("Upload Audio", type=["wav", "mp3"])
+
+    if uploaded_audio is not None:
+        # Save the uploaded audio file temporarily
+        with tempfile.NamedTemporaryFile(delete=False) as tmp_audio:
+            tmp_audio.write(uploaded_audio.read())
+            tmp_audio_path = tmp_audio.name
+
+        # Add an "Analyze Audio" button
+        if st.button("Analyze Audio"):
+            with st.spinner("Processing audio... Please wait."):
+                # Convert audio to WAV if it's in MP3 format
+                if uploaded_audio.type == "audio/mpeg":
+                    wav_audio_file = convert_mp3_to_wav(tmp_audio_path)
+                else:
+                    wav_audio_file = tmp_audio_path
+                
+                # Transcribe audio to text
+                transcription = transcribe_audio(wav_audio_file)
+
+                # Show the transcription
+                st.text_area("Transcription", transcription, height=300)
+
+                # Store transcription in session state
+                st.session_state.transcription_audio = transcription
+                st.session_state.wav_audio_file_audio = wav_audio_file
+
+                # Cleanup temporary audio file
+                os.remove(tmp_audio_path)
+
+        # 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
+            st.audio(st.session_state.wav_audio_file_audio, 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_audio,
+                file_name="transcription_audio.txt",
+                mime="text/plain"
+            )
+            
+            # 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"
+                )