Update app.py

app.py

@@ -41,12 +41,25 @@ def generate_frequency_visualization(audio_path, fps, num_bars):
 
         # Generate and save each frame
         for i, heights in enumerate(bar_heights):
-            fig, ax = plt.subplots(figsize=(10, 6))
-            ax.bar(range(num_bars), heights, color=plt.cm.viridis(np.linspace(0, 1, num_bars)))
-            ax.set_ylim(0, np.max(S))
-            ax.axis('off')
-            plt.savefig(f'frames/frame_{i:04d}.png', bbox_inches='tight', pad_inches=0)
-            plt.close()
+            # Create black background
+            img = np.zeros((720, 1280, 3), dtype=np.uint8)
+
+            # Normalize heights to fit the frame
+            heights = np.array(heights)
+            heights = (heights / np.max(heights) * 600).astype(int)
+
+            # Calculate bar positions
+            bar_width = 80
+            spacing = (1280 - num_bars * bar_width) // (num_bars + 1)
+            for j, height in enumerate(heights):
+                x = spacing + j * (bar_width + spacing)
+                y = 720 - height
+                color = tuple((np.array(plt.cm.viridis(j / num_bars))[:3] * 255).astype(int))  # Use Viridis colormap
+                cv2.rectangle(img, (x, 720), (x + bar_width, y), color, -1)
+
+            # Save the frame
+            frame_path = f'frames/frame_{i:04d}.png'
+            cv2.imwrite(frame_path, img)
 
         print(f"Generated {len(bar_heights)} frames for visualization.")
         return 'frames', duration