Update app.py

app.py

@@ -3,7 +3,6 @@ import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.animation import FuncAnimation, PillowWriter
 import scipy.io.wavfile as wavfile
-import os
 
 # Constants for sound generation
 SAMPLE_RATE = 48000
@@ -11,8 +10,10 @@ COLUMN_DURATION = 1  # Duration of each column in seconds
 
 # Mapping of matrix numbers to musical notes
 notes = ["A", "A#", "B", "C", "C#", "D", "D#", "E", "F", "F#", "G", "G#"]
-note_map = {2: "A", 3: "A#", 4: "B", 5: "C", 6: "C#", 7: "D", 8: "D#", 9: "E", 
-            10: "F", 11: "F#", 12: "G", 13: "G#"}
+note_map = {
+    2: "A", 3: "A#", 4: "B", 5: "C", 6: "C#", 7: "D", 8: "D#", 9: "E", 
+    10: "F", 11: "F#", 12: "G", 13: "G#"
+}
 
 def generate_tone(note, duration):
     """Generates a tone for a specified note and duration."""
@@ -146,7 +147,7 @@ def create_animation(matrix):
             has_changes, matrix = sim.step()
             if not has_changes:
                 ani.event_source.stop()
-                return
+                return circles + labels
         
         for i in range(rows):
             for j in range(cols):
@@ -171,7 +172,9 @@ def create_animation(matrix):
         current_frame[0] += 1
         return circles + labels
     
-    ani = FuncAnimation(fig, update, frames=None, interval=1000, blit=True)
+    # Estimate the number of frames needed
+    max_steps = 100  # Set a reasonable limit for the number of steps
+    ani = FuncAnimation(fig, update, frames=max_steps, interval=1000, blit=True, cache_frame_data=False)
     
     # Save the animation to a GIF file
     gif_path = "matrix_animation.gif"