Update app.py

app.py

@@ -332,11 +332,53 @@ def move_to_device(model, device):
         
 @spaces.GPU
 def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[str, str]:
-    gs, mesh, trial_id = unpack_state(state)
-    glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
-    glb_path = f"{TMP_DIR}/{trial_id}.glb"
-    glb.export(glb_path)
-    return glb_path, glb_path
+    try:
+        # GPU 메모리 정리
+        clear_gpu_memory()
+        
+        # 상태 언패킹
+        gs, mesh, trial_id = unpack_state(state)
+        
+        # GLB 변환 전 유효성 검사
+        if gs is None or mesh is None:
+            print("Error: Invalid gaussian or mesh data")
+            return None, None
+            
+        # GLB 변환
+        with torch.inference_mode():
+            glb = postprocessing_utils.to_glb(
+                gs, 
+                mesh, 
+                simplify=mesh_simplify, 
+                texture_size=texture_size, 
+                verbose=True  # 디버깅을 위해 verbose 모드 활성화
+            )
+            
+        if glb is None:
+            print("Error: GLB conversion failed")
+            return None, None
+            
+        # 파일 저장
+        glb_path = f"{TMP_DIR}/{trial_id}.glb"
+        try:
+            glb.export(glb_path)
+            if not os.path.exists(glb_path):
+                print(f"Error: GLB file was not created at {glb_path}")
+                return None, None
+        except Exception as e:
+            print(f"Error saving GLB file: {str(e)}")
+            return None, None
+            
+        print(f"Successfully created GLB file at: {glb_path}")
+        return glb_path, glb_path
+        
+    except Exception as e:
+        print(f"Error in extract_glb: {str(e)}")
+        return None, None
+        
+    finally:
+        # 정리 작업
+        clear_gpu_memory()
 
 
 def activate_button() -> gr.Button: