Update app.py

app.py

@@ -96,6 +96,7 @@ 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)
@@ -116,7 +117,7 @@ def generate_flux_image(
     filepath = os.path.join(SAVE_DIR, filename)
     image.save(filepath)
     # Devolver la imagen generada y el estado temporal (misma imagen)
-    return image, image
+    return image
 
 @spaces.GPU
 def image_to_3d(
@@ -231,10 +232,6 @@ with gr.Blocks() as demo:
         mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
         texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
 
-    # Estado para almacenar la imagen generada temporalmente
-    temp_image_state = gr.State()
-    output_buf = gr.State()
-
     # Event handlers
     demo.load(start_session)
     demo.unload(end_session)
@@ -243,7 +240,7 @@ with gr.Blocks() as demo:
     generate_image_btn.click(
         generate_flux_image,
         inputs=[prompt, seed, randomize_seed, width, height, guidance_scale],
-        outputs=[generated_image, temp_image_state],  # Almacenar la imagen en el estado temporal
+        outputs=[generated_image]
     ).then(
         lambda: gr.Button(interactive=True),  # Habilitar el botón "Generar Video"
         outputs=[generate_video_btn],
@@ -251,6 +248,10 @@ with gr.Blocks() as demo:
 
     # Generar video
     generate_video_btn.click(
+        get_seed,
+        inputs=[randomize_seed, seed],
+        outputs=[seed],
+    ).then(
         image_to_3d,
         inputs=[
             generated_image,