Update app.py

app.py

@@ -2,33 +2,76 @@ import streamlit as st
 import numpy as np
 import plotly.graph_objs as go
 
-st.title("Electromagnetic Wave Simulation")
-
-# Set up initial values
-c = 3e8  # Speed of light (m/s)
-lambda_ = 1e-9  # Wavelength (m)
-omega = 2 * np.pi * c / lambda_  # Angular frequency
-k = omega / c  # Wave number
-E0 = 1  # Amplitude of electric field
-B0 = E0 / c  # Amplitude of magnetic field
-t = np.linspace(0, 1e-14, 1000)  # Time (s)
-z = np.linspace(0, 1e-6, 100)  # Position (m)
-
-# Calculate the electric and magnetic fields
-E = E0 * np.sin(k * z - omega * t[:, np.newaxis])
-B = B0 * np.sin(k * z - omega * t[:, np.newaxis])
-
-# Create a 3D surface plot
-fig = go.Figure(
-    go.Surface(x=z, y=t, z=E.T, colorscale='Blues', showscale=False, name="Electric Field"),
-    go.Surface(x=z, y=t, z=B.T, colorscale='Reds', showscale=False, name="Magnetic Field"),
+st.title('Electromagnetic Wave Simulation with Plotly')
+
+# Set the default wavelength and amplitude values
+wavelength_default = 1
+amplitude_default = 1
+
+# Create sliders for adjusting wavelength and amplitude
+wavelength = st.slider('Wavelength', min_value=0.1, max_value=10.0, value=wavelength_default, step=0.1)
+amplitude = st.slider('Amplitude', min_value=0.1, max_value=5.0, value=amplitude_default, step=0.1)
+
+# Define the wave function for the electric field
+def electric_field(x, t):
+    return amplitude * np.sin(2*np.pi*x/wavelength - 2*np.pi*t)
+
+# Define the wave function for the magnetic field
+def magnetic_field(x, t):
+    return amplitude * np.sin(2*np.pi*x/wavelength - 2*np.pi*t + np.pi/2)
+
+# Create a plotly figure for the electric field
+fig_electric = go.Figure()
+
+# Add a line for the electric field
+fig_electric.add_trace(
+    go.Scatter(x=np.linspace(0, 1, 1000), y=electric_field(np.linspace(0, 1, 1000), 0),
+               mode='lines', name='Electric Field')
+)
+
+# Create a plotly figure for the magnetic field
+fig_magnetic = go.Figure()
+
+# Add a line for the magnetic field
+fig_magnetic.add_trace(
+    go.Scatter(x=np.linspace(0, 1, 1000), y=magnetic_field(np.linspace(0, 1, 1000), 0),
+               mode='lines', name='Magnetic Field')
+)
+
+# Create a plotly figure for the wave
+fig_wave = go.Figure()
+
+# Add a line for the electric field
+fig_wave.add_trace(
+    go.Scatter(x=np.linspace(0, 1, 1000), y=electric_field(np.linspace(0, 1, 1000), 0),
+               mode='lines', name='Electric Field')
 )
-fig.update_layout(
-    title="Electromagnetic Wave Simulation",
-    scene=dict(xaxis_title="Position (m)", yaxis_title="Time (s)", zaxis_title="Field Strength"),
-    margin=dict(l=0, r=0, b=0, t=40),
-    height=700,
+
+# Add a line for the magnetic field
+fig_wave.add_trace(
+    go.Scatter(x=np.linspace(0, 1, 1000), y=magnetic_field(np.linspace(0, 1, 1000), 0),
+               mode='lines', name='Magnetic Field')
+)
+
+# Set the title and axis labels for the wave figure
+fig_wave.update_layout(
+    title='Electromagnetic Wave',
+    xaxis_title='Position',
+    yaxis_title='Field Strength'
 )
 
-# Display the plot
-st.plotly_chart(fig, use_container_width=True)
+# Add a slider for adjusting time
+t_slider = fig_wave.add_slider(
+    dict(steps=[dict(method='animate', args=[None, {'frame': {'duration': 50, 'redraw': True}}])],
+         transition={'duration': 0},
+         x=0, y=0,
+         len=1.0,
+         currentvalue=dict(visible=True, xanchor='left'),
+         font=dict(size=10, color='#666')
+         ),
+    )
+
+# Define the animation frames
+frames = [go.Frame(data=[go.Scatter(x=np.linspace(0, 1, 1000), y=electric_field(np.linspace(0, 1, 1000), t),
+                                     mode='lines', name='Electric Field'),
+                         go.Scatter(x=np.linspace(0, 1, 1000), y=magnetic_field(np.linspace(0, 1,