from stl import mesh
from mpl_toolkits import mplot3d
from matplotlib import pyplot as plt

# Load the STL file
your_mesh = mesh.Mesh.from_file('sample_data.stl')

# Define three different viewing angles
viewing_angles = [(30, 45), (60, 90), (45, 135)]

# Iterate over each viewing angle and generate an image
for i, (elev, azim) in enumerate(viewing_angles, start=1):
    # Create a new plot with a larger figure size
    fig = plt.figure(figsize=(10, 10))
    ax = fig.add_subplot(111, projection='3d')

    # Add the STL file to the plot
    ax.add_collection3d(mplot3d.art3d.Poly3DCollection(your_mesh.vectors))

    # Calculate the limits of the mesh
    max_dim = max(your_mesh.points.flatten())
    min_dim = min(your_mesh.points.flatten())

    # Set the limits of the plot
    ax.set_xlim([min_dim, max_dim])
    ax.set_ylim([min_dim, max_dim])
    ax.set_zlim([min_dim, max_dim])

    # Set the viewing angle
    ax.view_init(elev=elev, azim=azim)

    # Save the plot as an image
    plt.savefig(f'mesh_{i}.png')

    # Close the plot to avoid memory leaks
    plt.close()

# Optional: Show the last generated plot
# plt.show()