eliphatfs commited on
Commit
44a6c30
·
1 Parent(s): 1caa49c
Files changed (1) hide show
  1. openshape/demo/misc_utils.py +11 -4
openshape/demo/misc_utils.py CHANGED
@@ -99,10 +99,17 @@ def trimesh_to_pc(scene_or_mesh):
99
 
100
 
101
  def input_3d_shape(key=None):
102
- objaid = st.text_input("Enter an Objaverse ID", key=key)
103
- model = st.file_uploader("Or upload a model (.glb/.obj/.ply)", key=key)
104
- npy = st.file_uploader("Or upload a point cloud numpy array (.npy of Nx3 XYZ or Nx6 XYZRGB)", key=key)
105
- swap_yz_axes = st.checkbox("Swap Y/Z axes of input (Y is up for OpenShape)", key=key)
 
 
 
 
 
 
 
106
  f32 = numpy.float32
107
 
108
  def load_data(prog):
 
99
 
100
 
101
  def input_3d_shape(key=None):
102
+ if key is None:
103
+ objaid_key = model_key = npy_key = swap_key = None
104
+ else:
105
+ objaid_key = key + "_objaid"
106
+ model_key = key + "_model"
107
+ npy_key = key + "_npy"
108
+ swap_key = key + "_swap"
109
+ objaid = st.text_input("Enter an Objaverse ID", key=objaid_key)
110
+ model = st.file_uploader("Or upload a model (.glb/.obj/.ply)", key=model_key)
111
+ npy = st.file_uploader("Or upload a point cloud numpy array (.npy of Nx3 XYZ or Nx6 XYZRGB)", key=npy_key)
112
+ swap_yz_axes = st.checkbox("Swap Y/Z axes of input (Y is up for OpenShape)", key=swap_key)
113
  f32 = numpy.float32
114
 
115
  def load_data(prog):