Fix Spaces file serving and LitModel3D output path wiring

app.py

@@ -71,6 +71,9 @@ if TRELLIS_RUNTIME not in {"original", "trellis2"}:
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join("cache")
 os.makedirs(TMP_DIR, exist_ok=True)
+TMP_DIR_ABS = os.path.abspath(TMP_DIR)
+ASSETS_DIR_ABS = os.path.abspath("assets")
+ALLOWED_PATHS = [TMP_DIR_ABS, ASSETS_DIR_ABS]
 
 if gr.NO_RELOAD:
     pipeline = None
@@ -646,9 +649,9 @@ def extract_model_from_state(
     texture_size: int,
     req: gr.Request,
     progress=gr.Progress(track_tqdm=True),
-) -> Tuple[Dict[str, Any], Dict[str, Any]]:
+) -> Tuple[Optional[str], Dict[str, Any]]:
     if state is None:
-        return gr.update(value=None, visible=True), gr.update(value=None, visible=True, interactive=False)
+        return None, gr.update(value=None, visible=True, interactive=False)
 
     if not torch.cuda.is_available():
         raise gr.Error("GPU is not ready. Please retry in a few seconds.")
@@ -717,12 +720,16 @@ def extract_model_from_state(
 
         stl_start = time.time()
         stl_path = export_stl_from_glb(glb_path_abs)
+        if stl_path is not None:
+            stl_path = os.path.abspath(stl_path)
         logger.info("STL Export Time: %.2fs", time.time() - stl_start)
         logger.info("Phase B Total Time: %.2fs", time.time() - job_start)
+        logger.info("Returning model file: %s", glb_path_abs)
+        logger.info("Returning STL file: %s", stl_path)
 
-        model_update = gr.update(value=glb_path_abs, visible=True)
         stl_update = gr.update(value=stl_path, visible=True, interactive=bool(stl_path))
-        return model_update, stl_update
+        # LitModel3D works most reliably when the output is the raw path string.
+        return glb_path_abs, stl_update
     except RuntimeError as re:
         if "out of memory" in str(re).lower():
             clear_cuda_cache()
@@ -917,7 +924,7 @@ if __name__ == "__main__":
         demo.queue(max_size=8, default_concurrency_limit=1, api_open=False).launch(
             show_api=False,
             share=False,
-            allowed_paths=["./cache", "./assets"],
+            allowed_paths=ALLOWED_PATHS,
         )
     elif prod:
         logger.info("Launching in PRODUCTION mode on port %s", port)
@@ -927,7 +934,7 @@ if __name__ == "__main__":
             show_api=False,
             favicon_path="assets/sb_3d_ai_logo.png",
             share=False,
-            allowed_paths=["./cache", "./assets"],
+            allowed_paths=ALLOWED_PATHS,
         )
     else:
         logger.info("Launching in DEVELOPMENT mode on port %s", port)
@@ -938,5 +945,5 @@ if __name__ == "__main__":
             favicon_path="assets/sb_3d_ai_logo.png",
             debug=True,
             share=True,
-            allowed_paths=["./cache", "./assets"],
+            allowed_paths=ALLOWED_PATHS,
         )