Update app.py

app.py

@@ -29,19 +29,14 @@ def generate_frequency_visualization(audio_path):
         os.makedirs('frames', exist_ok=True)
 
         # Generate and save each frame
-        for i, frame in enumerate(D.T):
+        for i in range(D.shape[1]):  # Iterate over columns of D (time frames)
             plt.figure(figsize=(10, 6))
-            librosa.display.specshow(frame.reshape(1, -1), sr=sr, x_axis='time', y_axis='log', hop_length=hop_length)
+            librosa.display.specshow(D[:, i].reshape(-1, 1), sr=sr, x_axis='time', y_axis='log', hop_length=hop_length, cmap='viridis')
             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 ZeroDivisionError as e:
-        print(f"ZeroDivisionError: {e}")
-        # Fallback: Generate a default visualization
-        generate_default_visualization()
+        print(f"Generated {D.shape[1]} frames for visualization.")
         return 'frames'
     except Exception as e:
         print(f"Error generating frequency visualization: {e}")
@@ -113,7 +108,3 @@ if __name__ == "__main__":
 # - numpy
 # - matplotlib
 # - moviepy
-# - gradio
-#
-# You can install these dependencies using pip:
-# pip install librosa numpy matplotlib moviepy gradio