Update app.py

app.py

@@ -214,6 +214,9 @@ def extract_glb(
     Returns:
         str: The path to the extracted GLB file.
     """
+    if state is None:
+        raise gr.Error("Please generate a 3D model first before extracting GLB.")
+    
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     gs, mesh = unpack_state(state)
     glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
@@ -235,6 +238,9 @@ def extract_gaussian(state: dict, req: gr.Request) -> Tuple[str, str]:
     Returns:
         str: The path to the extracted Gaussian file.
     """
+    if state is None:
+        raise gr.Error("Please generate a 3D model first before extracting Gaussian.")
+    
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     gs, _ = unpack_state(state)
     gaussian_path = os.path.join(user_dir, 'sample.ply')
@@ -428,4 +434,4 @@ if __name__ == "__main__":
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))    # Preload rembg
     except:
         pass
-    demo.launch()
+    demo.launch(ssr_mode=False)