Update app.py

app.py

@@ -1,5 +1,7 @@
-import open3d as o3d
 import streamlit as st
+import trimesh
+from pythreejs import *
+import numpy as np
 
 # Streamlit title and description
 st.title("Jewelry 3D Model Viewer")
@@ -9,28 +11,44 @@ st.write("Upload an OBJ file to visualize the 3D model of jewelry.")
 uploaded_file = st.file_uploader("Choose an OBJ file", type="obj")
 
 if uploaded_file is not None:
-    # Save the uploaded file to a temporary location
-    with open("uploaded_model.obj", "wb") as f:
-        f.write(uploaded_file.getbuffer())
-
-    # Load the mesh using Open3D
-    mesh = o3d.io.read_triangle_mesh("uploaded_model.obj")
-    mesh.compute_vertex_normals()
-    
-    # Convert the mesh to a visualizable format for Streamlit
-    vertices = mesh.vertices
-    triangles = mesh.triangles
-    vertex_colors = mesh.vertex_colors
-
-    st.write("### 3D Model Preview")
-    st.write("Mesh Vertices: ", len(vertices))
-    st.write("Mesh Triangles: ", len(triangles))
-    
-    # Display the 3D mesh using Streamlit's 3D visualization support
-    st.write("You can download Open3D and use it to visualize the mesh locally.")
-    
-    # Optionally, display the mesh information
-    st.write(mesh)
+    try:
+        # Load the mesh using trimesh
+        mesh = trimesh.load(uploaded_file, force='mesh')
+
+        # Check if mesh is valid
+        if mesh.is_empty:
+            st.error("The OBJ file does not contain valid geometry. Please upload a different file.")
+        else:
+            # Get vertices and faces from the mesh
+            vertices = mesh.vertices
+            faces = mesh.faces
+
+            # Create geometry and material for the 3D model
+            geometry = BufferGeometry(
+                attributes={
+                    'position': BufferAttribute(vertices, normalized=False),
+                    'index': BufferAttribute(faces.flatten(), normalized=False)
+                }
+            )
+
+            material = MeshStandardMaterial(color='gold', roughness=0.5, metalness=0.8)
+            model = Mesh(geometry, material)
+
+            # Setup the scene
+            camera = PerspectiveCamera(position=[3, 3, 3], fov=75)
+            scene = Scene(children=[model, camera, AmbientLight(color='#777777')])
+            renderer = Renderer(camera=camera, scene=scene, controls=[OrbitControls(controlling=camera)],
+                                width=800, height=600)
+
+            st.write("### 3D Model Preview")
+            st.write(renderer)
+            
+            # Print success message
+            st.success("The OBJ file was successfully loaded and visualized.")
+
+    except Exception as e:
+        # Print error message
+        st.error(f"An error occurred while loading the OBJ file: {str(e)}")
 
 else:
     st.write("Please upload an OBJ file to visualize the jewelry model.")