cpu won't work (show message)

.gitignore

@@ -0,0 +1 @@
+*.pyc

app.py

@@ -1,22 +1,54 @@
 import torch
 import sys
+import numpy as np
+import tempfile
+
 if torch.cuda.is_available():
-	try:
-		torch.ones(1, 1).cuda().float()
-		sys.path.append("/home/user/app/Dynamic3DGaussians")
-		from Dynamic3DGaussians.visualize import visualize
-	except:
-		sys.path.append("/home/user/app/Dynamic3DGaussians_cpu")
-		from Dynamic3DGaussians_cpu.visualize import visualize
+    try:
+        torch.ones(1, 1).cuda().float()
+        sys.path.insert(0,"/home/user/app/wrapper")
+        sys.path.append("/home/user/app/Dynamic3DGaussians")
+        from Dynamic3DGaussians.visualize import visualize
+        import open3d as o3d
+        IS_CPU = False
+    except Exception as e:
+        print(e)
+        IS_CPU = True
 else:
-	sys.path.append("/home/user/app/Dynamic3DGaussians_cpu")
-	from Dynamic3DGaussians_cpu.visualize import visualize
+    IS_CPU = True
 
 import gradio as gr
 
-def basketball(name):
-    visualize("basketball", "pretrained")
-    return "Hello " + name + "!!!"
+def basketball(what):
+    if IS_CPU:
+        return None, None, "Not possible on CPU"
+    visualize(what, "pretrained")
+    result = "Done"
+    pcd = o3d.get_object()
+    pcd.normals = o3d.utility.Vector3dVector(
+        np.zeros((1, 3)))  # invalidate existing normals
+    pcd.estimate_normals(
+        search_param=o3d.geometry.KDTreeSearchParamHybrid(radius=0.01, max_nn=30))
+    pcd.orient_normals_towards_camera_location(
+        camera_location=np.array([0., 0., 1000.]))
+    pcd.transform([[1, 0, 0, 0],
+                   [0, -1, 0, 0],
+                   [0, 0, -1, 0],
+                   [0, 0, 0, 1]])
+    pcd.transform([[-1, 0, 0, 0],
+                   [0, 1, 0, 0],
+                   [0, 0, 1, 0],
+                   [0, 0, 0, 1]])
+    mesh_raw, densities = o3d.geometry.TriangleMesh.create_from_point_cloud_poisson(
+            pcd, width=0, scale=1.1, linear_fit=True)
+    temp_name = next(tempfile._get_candidate_names())
+    file_path = f"/tmp/{temp_name}.gltf"
+    file_path_pcd = f"/tmp/{temp_name}.pcd"
+    if not o3d.io.write_triangle_mesh(file_path, mesh_raw):
+        result = "ERROR #1"
+    if not o3d.io.write_point_cloud(file_path_pcd, pcd):
+        result = "ERROR #2"
+    return file_path, file_path_pcd, result
 
-iface = gr.Interface(fn=basketball, inputs=None, outputs="text", allow_flagging="never")
+iface = gr.Interface(fn=basketball, inputs=gr.Dropdown(["basketball", "boxes", "football", "juggle", "softball", "tennis"], value="basketball"), outputs=[gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model"), gr.File(), gr.Textbox(label="debug output")], allow_flagging="never")
 iface.launch(server_name="0.0.0.0")

runLocalDocker.bat

@@ -1,2 +1,2 @@
-docker run --gpus=all --entrypoint "python" -v .:/home/user/app -p 7860:7860 --rm -ti dynamic3dgaussians /usr/local/bin/gradio app.py
+docker run --gpus=all --entrypoint "python" -v .\wrapper:/home/user/app/wrapper -v .\app.py:/home/user/app/app.py -p 7860:7860 --rm -ti registry.hf.space/crazyt-dynamic3dgaussians:latest /usr/local/bin/gradio app.py
 pause

wrapper/open3d/__init__.py

@@ -0,0 +1,63 @@
+from importlib import reload
+import sys
+import numpy as np
+if("wrapper" in sys.path[0]):
+	sys.path.remove(sys.path[0])
+import open3d
+reload(open3d)
+from open3d import visualization
+
+class ObjectHolder:
+    def __init__(self):
+        self.obj = None
+    
+objHolder = ObjectHolder()
+
+def get_object():
+    global objHolder
+    return objHolder.obj
+    
+class do_nothing_obj_simple:
+    def __init__(self):
+        self.intrinsic_matrix = np.array([[320, 0, 320], [0, 320, 180], [0, 0, 1]])
+class do_nothing_obj:
+    def __init__(self):
+        self.intrinsic = do_nothing_obj_simple()
+        self.extrinsic = np.array([[1., 0., 0., 0.],
+                                   [0., 1., 0., 0.],
+                                   [0., 0., 1., 0.],
+                                   [0., 0., 0., 1.]])
+
+class do_nothing_class_ViewControl:
+    def convert_from_pinhole_camera_parameters(self,*args,**kwargs):
+        pass
+    def convert_to_pinhole_camera_parameters(self,*args,**kwargs):
+        return do_nothing_obj()
+class proxy_class_Visualizer:
+    def __init__(self):
+        global orgVisualizer
+        self.geometries = []
+        self.delegate = orgVisualizer()
+    def get_render_option(self):
+        return do_nothing_obj()
+    def create_window(self,*args,**kwargs):
+        pass
+    def destroy_window(self):
+        pass
+    def update_renderer(self):
+        return self.update_renderer()
+    def get_view_control(self):
+        return do_nothing_class_ViewControl()
+    def add_geometry(self,pcd):
+        self.geometries.append(pcd)
+        return self.delegate.add_geometry(pcd)
+    def update_geometry(self,pcd):
+        global objHolder
+        objHolder.obj = pcd
+        return self.delegate.update_geometry(pcd)
+    def poll_events(self):
+        return self.delegate.poll_events()
+
+orgVisualizer = visualization.Visualizer
+visualization.Visualizer = proxy_class_Visualizer
+open3d.get_object = get_object