Update app.py

app.py

@@ -5,85 +5,77 @@ import time
 
 # Page Configurations
 st.set_page_config(
-    page_title="Gun Firing Simulation",
+    page_title="CBT: Gun Firing Internal Mechanism",
     layout="centered"
 )
 
 # Title and Description
-st.title("Gun Firing Simulation")
-st.write("This app simulates the firing of a bullet from a gun. Watch the bullet travel along its trajectory!")
+st.title("CBT: Gun Firing Internal Mechanism")
+st.write("This app simulates the internal mechanism of a gun firing process, including the combustion and pressure dynamics inside the chamber.")
 
 # Parameters Input
 st.sidebar.header("Simulation Parameters")
-initial_velocity = st.sidebar.number_input("Initial Velocity (m/s):", min_value=1, max_value=1000, value=500, step=10)
-angle = st.sidebar.slider("Firing Angle (degrees):", min_value=0, max_value=90, value=45)
-gravity = st.sidebar.number_input("Gravity (m/s²):", min_value=1.0, max_value=20.0, value=9.8, step=0.1)
+chamber_pressure = st.sidebar.number_input("Chamber Pressure (MPa):", min_value=1, max_value=100, value=50, step=1)
+barrel_length = st.sidebar.number_input("Barrel Length (cm):", min_value=10, max_value=200, value=50, step=5)
+bullet_mass = st.sidebar.number_input("Bullet Mass (grams):", min_value=1, max_value=50, value=10, step=1)
+gas_expansion_ratio = st.sidebar.slider("Gas Expansion Ratio:", min_value=1.0, max_value=5.0, value=1.5, step=0.1)
 
-def simulate_bullet():
+def simulate_internal_mechanism():
     # Constants
-    angle_rad = np.radians(angle)
-    
-    # Time of flight
-    time_of_flight = (2 * initial_velocity * np.sin(angle_rad)) / gravity
-    
-    # Maximum range
-    max_range = (initial_velocity**2 * np.sin(2 * angle_rad)) / gravity
-    
-    # Maximum height
-    max_height = (initial_velocity**2 * (np.sin(angle_rad)**2)) / (2 * gravity)
-    
-    # Time intervals
-    t = np.linspace(0, time_of_flight, num=500)
-
-    # Trajectory equations
-    x = initial_velocity * np.cos(angle_rad) * t
-    y = initial_velocity * np.sin(angle_rad) * t - 0.5 * gravity * t**2
-
-    return x, y, max_range, max_height, time_of_flight
+    chamber_pressure_pa = chamber_pressure * 1e6  # Convert MPa to Pa
+    barrel_length_m = barrel_length / 100  # Convert cm to meters
+    bullet_mass_kg = bullet_mass / 1000  # Convert grams to kg
+
+    # Time array for simulation (arbitrary resolution)
+    time_steps = np.linspace(0, 0.01, num=500)  # 10 ms simulation
+
+    # Pressure dynamics (simplified as linear decay for simulation purposes)
+    pressure_profile = chamber_pressure_pa * np.exp(-time_steps / (gas_expansion_ratio * 0.002))
+
+    # Force on bullet
+    force_profile = pressure_profile * (np.pi * (0.01)**2)  # Assuming a barrel diameter of 2 cm
+
+    # Acceleration of bullet
+    acceleration_profile = force_profile / bullet_mass_kg
+
+    # Velocity and displacement
+    velocity = np.cumsum(acceleration_profile * np.diff(np.insert(time_steps, 0, 0)))
+    displacement = np.cumsum(velocity * np.diff(np.insert(time_steps, 0, 0)))
+
+    return time_steps, pressure_profile, velocity, displacement
 
 # Simulate and Plot
-if st.button("Fire Bullet!"):
-    st.write("## Bullet Trajectory")
-    st.write("Simulating... Hold on!")
-
-    x, y, max_range, max_height, time_of_flight = simulate_bullet()
-
-    # Plot
-    fig, ax = plt.subplots()
-    ax.plot(x, y, label="Bullet Trajectory")
-    ax.axhline(0, color='black', linewidth=0.5, linestyle='--')
-    ax.set_xlabel("Distance (m)")
-    ax.set_ylabel("Height (m)")
-    ax.set_title("Bullet Trajectory Simulation")
-    ax.legend()
-
-    st.pyplot(fig)
-
-    # Display parameters
-    st.write(f"**Maximum Range:** {max_range:.2f} meters")
-    st.write(f"**Maximum Height:** {max_height:.2f} meters")
-    st.write(f"**Time of Flight:** {time_of_flight:.2f} seconds")
-
-    # Animation Simulation
-    st.write("### Animation of the Bullet")
-    progress_bar = st.progress(0)
-    bullet_fig, bullet_ax = plt.subplots()
-
-    for i in range(len(x)):
-        if y[i] < 0:  # Stop animation if bullet hits the ground
-            break
-        bullet_ax.clear()
-        bullet_ax.plot(x, y, label="Bullet Trajectory")
-        bullet_ax.plot(x[i], y[i], 'ro', label="Bullet Position")
-        bullet_ax.axhline(0, color='black', linewidth=0.5, linestyle='--')
-        bullet_ax.set_xlim(0, max(x))
-        bullet_ax.set_ylim(0, max(y) + 5)
-        bullet_ax.set_xlabel("Distance (m)")
-        bullet_ax.set_ylabel("Height (m)")
-        bullet_ax.set_title("Bullet Animation")
-        bullet_ax.legend()
-        st.pyplot(bullet_fig)
-        progress_bar.progress(int((i / len(x)) * 100))
-        time.sleep(0.01)
-
-    st.write("Simulation Complete!")
+if st.button("Simulate Gun Firing!"):
+    st.write("## Internal Mechanism Simulation")
+    st.write("Simulating the internal dynamics of the gun firing process...")
+
+    time_steps, pressure_profile, velocity, displacement = simulate_internal_mechanism()
+
+    # Plot Chamber Pressure
+    fig1, ax1 = plt.subplots()
+    ax1.plot(time_steps * 1000, pressure_profile / 1e6, label="Chamber Pressure")
+    ax1.set_xlabel("Time (ms)")
+    ax1.set_ylabel("Pressure (MPa)")
+    ax1.set_title("Chamber Pressure Over Time")
+    ax1.legend()
+    st.pyplot(fig1)
+
+    # Plot Bullet Velocity
+    fig2, ax2 = plt.subplots()
+    ax2.plot(time_steps * 1000, velocity, label="Bullet Velocity")
+    ax2.set_xlabel("Time (ms)")
+    ax2.set_ylabel("Velocity (m/s)")
+    ax2.set_title("Bullet Velocity Over Time")
+    ax2.legend()
+    st.pyplot(fig2)
+
+    # Plot Bullet Displacement
+    fig3, ax3 = plt.subplots()
+    ax3.plot(time_steps * 1000, displacement, label="Bullet Displacement")
+    ax3.set_xlabel("Time (ms)")
+    ax3.set_ylabel("Displacement (m)")
+    ax3.set_title("Bullet Displacement Over Time")
+    ax3.legend()
+    st.pyplot(fig3)
+
+    st.write("Simulation Complete! Explore the dynamics of the gun's internal firing mechanism.")