Update app.py

app.py

@@ -171,46 +171,121 @@ def main():
         col1, col2 = st.columns(2)
         
         with col1:
-            if st.button("🎤 Start Recording", disabled=st.session_state.get('recording', False)):
+            # Check if we have any audio devices
+            devices_available = any(device.get('max_input_channels', 0) > 0 
+                                 for device in app_config.get('AUDIO_DEVICES', []))
+            
+            if not devices_available:
+                st.warning("⚠️ No audio input devices detected. Recording may not work correctly.")
+            
+            if st.button("🎤 Start Recording", 
+                        disabled=st.session_state.get('recording', False), 
+                        key='start_recording_btn'):
                 try:
+                    # Create a new recorder instance if needed
                     if 'recorder' not in st.session_state or st.session_state.recorder is None:
+                        print("Creating new AudioRecorder instance...")
                         st.session_state.recorder = AudioRecorder(device_index=selected_device)
-                    st.session_state.recorder.start_recording()
-                    st.session_state.recording = True
-                    st.success("Recording started...")
-                    st.rerun()
+                    
+                    print("Starting recording...")
+                    if not devices_available:
+                        st.warning("Recording started without an audio device. This will generate silent audio.")
+                    
+                    if st.session_state.recorder.start_recording():
+                        st.session_state.recording = True
+                        st.session_state.recording_started = True
+                        st.session_state.recording_error = None
+                        print("Recording started successfully")
+                        st.rerun()
+                    else:
+                        error_msg = "Failed to start recording. Please try again."
+                        print(error_msg)
+                        st.error(error_msg)
+                        st.session_state.recording_error = error_msg
+                        st.session_state.recording = False
+                        st.session_state.recording_started = False
+                        
                 except Exception as e:
-                    st.error(f"Error starting recording: {str(e)}")
+                    error_msg = f"Error starting recording: {str(e)}"
+                    print(error_msg)
+                    st.error(error_msg)
+                    st.session_state.recording_error = error_msg
+                    st.session_state.recording = False
+                    st.session_state.recording_started = False
         
         with col2:
-            if st.button("⏹️ Stop Recording", disabled=not st.session_state.get('recording', False)):
+            if st.button("⏹️ Stop Recording", 
+                        disabled=not st.session_state.get('recording', False),
+                        key='stop_recording_btn'):
                 try:
                     if 'recorder' in st.session_state and st.session_state.recorder is not None:
-                        st.session_state.recorder.stop_recording()
+                        print("Stopping recording...")
+                        
+                        # Stop the recording
+                        audio_data = st.session_state.recorder.stop_recording()
+                        
+                        if audio_data is None:
+                            st.warning("No audio data was recorded")
+                            st.session_state.recording = False
+                            st.session_state.recording_started = False
+                            return
                         
-                        # Save recording
+                        # Save the recording
                         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
                         os.makedirs("recordings", exist_ok=True)
                         audio_file = os.path.join("recordings", f"recording_{timestamp}.wav")
-                        st.session_state.recorder.save_recording(audio_file)
                         
-                        st.success("Recording saved!")
+                        print(f"Saving recording to {audio_file}...")
                         try:
-                            st.audio(audio_file)
-                        except Exception as e:
-                            st.warning(f"Could not display audio preview: {str(e)}")
-                        
-                        # Keep the audio_file in session state for transcription
-                        st.session_state.last_recorded_audio = audio_file
-                        # Clean up old recordings (keep last 5)
-                        clean_up_recordings(keep_last=5)
-                        # Update UI without rerun to prevent state issues
-                        st.experimental_rerun()
+                            saved_file = st.session_state.recorder.save_recording(audio_file)
+                            if saved_file and os.path.exists(saved_file):
+                                print(f"Successfully saved recording to {saved_file}")
+                                st.session_state.last_recording = saved_file
+                                st.session_state.last_recorded_audio = saved_file
+                                
+                                # Clean up old recordings
+                                clean_up_recordings(keep_last=5)
+                                
+                                # Display success and audio player
+                                st.success(f"Recording saved successfully!")
+                                st.audio(saved_file)
+                                
+                                # Rerun to update the UI
+                                st.rerun()
+                            else:
+                                error_msg = "Failed to save recording. No audio data was captured."
+                                print(error_msg)
+                                st.error(error_msg)
+                                st.session_state.recording_error = error_msg
+                                
+                        except Exception as save_error:
+                            error_msg = f"Error saving recording: {str(save_error)}"
+                            print(f"Save error details: {error_msg}")
+                            st.error(error_msg)
+                            st.session_state.recording_error = error_msg
+                    
+                    # Reset recording state
+                    st.session_state.recording = False
+                    st.session_state.recording_started = False
+                    
                 except Exception as e:
-                    st.error(f"Error stopping recording: {str(e)}")
-                finally:
+                    error_msg = f"Error stopping recording: {str(e)}"
+                    print(error_msg)
+                    st.error(error_msg)
+                    st.session_state.recording_error = error_msg
+                    
+                    # Ensure we reset the recording state
                     st.session_state.recording = False
-                    # Don't use rerun() here as it can cause state issues
+                    st.session_state.recording_started = False
+                finally:
+                    # Always clean up the recorder
+                    if 'recorder' in st.session_state:
+                        try:
+                            st.session_state.recorder = None
+                        except:
+                            pass
+                    
+                    # Don't use rerun() in finally as it can cause infinite loops
                     # The UI will update automatically due to Streamlit's reactivity
     
     # Transcription Section