Update app.py

app.py

@@ -4,13 +4,15 @@ from mpl_toolkits import mplot3d
 import matplotlib.pyplot as plt
 import tempfile
 import os
+import trimesh
+import pyvista as pv
 
 # Page configuration
 st.set_page_config(page_title="AR CAD Viewer", layout="wide")
 
 # Title
 st.title("Augmented Reality CAD Viewer")
-st.write("Upload your CAD models and view them in a 3D interactive environment.")
+st.write("Upload your CAD models, preview them, and convert them to AR-compatible formats.")
 
 # File uploader for CAD files
 uploaded_file = st.file_uploader("Upload a CAD file (.stl format)", type=["stl"])
@@ -20,29 +22,51 @@ if uploaded_file is not None:
     with tempfile.NamedTemporaryFile(delete=False, suffix=".stl") as temp_file:
         temp_file.write(uploaded_file.read())
         temp_filepath = temp_file.name
-    
+
     # Load the STL file
     stl_mesh = mesh.Mesh.from_file(temp_filepath)
-    
-    # Create a new figure for plotting
+
+    # Preview the model in 3D
+    st.subheader("3D Model Preview")
     fig = plt.figure(figsize=(10, 10))
     ax = fig.add_subplot(111, projection="3d")
-    
-    # Plot the CAD model
     ax.add_collection3d(mplot3d.art3d.Poly3DCollection(stl_mesh.vectors, alpha=0.7, color="cyan"))
     ax.auto_scale_xyz(stl_mesh.points.flatten(), stl_mesh.points.flatten(), stl_mesh.points.flatten())
-    
-    # Display the 3D plot in Streamlit
     st.pyplot(fig)
-    
-    # Clean up the temporary file
-    os.remove(temp_filepath)
-
-# AR.js link
-st.markdown(
-    """
-    ## AR Feature
-    Use [AR.js](https://ar-js-org.github.io/AR.js/) to host your model online for AR visualization.
-    Upload this model to a compatible AR hosting service or modify the file to be AR.js-ready.
-    """
-)
+
+    # Convert to glTF for AR
+    st.subheader("Convert to AR Format")
+    st.write("Click below to convert the STL file into an AR-compatible `.glTF` format.")
+
+    if st.button("Convert to glTF"):
+        # Use Trimesh for conversion
+        trimesh_mesh = trimesh.load(temp_filepath)
+        gltf_filepath = temp_filepath.replace(".stl", ".gltf")
+        trimesh.exchange.gltf.export_gltf(trimesh_mesh, gltf_filepath)
+
+        # Display download link for glTF
+        with open(gltf_filepath, "rb") as gltf_file:
+            st.download_button(
+                label="Download glTF Model",
+                data=gltf_file,
+                file_name="model.gltf",
+                mime="model/gltf+json"
+            )
+
+        # Clean up temporary files
+        os.remove(gltf_filepath)
+        os.remove(temp_filepath)
+
+    # AR Hosting Info
+    st.subheader("Host in AR")
+    st.markdown(
+        """
+        Upload your `.glTF` file to [GitHub Pages](https://pages.github.com/) or use a service like [Google Poly](https://poly.google.com/) (now integrated into other platforms).
+        
+        For AR.js:
+        - Host the `.glTF` file on a web server.
+        - Create an AR marker or use markerless AR for viewing.
+
+        Example: `<a-scene><a-entity gltf-model="model.gltf"></a-entity></a-scene>`
+        """
+    )