Update app.py

app.py

@@ -15,12 +15,15 @@ def generate_frequency_visualization(audio_path):
     try:
         # Load the audio file
         y, sr = librosa.load(audio_path, sr=None)
+        print(f"Loaded audio file with sampling rate: {sr}, and duration: {librosa.get_duration(y=y, sr=sr)} seconds.")
 
         if sr == 0 or len(y) == 0:
             raise ValueError("Invalid audio file: sampling rate or audio data is zero.")
 
         # Perform Short-Time Fourier Transform (STFT)
-        D = librosa.amplitude_to_db(np.abs(librosa.stft(y)), ref=np.max)
+        n_fft = 2048  # Ensure n_fft is set to a valid number
+        hop_length = 512  # Ensure hop_length is set to a valid number
+        D = librosa.amplitude_to_db(np.abs(librosa.stft(y, n_fft=n_fft, hop_length=hop_length)), ref=np.max)
 
         # Create a directory to save the frames
         os.makedirs('frames', exist_ok=True)
@@ -28,11 +31,12 @@ def generate_frequency_visualization(audio_path):
         # Generate and save each frame
         for i, frame in enumerate(D.T):
             plt.figure(figsize=(10, 6))
-            librosa.display.specshow(frame.reshape(1, -1), sr=sr, x_axis='time', y_axis='log')
+            librosa.display.specshow(frame.reshape(1, -1), sr=sr, x_axis='time', y_axis='log', hop_length=hop_length)
             plt.axis('off')
             plt.savefig(f'frames/frame_{i:04d}.png', bbox_inches='tight', pad_inches=0)
             plt.close()
 
+        print(f"Generated {len(D.T)} frames for visualization.")
         return 'frames'
     except Exception as e:
         print(f"Error generating frequency visualization: {e}")
@@ -68,6 +72,7 @@ def create_video_from_frames(frames_directory):
         video_path = 'output_video.mp4'
         clip.write_videofile(video_path, codec='libx264')
 
+        print(f"Video created with {len(frame_files)} frames.")
         return video_path
     except Exception as e:
         print(f"Error creating video from frames: {e}")