Spaces:
Runtime error
Runtime error
File size: 2,094 Bytes
855f791 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import streamlit as st
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.animation as animation
# Set up page configuration
st.set_page_config(page_title="Hithesh Rai - Interactive Physics Model", page_icon="🔬")
# Header
st.title("Interactive Bohr Model of the Atom")
st.subheader("Visualize electrons orbiting around a nucleus")
# About Section
st.write("""
Explore a simple interactive model of an atom based on the Bohr model, where electrons orbit a nucleus in defined circular paths. Use the controls to set the number of electrons and watch them orbit in real time!
""")
# Input for number of electrons
num_electrons = st.slider("Select the number of electrons", 1, 5, 3)
# Create figure and axis
fig, ax = plt.subplots(figsize=(6, 6))
ax.set_xlim(-10, 10)
ax.set_ylim(-10, 10)
ax.set_aspect('equal')
ax.axis('off')
# Set up nucleus
nucleus, = ax.plot(0, 0, 'o', color='orange', markersize=10, label="Nucleus")
# Colors for electron orbits
orbit_colors = ['blue', 'green', 'red', 'purple', 'cyan']
# Set up electron orbits based on selected number
electrons = []
orbits = []
for i in range(num_electrons):
radius = 2 + 2 * i
theta = np.linspace(0, 2 * np.pi, 100)
x = radius * np.cos(theta)
y = radius * np.sin(theta)
orbit, = ax.plot(x, y, linestyle="--", color=orbit_colors[i % len(orbit_colors)], alpha=0.5)
electron, = ax.plot([], [], 'o', color=orbit_colors[i % len(orbit_colors)], markersize=6)
electrons.append(electron)
orbits.append(orbit)
# Animation update function
def update(frame):
for i, electron in enumerate(electrons):
radius = 2 + 2 * i
x = radius * np.cos(frame * 0.1 + i * np.pi / num_electrons)
y = radius * np.sin(frame * 0.1 + i * np.pi / num_electrons)
electron.set_data(x, y)
return electrons
# 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.")
|