refactor session handling to use a fixed temporary directory and remove session management functions

app.py

@@ -53,27 +53,6 @@ _image_gen_pipeline = None
 _trellis_pipeline = None
 
 
-def start_session(req: gr.Request):
-    """Create a temporary directory for the user session"""
-    try:
-        user_dir = os.path.join(TMP_DIR, str(req.session_hash))
-        os.makedirs(user_dir, exist_ok=True)
-        print(f"Session started: {req.session_hash}")
-    except Exception as e:
-        print(f"Error starting session: {str(e)}")
-    
-    
-def end_session(req: gr.Request):
-    """Clean up the temporary directory when the session ends"""
-    try:
-        user_dir = os.path.join(TMP_DIR, str(req.session_hash))
-        if os.path.exists(user_dir):
-            shutil.rmtree(user_dir)
-            print(f"Session ended: {req.session_hash}")
-    except Exception as e:
-        print(f"Error ending session: {str(e)}")
-
-
 @spaces.GPU()
 def get_image_gen_pipeline():
     global _image_gen_pipeline
@@ -320,10 +299,11 @@ def image_to_3d(
     ss_sampling_steps: int,
     slat_guidance_strength: float,
     slat_sampling_steps: int,
-    req: gr.Request,
 ) -> Tuple[dict, str]:
     try:
-        user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+        # Use a fixed temp directory instead of user-specific
+        temp_dir = os.path.join(TMP_DIR, "temp_output")
+        os.makedirs(temp_dir, exist_ok=True)
         
         # Get the pipeline using the getter function
         pipeline = get_trellis_pipeline()
@@ -348,7 +328,7 @@ def image_to_3d(
         video = render_utils.render_video(outputs['gaussian'][0], num_frames=120)['color']
         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')
+        video_path = os.path.join(temp_dir, 'sample.mp4')
         imageio.mimsave(video_path, video, fps=15)
         state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
         torch.cuda.empty_cache()
@@ -363,7 +343,6 @@ def extract_glb(
     state: dict,
     mesh_simplify: float,
     texture_size: int,
-    req: gr.Request,
 ) -> Tuple[str, str]:
     """
     Extract a GLB file from the 3D model.
@@ -376,17 +355,17 @@ def extract_glb(
     Returns:
         str: The path to the extracted GLB file.
     """
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    temp_dir = os.path.join(TMP_DIR, "temp_output")
     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_path = os.path.join(temp_dir, 'sample.glb')
     glb.export(glb_path)
     torch.cuda.empty_cache()
     return glb_path, glb_path
 
 
 @spaces.GPU
-def extract_gaussian(state: dict, req: gr.Request) -> Tuple[str, str]:
+def extract_gaussian(state: dict) -> Tuple[str, str]:
     """
     Extract a Gaussian file from the 3D model.
 
@@ -396,9 +375,9 @@ 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))
+    temp_dir = os.path.join(TMP_DIR, "temp_output")
     gs, _ = unpack_state(state)
-    gaussian_path = os.path.join(user_dir, 'sample.ply')
+    gaussian_path = os.path.join(temp_dir, 'sample.ply')
     gs.save_ply(gaussian_path)
     torch.cuda.empty_cache()
     return gaussian_path, gaussian_path
@@ -430,10 +409,6 @@ def create_interface():
         model_status = "ℹ️ Models will be loaded on demand"
 
     with gr.Blocks(css=css) as demo:
-        # Set up session management - COMMENT THESE OUT FOR TESTING
-        # demo.load(start_session)
-        # demo.unload(end_session)
-
         gr.Info(model_status)
         
         # State for storing 3D model data
@@ -604,4 +579,4 @@ if __name__ == "__main__":
         print(status)
         
     demo = create_interface()
-    demo.launch()
+    demo.launch()