Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import matplotlib.pyplot as plt
 import numpy as np
 import matplotlib.animation as animation
+from io import BytesIO
+import base64
 
 # Set up page configuration
 st.set_page_config(page_title="Hithesh Rai - Interactive Bohr Model of the Atom", page_icon="🔬")
@@ -53,11 +55,11 @@ def update(frame):
         electron.set_data([x], [y])  # Wrap x and y in lists to avoid RuntimeError
     return electrons
 
-# Animation
+# Generate animation
 ani = animation.FuncAnimation(fig, update, frames=range(200), interval=50, blit=True)
 
-# Display animation in Streamlit
-st.pyplot(fig)
-
-# Footer
-st.write("Explore the Bohr Model! Adjust the number of electrons and watch them orbit around the nucleus.")
+# 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)