Update app.py

app.py

@@ -1,44 +1,49 @@
-import cadquery as cq
-from cadquery import exporters
-import trimesh
-import matplotlib.pyplot as plt
 import gradio as gr
+import trimesh
+import io
+import base64
+from PIL import Image
+from matplotlib import pyplot as plt
 
-# Function to generate CAD model
-def generate_cad_model(length, width, height):
-    # Create a simple box model
-    model = cq.Workplane("XY").box(length, width, height)
+def generate_cad_model(base_length, base_width, base_height):
+    """
+    Generate a CAD model as a 3D mesh and return a 2D projection image for demo purposes.
+    """
+    # Create a 3D box (CAD model representation)
+    mesh = trimesh.creation.box(extents=(base_length, base_width, base_height))
     
-    # Save the model as a STEP file
-    step_filename = "customer_model.step"
-    exporters.export(model, step_filename)
+    # Create an image buffer to save a rendered view
+    image_buffer = io.BytesIO()
+    # Render the model to an image
+    mesh.show(smooth=False)
     
-    # Generate a preview using trimesh
-    preview = trimesh.creation.box(extents=(length, width, height))
-    preview_filename = "preview.png"
-    preview.export(preview_filename)
-
-    return preview_filename, step_filename
+    # Generate a 2D projection image (top view for demo)
+    projection = mesh.section(plane_origin=(0, 0, 0), plane_normal=(0, 0, 1))
+    projection_img = trimesh.util.grid_2d(projection.to_image(), tile=(1, 1))
+    plt.imshow(projection_img, cmap="gray")
+    plt.axis("off")
+    plt.savefig(image_buffer, format="png")
+    image_buffer.seek(0)
+    
+    # Convert to Base64 for Gradio to display as an image
+    encoded_image = base64.b64encode(image_buffer.getvalue()).decode("utf-8")
+    return f"data:image/png;base64,{encoded_image}"
 
-# Gradio interface for customer demo
-def customer_demo(length, width, height):
-    preview_image, step_file = generate_cad_model(length, width, height)
-    return preview_image, step_file
+# Gradio Interface
+def display_cad_model(base_length, base_width, base_height):
+    cad_image = generate_cad_model(base_length, base_width, base_height)
+    return cad_image
 
-# Launch the Gradio app
+# Hosting the interface with Gradio
 demo = gr.Interface(
-    fn=customer_demo,
+    fn=display_cad_model,
     inputs=[
-        gr.Number(label="Length (mm)", value=100),
-        gr.Number(label="Width (mm)", value=50),
-        gr.Number(label="Height (mm)", value=30)
-    ],
-    outputs=[
-        gr.Image(label="CAD Preview"),
-        gr.File(label="Download STEP File")
+        gr.Number(label="Base Length (mm)", value=10),
+        gr.Number(label="Base Width (mm)", value=10),
+        gr.Number(label="Base Height (mm)", value=5),
     ],
-    title="CAD Model Generator",
-    description="Input dimensions to generate a CAD model for demo."
+    outputs=gr.Image(label="CAD Model Projection"),
+    live=True,
 )
 
 demo.launch()