Update app.py

app.py

@@ -135,52 +135,54 @@ def main():
         audio_file = st.session_state.last_recorded_audio
         
         if st.button("🎵 Transcribe Audio"):
-            with st.spinner("Transcribing audio..."):
-                try:
+            try:
+                with st.spinner("Transcribing audio..."):
                     # Initialize the transcriber
                     if use_hf:
                         transcriber = HFTranscriber(model_name=model_name)
                     else:
                         transcriber = AudioTranscriber()
                 
-                    # Create output directory
-                    os.makedirs("outputs", exist_ok=True)
-                    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-                    midi_output = os.path.join("outputs", f"output_{timestamp}.mid")
-                    musicxml_output = os.path.join("outputs", f"output_{timestamp}.musicxml")
-                    # Transcribe
-                    transcriber.transcribe(audio_file, midi_output, musicxml_output)
-                    
-                    # Show success message
-                    st.success("🎵 Transcription completed successfully!")
-                    
-                    # Show download links
-                    st.markdown("### Download Results")
-                    
-                    # Display MIDI file
-                    if os.path.exists(midi_output):
-                        st.markdown(f"**MIDI File:** {get_binary_file_downloader_html(midi_output, 'Download MIDI')}", 
-                                  unsafe_allow_html=True)
-                    
-                    # Display MusicXML file
-                    if os.path.exists(musicxml_output):
-                        st.markdown(f"**MusicXML File:** {get_binary_file_downloader_html(musicxml_output, 'Download MusicXML')}", 
-                                  unsafe_allow_html=True)
-                    
-                    # Display a preview of the transcription if possible
                     try:
-                        from IPython.display import display, Audio
-                        audio = Audio(audio_file)
-                        st.audio(audio_file, format='audio/wav')
-                    except Exception as e:
-                        st.warning(f"Could not display audio preview: {str(e)}")
+                        # Create output directory
+                        os.makedirs("outputs", exist_ok=True)
+                        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+                        midi_output = os.path.join("outputs", f"output_{timestamp}.mid")
+                        musicxml_output = os.path.join("outputs", f"output_{timestamp}.musicxml")
                         
-                except Exception as e:
-                    st.error(f"❌ Error during transcription: {str(e)}")
-                    st.exception(e)  # Show full traceback for debugging
-                
+                        # Transcribe
+                        transcriber.transcribe(audio_file, midi_output, musicxml_output)
+                        
+                        # Show success message
+                        st.success("🎵 Transcription completed successfully!")
+                        
+                        # Show download links
+                        st.markdown("### Download Results")
+                        
+                        # Display MIDI file
+                        if os.path.exists(midi_output):
+                            st.markdown(f"**MIDI File:** {get_binary_file_downloader_html(midi_output, 'Download MIDI')}", 
+                                      unsafe_allow_html=True)
+                        
+                        # Display MusicXML file
+                        if os.path.exists(musicxml_output):
+                            st.markdown(f"**MusicXML File:** {get_binary_file_downloader_html(musicxml_output, 'Download MusicXML')}", 
+                                      unsafe_allow_html=True)
+                        
+                        # Display a preview of the transcription if possible
+                        try:
+                            from IPython.display import display, Audio
+                            audio = Audio(audio_file)
+                            st.audio(audio_file, format='audio/wav')
+                        except Exception as e:
+                            st.warning(f"Could not display audio preview: {str(e)}")
+                            
+                    except Exception as e:
+                        st.error(f"❌ Error during transcription: {str(e)}")
+                        st.exception(e)  # Show full traceback for debugging
+            
             except Exception as e:
-                st.error(f"An error occurred during transcription: {str(e)}")
+                st.error(f"An error occurred during transcription setup: {str(e)}")
                 st.exception(e)  # Show full traceback for debugging
     
     # Clean up temporary files