fix user dir errors from API

app.py

@@ -105,7 +105,11 @@ def text_to_3d(
         dict: The information of the generated 3D model.
         str: The path to the video of the 3D model.
     """
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    # --- Determine user_dir robustly ---
+    session_hash_str = str(req.session_hash) if hasattr(req, 'session_hash') and req.session_hash else f"api_call_{np.random.randint(10000)}"
+    user_dir = os.path.join(TMP_DIR, session_hash_str)
+    os.makedirs(user_dir, exist_ok=True) # Ensure directory exists
+
     outputs = pipeline.run(
         prompt,
         seed=seed,
@@ -123,7 +127,13 @@ def text_to_3d(
     video_geo = render_utils.render_video(outputs['mesh'][0], num_frames=120)['normal']
     video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
     video_path = os.path.join(user_dir, 'sample.mp4')
-    imageio.mimsave(video_path, video, fps=15)
+    try:
+        imageio.mimsave(video_path, video, fps=15) # Now the directory should exist
+    except FileNotFoundError:
+         print(f"ERROR: Directory {user_dir} still not found before mimsave!", file=sys.stderr)
+         # Decide if we should raise or return an error state?
+         # Returning a dummy path might hide the error, so let's raise for now
+         raise
     state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
     torch.cuda.empty_cache()
     return state, video_path
@@ -147,11 +157,19 @@ def extract_glb(
     Returns:
         str: The path to the extracted GLB file.
     """
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    # --- Determine user_dir robustly ---
+    session_hash_str = str(req.session_hash) if hasattr(req, 'session_hash') and req.session_hash else f"api_call_{np.random.randint(10000)}"
+    user_dir = os.path.join(TMP_DIR, session_hash_str)
+    os.makedirs(user_dir, exist_ok=True) # Ensure directory exists
+
     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)
+    try:
+        glb.export(glb_path) # Now the directory should exist
+    except FileNotFoundError:
+        print(f"ERROR: Directory {user_dir} still not found before glb.export!", file=sys.stderr)
+        raise
     torch.cuda.empty_cache()
     return glb_path, glb_path
 
@@ -167,10 +185,18 @@ def extract_gaussian(state: dict, req: gr.Request) -> Tuple[str, str]:
     Returns:
         str: The path to the extracted Gaussian file.
     """
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    # --- Determine user_dir robustly ---
+    session_hash_str = str(req.session_hash) if hasattr(req, 'session_hash') and req.session_hash else f"api_call_{np.random.randint(10000)}"
+    user_dir = os.path.join(TMP_DIR, session_hash_str)
+    os.makedirs(user_dir, exist_ok=True) # Ensure directory exists
+
     gs, _ = unpack_state(state)
     gaussian_path = os.path.join(user_dir, 'sample.ply')
-    gs.save_ply(gaussian_path)
+    try:
+        gs.save_ply(gaussian_path) # Now the directory should exist
+    except FileNotFoundError:
+        print(f"ERROR: Directory {user_dir} still not found before gs.save_ply!", file=sys.stderr)
+        raise
     torch.cuda.empty_cache()
     return gaussian_path, gaussian_path