extract_glb modif (relative path)

app.py

@@ -168,25 +168,18 @@ def extract_glb(
     texture_size: int,
     req: gr.Request,
 ) -> Tuple[str, str]:
-    """
-    Extract a GLB file from the 3D model.
-
-    Args:
-        state (dict): The state of the generated 3D model.
-        mesh_simplify (float): The mesh simplification factor.
-        texture_size (int): The texture resolution.
-
-    Returns:
-        str: The path to the extracted GLB file.
-    """
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     os.makedirs(user_dir, exist_ok=True)
     gs, mesh = unpack_state(state)
     glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
-    glb_path = os.path.join(user_dir, 'sample.glb')
-    glb.export(glb_path)
+    # Save the GLB file to an absolute path
+    glb_path_absolute = os.path.join(user_dir, 'sample.glb')
+    glb.export(glb_path_absolute)
     torch.cuda.empty_cache()
-    return glb_path, glb_path
+    # Compute a relative path that Gradio can serve, e.g. "tmp/<session_hash>/sample.glb"
+    relative_path = os.path.join("tmp", str(req.session_hash), "sample.glb")
+    return relative_path, relative_path
+
 
 
 def prepare_multi_example() -> List[Image.Image]: