Update app.py

app.py

@@ -24,12 +24,12 @@ os.makedirs(CUSTOM_SAVE_DIR, exist_ok=True)
 
 
 def start_session(req: gr.Request):
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    user_dir = os.path.join(CUSTOM_SAVE_DIR, str(req.session_hash))
     os.makedirs(user_dir, exist_ok=True)
     
     
 def end_session(req: gr.Request):
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    user_dir = os.path.join(CUSTOM_SAVE_DIR, str(req.session_hash))
     shutil.rmtree(user_dir)
 
 
@@ -224,7 +224,7 @@ def extract_gaussian(state: dict, req: gr.Request) -> Tuple[str, str]:
     Returns:
         Tuple[str, str]: Paths to the extracted Gaussian file (for display and download)
     """
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    user_dir = os.path.join(CUSTOM_SAVE_DIR, str(req.session_hash))
     gs, _ = unpack_state(state)
     gaussian_path = os.path.join(user_dir, 'sample.ply')
     gs.save_ply(gaussian_path)