Update app.py

app.py

@@ -169,6 +169,7 @@ def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,t
                 video_path = f"{TMP_DIR}/{trial_id}.mp4"
                 os.makedirs(os.path.dirname(video_path), exist_ok=True)
                 imageio.mimsave(video_path, video, fps=15)
+                yield _chatbot,fig,viewer_mesh,task_new,video_path
 
                 glb = postprocessing_utils.to_glb(
                     outputs['gaussian'][0],
@@ -177,9 +178,9 @@ def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,t
                     texture_size=1024,    
                     verbose=False  
                 )
-                glb.export(f"temper.glb")
+                glb.export(f"{TMP_DIR}/{trial_id}.glb")
                 print("processing mesh over...")
-                yield _chatbot,fig,"temper.glb",task_new,video_path
+                yield _chatbot,fig,f"{TMP_DIR}/{trial_id}.glb",task_new,video_path
             else:
                 # image to 3d
                 with torch.no_grad():
@@ -195,6 +196,7 @@ def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,t
                 video_path = f"{TMP_DIR}/{trial_id}.mp4"
                 os.makedirs(os.path.dirname(video_path), exist_ok=True)
                 imageio.mimsave(video_path, video, fps=15)
+                yield _chatbot,fig,viewer_mesh,task_new,video_path
                 
                 glb = postprocessing_utils.to_glb(
                     outputs['gaussian'][0],
@@ -203,11 +205,11 @@ def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,t
                     texture_size=1024,    
                     verbose=False  
                 )
-                glb.export(f"temper.glb")
+                glb.export(f"{TMP_DIR}/{trial_id}.glb")
                 print("processing mesh over...")
-                yield _chatbot,fig,"temper.glb",task_new,video_path,video_path
-        except:
-            print("processing mesh...bug")
+                yield _chatbot,fig,f"{TMP_DIR}/{trial_id}.glb",task_new,video_path
+        except Exception as e:
+            print(e)
             yield _chatbot,fig,viewer_mesh,task_new,video_path
 
 def regenerate(_chatbot, task_history):
@@ -329,8 +331,6 @@ def make_pointcloud_figure(verts,rotate=False):
         projection=dict(type="orthographic")  
     )
 
-    print(len(verts[:, 0].tolist()))
-    print(verts[:, 0].tolist())
     scatter = go.Scatter3d(
         x=verts[:, 0].tolist(),
         y=verts[:, 1].tolist(),