Logging in app

app.py

@@ -16,18 +16,33 @@ from trellis.utils import render_utils, postprocessing_utils
 
 import traceback
 import sys
+import logging
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - HF_SPACE - %(levelname)s - %(message)s')
 
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
 
 def start_session(req: gr.Request):
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    session_hash = str(req.session_hash)
+    user_dir = os.path.join(TMP_DIR, session_hash)
+    logging.info(f"START SESSION: Creando directorio para la sesión {session_hash} en {user_dir}")
     os.makedirs(user_dir, exist_ok=True)
       
 def end_session(req: gr.Request):
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
-    shutil.rmtree(user_dir)
+    session_hash = str(req.session_hash)
+    user_dir = os.path.join(TMP_DIR, session_hash)
+    logging.info(f"END SESSION: Intentando eliminar el directorio de la sesión {session_hash} en {user_dir}")
+    # Hacemos la eliminación más robusta.
+    if os.path.exists(user_dir):
+        try:
+            shutil.rmtree(user_dir)
+            logging.info(f"Directorio de la sesión {session_hash} eliminado correctamente.")
+        except Exception as e:
+            logging.error(f"Error al eliminar el directorio de la sesión {session_hash}: {e}")
+    else:
+        logging.warning(f"El directorio de la sesión {session_hash} no fue encontrado al intentar eliminarlo. Es posible que ya haya sido limpiado.")
 
 def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
     return {
@@ -68,7 +83,9 @@ def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
     return gs, mesh
 
 def get_seed(randomize_seed: bool, seed: int) -> int:
-    return np.random.randint(0, MAX_SEED) if randomize_seed else seed
+    new_seed = np.random.randint(0, MAX_SEED) if randomize_seed else seed
+    logging.info(f"Usando seed: {new_seed}")
+    return new_seed
 
 @spaces.GPU
 def text_to_3d(
@@ -80,7 +97,10 @@ def text_to_3d(
     slat_sampling_steps: int,
     req: gr.Request,
 ) -> Tuple[dict, str]:
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    session_hash = str(req.session_hash)
+    logging.info(f"[{session_hash}] Iniciando text_to_3d con prompt: '{prompt[:50]}...'")
+    user_dir = os.path.join(TMP_DIR, session_hash)
+    
     outputs = pipeline.run(
         prompt,
         seed=seed,
@@ -94,13 +114,17 @@ def text_to_3d(
             "cfg_strength": slat_guidance_strength,
         },
     )
+    
+    logging.info(f"[{session_hash}] Generación del modelo completada. Renderizando video...")
     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')
     imageio.mimsave(video_path, video, fps=15)
+    
     state = pack_state(outputs['gaussian'][0], outputs['mesh'][0])
     torch.cuda.empty_cache()
+    logging.info(f"[{session_hash}] Video renderizado y estado empaquetado. Devolviendo: {video_path}")
     return state, video_path
 
 @spaces.GPU(duration=90)
@@ -110,12 +134,17 @@ def extract_glb(
     texture_size: int,
     req: gr.Request,
 ) -> Tuple[str, str]:
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    session_hash = str(req.session_hash)
+    logging.info(f"[{session_hash}] Iniciando extract_glb...")
+    user_dir = os.path.join(TMP_DIR, session_hash)
+    
     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)
+    
     torch.cuda.empty_cache()
+    logging.info(f"[{session_hash}] GLB extraído. Devolviendo: {glb_path}")
     return glb_path, glb_path
 
 @spaces.GPU