Update app.py

app.py

@@ -5,61 +5,71 @@ import trimesh
 import gradio as gr
 from cadquery.occ_impl.exporters import exportShape
 
-# Function to create the CAD model
 def create_bird_ear_shell(height, radius, thickness):
-    outer_shell = (
-        cq.Workplane("XY")
-        .lineTo(0, height)                # Vertical height
-        .radiusArc((radius, 0), radius)  # Curve to base
-        .close()
-        .revolve()                       # Revolve to create 3D shape
-    )
-    inner_shell = (
-        cq.Workplane("XY")
-        .lineTo(0, height - thickness)
-        .radiusArc((radius - thickness, 0), radius - thickness)
-        .close()
-        .revolve()
-    )
-    shell = outer_shell.cut(inner_shell)
-    return shell
+    try:
+        outer_shell = (
+            cq.Workplane("XY")
+            .lineTo(0, height)
+            .radiusArc((radius, 0), radius)
+            .close()
+            .revolve()
+        )
+        inner_shell = (
+            cq.Workplane("XY")
+            .lineTo(0, height - thickness)
+            .radiusArc((radius - thickness, 0), radius - thickness)
+            .close()
+            .revolve()
+        )
+        return outer_shell.cut(inner_shell)
+    except Exception as e:
+        print(f"Error in creating shell geometry: {e}")
+        return None
 
-# Function to generate a 2D visualization
 def generate_2d_image(height, radius):
-    theta = np.linspace(0, 2 * np.pi, 100)
-    x_outer = radius * np.cos(theta)
-    y_outer = height * np.sin(theta) / max(np.sin(theta))
+    try:
+        theta = np.linspace(0, 2 * np.pi, 100)
+        x_outer = radius * np.cos(theta)
+        y_outer = height * np.sin(theta) / max(np.sin(theta))
 
-    plt.figure(figsize=(6, 6))
-    plt.plot(x_outer, y_outer, label="Outer Shell", color="blue")
-    plt.fill_between(x_outer, y_outer, color="lightblue", alpha=0.5)
-    plt.axhline(0, color="black", linewidth=0.5, linestyle="--")
-    plt.axvline(0, color="black", linewidth=0.5, linestyle="--")
-    plt.title("2D Cross-Section of Ear-Like Bird Shell")
-    plt.xlabel("Radius (mm)")
-    plt.ylabel("Height (mm)")
-    plt.grid()
-    plt.legend()
-    plt.savefig("2d_visualization.png")
-    return "2d_visualization.png"
+        plt.figure(figsize=(6, 6))
+        plt.plot(x_outer, y_outer, label="Outer Shell", color="blue")
+        plt.fill_between(x_outer, y_outer, color="lightblue", alpha=0.5)
+        plt.axhline(0, color="black", linewidth=0.5, linestyle="--")
+        plt.axvline(0, color="black", linewidth=0.5, linestyle="--")
+        plt.title("2D Cross-Section of Ear-Like Bird Shell")
+        plt.xlabel("Radius (mm)")
+        plt.ylabel("Height (mm)")
+        plt.grid()
+        plt.legend()
+        plt.savefig("2d_visualization.png")
+        return "2d_visualization.png"
+    except Exception as e:
+        print(f"Error in generating 2D image: {e}")
+        return None
 
-# Function to generate a 3D STL file
 def generate_3d_stl(height, radius, thickness):
-    bird_shell_model = create_bird_ear_shell(height, radius, thickness)
-    stl_path = "ear_bird_shell.stl"
-    with open(stl_path, "wb") as f:
-        exportShape(bird_shell_model.val(), "STL", f)
-    return stl_path
+    try:
+        bird_shell_model = create_bird_ear_shell(height, radius, thickness)
+        if bird_shell_model is None:
+            raise ValueError("Invalid CAD geometry.")
+        stl_path = "ear_bird_shell.stl"
+        with open(stl_path, "wb") as f:
+            exportShape(bird_shell_model.val(), "STL", f)
+        return stl_path
+    except Exception as e:
+        print(f"Error in generating 3D STL: {e}")
+        return None
 
-# Gradio interface functions
 def generate_visuals(height, radius, thickness):
-    # Generate 2D image
     two_d_image = generate_2d_image(height, radius)
-    # Generate 3D STL file
     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
 
-# Gradio interface
 interface = gr.Interface(
     fn=generate_visuals,
     inputs=[
@@ -75,5 +85,4 @@ interface = gr.Interface(
     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."
 )
 
-# Launch the app
 interface.launch()