Create simluation.py

simluation.py

@@ -0,0 +1,40 @@
+import streamlit as st
+import numpy as np
+import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
+
+# Define the time variable
+t = np.linspace(0, 2*np.pi, 100)
+
+# Create a meshgrid for the x and y axes
+x, y = np.meshgrid(np.linspace(-1, 1, 100), np.linspace(-1, 1, 100))
+
+# Define the amplitudes and frequencies of the waves
+amp_e = st.sidebar.slider('Amplitude of Electric Field', min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+amp_b = st.sidebar.slider('Amplitude of Magnetic Field', min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+freq = st.sidebar.slider('Frequency of Waves', min_value=1, max_value=10, value=5, step=1)
+
+# Calculate the electric and magnetic fields
+e_x = amp_e * np.cos(freq*t)
+e_y = np.zeros_like(e_x)
+e_z = np.zeros_like(e_x)
+
+b_x = np.zeros_like(e_x)
+b_y = amp_b * np.cos(freq*t)
+b_z = np.zeros_like(e_x)
+
+# Plot the electric and magnetic fields
+fig = plt.figure(figsize=(10, 6))
+ax = fig.add_subplot(111, projection='3d')
+ax.quiver(x, y, np.zeros_like(x), e_x, e_y, e_z, length=0.1, color='r')
+ax.quiver(x, y, np.zeros_like(x), b_x, b_y, b_z, length=0.1, color='b')
+ax.set_xlim(-1, 1)
+ax.set_ylim(-1, 1)
+ax.set_zlim(-1, 1)
+ax.set_xlabel('X')
+ax.set_ylabel('Y')
+ax.set_zlabel('Z')
+ax.view_init(elev=30, azim=120)
+
+# Display the plot in Streamlit
+st.pyplot(fig)