Update app.py

app.py

@@ -9,8 +9,7 @@ import numpy as np
 import shutil
 
 # --------------------------------------------------------------------------
-# BLOQUE 1: INSTALACIÓN DEL ENTORNO Y DEPENDENCIAS
-# Esta sección DEBE ejecutarse en su totalidad ANTES de importar torch o gradio.
+# BLOQUE 1: INSTALACIÓN DEL ENTORNO Y DEPENDENCIAS (Sin cambios)
 # --------------------------------------------------------------------------
 
 @spaces.GPU(duration=120)
@@ -67,7 +66,6 @@ def setup_environment():
 
     print("--- CONFIGURACIÓN DEL ENTORNO FINALIZADA ---")
 
-# Ejecutar la configuración del entorno antes de cualquier otra cosa
 setup_environment()
 
 # --------------------------------------------------------------------------
@@ -127,10 +125,11 @@ def end_session(req: gr.Request):
 # --------------------------------------------------------------------------
 
 def get_random_seed(randomize_seed, seed):
-    """Genera una semilla aleatoria si se solicita."""
+    """Función auxiliar para determinar la semilla a usar."""
     if randomize_seed:
-        return random.randint(0, MAX_SEED)
-    return int(seed)
+        seed = random.randint(0, MAX_SEED)
+    print(f"Usando semilla: {seed}")
+    return seed
 
 @spaces.GPU(duration=180)
 def generate_texture_for_user_mesh(
@@ -138,10 +137,9 @@ def generate_texture_for_user_mesh(
     input_mesh_path,
     guidance_scale,
     inference_steps,
-    seed,
-    randomize_seed,
+    seed, # <-- Ahora recibe la semilla final, no el checkbox
     reference_conditioning_scale,
-    req: gr.Request, # Se añade el objeto request para obtener el hash de la sesión
+    req: gr.Request,
     progress=gr.Progress(track_tqdm=True)
 ):
     """
@@ -156,8 +154,7 @@ def generate_texture_for_user_mesh(
     user_dir = os.path.join(TMP_DIR, session_hash)
     print(f"[{session_hash}] Iniciando generación de textura...")
     
-    # Obtener semilla
-    final_seed = get_random_seed(randomize_seed, seed)
+    final_seed = int(seed)
     
     # Actualizar la configuración del pipeline
     texture_model.config.guidance_scale = float(guidance_scale)
@@ -193,6 +190,7 @@ def generate_texture_for_user_mesh(
     
     return textured_save_path
 
+
 # --------------------------------------------------------------------------
 # 6. INTERFAZ DE USUARIO CON GRADIO
 # --------------------------------------------------------------------------
@@ -221,20 +219,21 @@ with gr.Blocks(title="Step1X-3D Texture Generator") as demo:
 
     # --- Lógica de la interfaz ---
     
-    # Conecta las funciones de sesión al ciclo de vida de la demo
-    demo.load(start_session)
-    demo.unload(end_session)
+    demo.load(fn=start_session)
+    demo.unload(fn=end_session)
 
-    # Conecta el botón a la función de generación
     btn_generate.click(
+        fn=get_random_seed,
+        inputs=[randomize_seed_checkbox, seed_input],
+        outputs=[seed_input] 
+    ).then(
         fn=generate_texture_for_user_mesh,
         inputs=[
             input_image,
             input_mesh,
             guidance_scale,
             inference_steps,
-            seed_input,
-            randomize_seed_checkbox,
+            seed_input, 
             reference_conditioning_scale,
         ],
         outputs=[output_model]