Update app.py

app.py

@@ -92,7 +92,6 @@ def generate_flux_image(
     progress: gr.Progress = gr.Progress(track_tqdm=True),
 ) -> Tuple[Image.Image, Image.Image]:
     """Generate image using Flux pipeline"""
-    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
@@ -106,13 +105,17 @@ def generate_flux_image(
         generator=generator,
     ).images[0]
     
-    # Save the generated image
+    # Guardar la imagen en el directorio temporal
+    user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    os.makedirs(user_dir, exist_ok=True)  # Asegúrate de que el directorio exista
+    
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     unique_id = str(uuid.uuid4())[:8]
     filename = f"{timestamp}_{unique_id}.png"
-    filepath = os.path.join(TMP_DIR, filename)
+    filepath = os.path.join(user_dir, filename)
     image.save(filepath)
-    # Devolver la imagen generada y el estado temporal (misma imagen)
+    print(f"Imagen guardada en: {filepath}")
+    
     return image
 
 @spaces.GPU