Update app.py

app.py

@@ -122,6 +122,8 @@ def main():
                         
                         # Keep the audio_file in session state for transcription
                         st.session_state.last_recorded_audio = audio_file
+                        # Force a rerun to update the UI
+                        st.rerun()
                 except Exception as e:
                     st.error(f"Error stopping recording: {str(e)}")
                 finally:
@@ -129,13 +131,16 @@ def main():
                     st.rerun()
     
     # Transcription Section
-    if audio_file and st.button("Transcribe Audio"):
-        with st.spinner("Transcribing audio..."):
-            try:
-                if use_hf:
-                    transcriber = HFTranscriber(model_name=model_name)
-                else:
-                    transcriber = AudioTranscriber()
+    if 'last_recorded_audio' in st.session_state and st.session_state.last_recorded_audio:
+        audio_file = st.session_state.last_recorded_audio
+        
+        if st.button("🎵 Transcribe Audio"):
+            with st.spinner("Transcribing audio..."):
+                try:
+                    if use_hf:
+                        transcriber = HFTranscriber(model_name=model_name)
+                    else:
+                        transcriber = AudioTranscriber()
                 
                 # Create output directory
                 os.makedirs("outputs", exist_ok=True)
@@ -143,18 +148,37 @@ def main():
                 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")
-                st.markdown(f"**MIDI File:** {get_binary_file_downloader_html(midi_output, 'Download MIDI')}", 
-                           unsafe_allow_html=True)
-                st.markdown(f"**MusicXML File:** {get_binary_file_downloader_html(musicxml_output, 'Download MusicXML')}", 
-                          unsafe_allow_html=True)
+                try:
+                    # 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)}")