Session and tmp clean up changes

app.py

@@ -128,18 +128,20 @@ depth_model: Optional[DPTForDepthEstimation] = None
 TRELLIS_PIPELINE: Optional[TrellisImageTo3DPipeline] = None
 pipe: Optional[Union[FluxPipeline, FluxImg2ImgPipeline, FluxControlPipeline]] = None
 
-# Register the cleanup function
-atexit.register(cleanup_temp_files)
-
 def start_session(req: gr.Request):
+    print(f"Starting session with hash: {req.session_hash}")
     user_dir = os.path.join(constants.TMPDIR, str(req.session_hash))
     os.makedirs(user_dir, exist_ok=True)
     
     
 def end_session(req: gr.Request):
+    print(f"Ending session with hash: {req.session_hash}")
     user_dir = os.path.join(constants.TMPDIR, str(req.session_hash))
     shutil.rmtree(user_dir)
 
+# Register the cleanup function
+atexit.register(end_session)
+
 def hex_create(hex_size, border_size, input_image_path, start_x, start_y, end_x, end_y, rotation, background_color_hex, background_opacity, border_color_hex, border_opacity, fill_hex, excluded_colors_var, filter_color, x_spacing, y_spacing, add_hex_text_option=None, custom_text_list=None, custom_text_color_list=None):
     global input_image_palette
 

utils/file_utils.py

@@ -10,6 +10,7 @@ def cleanup_temp_files():
             os.remove(file_path)
         except Exception as e:
             print(f"Failed to delete temp file {file_path}: {e}")
+    print("Temp files cleaned up. Goodbye")
 
 def rename_file_to_lowercase_extension(file_path: str) -> str:
     """