Update app.py

app.py

@@ -1,12 +1,15 @@
 import cadquery as cq
 import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
 import numpy as np
-import trimesh
 import gradio as gr
 from cadquery.occ_impl.exporters import exportShape
 
+
+# Function to create 3D ear-like shell using CadQuery
 def create_bird_ear_shell(height, radius, thickness):
     try:
+        # Create outer shell
         outer_shell = (
             cq.Workplane("XY")
             .lineTo(0, height)
@@ -14,6 +17,7 @@ def create_bird_ear_shell(height, radius, thickness):
             .close()
             .revolve()
         )
+        # Create inner shell
         inner_shell = (
             cq.Workplane("XY")
             .lineTo(0, height - thickness)
@@ -21,11 +25,15 @@ def create_bird_ear_shell(height, radius, thickness):
             .close()
             .revolve()
         )
-        return outer_shell.cut(inner_shell)
+        # Subtract inner shell from outer to create a hollow shell
+        shell = outer_shell.cut(inner_shell)
+        return shell
     except Exception as e:
-        print(f"Error in creating shell geometry: {e}")
+        print(f"Error in shell creation: {e}")
         return None
 
+
+# Function to generate a 2D visualization using Matplotlib
 def generate_2d_image(height, radius):
     try:
         theta = np.linspace(0, 2 * np.pi, 100)
@@ -48,6 +56,8 @@ def generate_2d_image(height, radius):
         print(f"Error in generating 2D image: {e}")
         return None
 
+
+# Function to generate a 3D STL file using CadQuery
 def generate_3d_stl(height, radius, thickness):
     try:
         bird_shell_model = create_bird_ear_shell(height, radius, thickness)
@@ -61,15 +71,42 @@ def generate_3d_stl(height, radius, thickness):
         print(f"Error in generating 3D STL: {e}")
         return None
 
+
+# Function to create a 3D surface plot using Matplotlib
+def plot_3d_surface(height, radius):
+    try:
+        fig = plt.figure(figsize=(8, 8))
+        ax = fig.add_subplot(111, projection="3d")
+
+        # Generate dummy data for the 3D surface
+        theta = np.linspace(0, 2 * np.pi, 100)
+        z = np.linspace(0, height, 50)
+        theta, z = np.meshgrid(theta, z)
+        x = (radius + 5 * np.cos(z)) * np.cos(theta)
+        y = (radius + 5 * np.cos(z)) * np.sin(theta)
+
+        # Plot the surface
+        ax.plot_surface(x, y, z, cmap="viridis", edgecolor="k", alpha=0.8)
+        ax.set_title("3D Surface of Shell")
+        ax.set_xlabel("X")
+        ax.set_ylabel("Y")
+        ax.set_zlabel("Height")
+        plt.savefig("3d_surface_plot.png")
+        return "3d_surface_plot.png"
+    except Exception as e:
+        print(f"Error in generating 3D plot: {e}")
+        return None
+
+
+# Main function to generate visuals
 def generate_visuals(height, radius, thickness):
     two_d_image = generate_2d_image(height, radius)
     three_d_model = generate_3d_stl(height, radius, thickness)
-    if two_d_image is None:
-        two_d_image = "Error: Unable to generate 2D visualization."
-    if three_d_model is None:
-        three_d_model = "Error: Unable to generate 3D STL file."
-    return two_d_image, three_d_model
+    three_d_plot = plot_3d_surface(height, radius)
+    return two_d_image, three_d_model, three_d_plot
+
 
+# Gradio Interface
 interface = gr.Interface(
     fn=generate_visuals,
     inputs=[
@@ -80,9 +117,10 @@ interface = gr.Interface(
     outputs=[
         gr.Image(label="2D Visualization"),
         gr.File(label="Download 3D Model (STL)"),
+        gr.Image(label="3D Surface Visualization"),
     ],
     title="Ear-Like Bird Shell Simulation",
-    description="Input the parameters to create an ear-like bird shell model. You can visualize it in 2D and download the 3D STL file for further use."
+    description="Input parameters to create an ear-like bird shell model. Visualize it in 2D and 3D, and download the 3D STL file."
 )
 
 interface.launch()