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 io
 
 # Constants for sound generation
 SAMPLE_RATE = 48000
@@ -173,20 +172,19 @@ def create_animation(matrix):
     
     ani = FuncAnimation(fig, update, frames=None, interval=1000, blit=True)
     
-    # Save the animation to a GIF
-    gif_buffer = io.BytesIO()
-    ani.save(gif_buffer, format='gif', writer=PillowWriter(fps=1))
+    # Save the animation to a GIF file
+    gif_path = "matrix_animation.gif"
+    ani.save(gif_path, format='gif', writer=PillowWriter(fps=1))
     plt.close(fig)
-    gif_buffer.seek(0)
     
-    return gif_buffer
+    return gif_path
 
 def run_simulation(matrix_input):
     """
     Run the full simulation based on user-input matrix
     
     :param matrix_input: 2D list of integers representing the matrix
-    :return: tuple of (audio_path, gif_buffer)
+    :return: tuple of (audio_path, gif_path)
     """
     # Convert input to numpy array
     matrix = np.array(matrix_input, dtype=int)
@@ -209,9 +207,9 @@ def run_simulation(matrix_input):
     wavfile.write(audio_path, SAMPLE_RATE, final_audio)
     
     # Create animation
-    gif_buffer = create_animation(matrix)
+    gif_path = create_animation(matrix)
     
-    return audio_path, gif_buffer
+    return audio_path, gif_path
 
 # Gradio Interface
 def create_gradio_interface():
@@ -238,7 +236,7 @@ def create_gradio_interface():
         ],
         outputs=[
             gr.Audio(type="filepath", label="Generated Sound"),
-            gr.Image(type="file", label="Matrix Animation")
+            gr.Image(type="filepath", label="Matrix Animation")
         ],
         title="Number Spread Simulator",
         description="Edit the matrix and see how numbers spread, generating a unique sound and animation!"