Update app.py

app.py

@@ -1,77 +1,51 @@
-import streamlit as st
 import numpy as np
 import plotly.graph_objs as go
+from plotly.subplots import make_subplots
+import streamlit as st
 
-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')
-)
-
-# 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'
-)
-
-# 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,
+def calculate_wave(wavelength, amplitude, frequency, num_periods, time_step):
+    k = 2*np.pi/wavelength
+    omega = 2*np.pi*frequency
+    period = 1/frequency
+    time_array = np.arange(0, num_periods*period, time_step)
+    wave = amplitude * np.sin(k * np.arange(0, 1, wavelength/1000)[:, None] - omega * time_array)
+    return time_array, wave
+
+def plot_3d_wave(wavelength, amplitude, frequency):
+    fig = make_subplots(rows=1, cols=2,
+                        specs=[[{'type': 'surface'}, {'type': 'surface'}]],
+                        subplot_titles=('Electric field', 'Magnetic field'),
+                        )
+    time_step = 1/(100*frequency)
+    num_periods = 5
+    time_array, wave = calculate_wave(wavelength, amplitude, frequency, num_periods, time_step)
+    E_x = wave*np.cos(2*np.pi*time_array*frequency)
+    E_y = np.zeros_like(E_x)
+    E_z = np.zeros_like(E_x)
+    B_x = np.zeros_like(E_x)
+    B_y = -wave*np.sin(2*np.pi*time_array*frequency)
+    B_z = np.zeros_like(E_x)
+    
+    fig.add_trace(go.Surface(x=wave, y=E_x, z=E_y, colorscale='Blues'), row=1, col=1)
+    fig.add_trace(go.Surface(x=wave, y=B_x, z=B_y, colorscale='Reds'), row=1, col=2)
+    fig.update_layout(scene_aspectratio=dict(x=1, y=1, z=1), width=800, height=400,
+                      scene=dict(xaxis_title='Wave', yaxis_title='E_x', zaxis_title='E_y'),
+                      scene2=dict(xaxis_title='Wave', yaxis_title='B_x', zaxis_title='B_y'),
+                     )
+    
+    fig.update_xaxes(range=[0, wavelength], row=1, col=1)
+    fig.update_xaxes(range=[0, wavelength], row=1, col=2)
+    fig.update_yaxes(range=[-amplitude, amplitude], row=1, col=1)
+    fig.update_yaxes(range=[-amplitude, amplitude], row=1, col=2)
+    fig.update_zaxes(range=[-amplitude, amplitude], row=1, col=1)
+    fig.update_zaxes(range=[-amplitude, amplitude], row=1, col=2)
+    
+    return fig
+
+wavelength = st.slider('Wavelength', 0.1, 10.0, 1.0)
+amplitude = st.slider('Amplitude', 0.1, 1.0, 0.5)
+frequency = st.slider('Frequency', 0.1, 10.0, 1.0)
+
+fig = plot_3d_wave(wavelength, amplitude, frequency)
+
+st.plotly_chart(fig)