Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import matplotlib.pyplot as plt
 import numpy as np
 import matplotlib.animation as animation
-from io import BytesIO
+import tempfile
 import base64
 
 # Set up page configuration
@@ -55,11 +55,12 @@ def update(frame):
         electron.set_data([x], [y])  # Wrap x and y in lists to avoid RuntimeError
     return electrons
 
-# Generate animation
-ani = animation.FuncAnimation(fig, update, frames=range(200), interval=50, blit=True)
+# Generate animation and save as temporary HTML file
+with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as tmpfile:
+    ani = animation.FuncAnimation(fig, update, frames=range(200), interval=50, blit=True)
+    ani.save(tmpfile.name, writer="html", fps=20)
+    tmpfile.seek(0)
+    html_data = tmpfile.read().decode("utf-8")
 
-# Save animation as HTML and embed in Streamlit
-html_anim = BytesIO()
-ani.save(html_anim, writer="html", fps=20)
-html_data = base64.b64encode(html_anim.getvalue()).decode("utf-8")
-st.components.v1.html(f"<div align='center'><img src='data:image/gif;base64,{html_data}'></div>", height=500)
+# Display animation in Streamlit
+st.components.v1.html(html_data, height=500)