Update app.py

app.py

@@ -90,12 +90,12 @@ def generate_flux_image(
     guidance_scale: float,
     req: gr.Request,
     progress: gr.Progress = gr.Progress(track_tqdm=True),
-) -> Tuple[Image.Image, Image.Image]:
+) -> Image.Image:
     """Generate image using Flux pipeline"""
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
-    prompt = "wbgmsst, " + prompt + ", 3D isometric, white background"
+    prompt = "wbgmsst, " + prompt + ", 3D isometric, black background"
     image = flux_pipeline(
         prompt=prompt,
         guidance_scale=guidance_scale,
@@ -114,7 +114,6 @@ def generate_flux_image(
     filename = f"{timestamp}_{unique_id}.png"
     filepath = os.path.join(user_dir, filename)
     image.save(filepath)
-    print(f"Imagen guardada en: {filepath}")
     
     return image
 
@@ -252,10 +251,6 @@ with gr.Blocks() as demo:
         get_seed,
         inputs=[randomize_seed, seed],
         outputs=[seed],
-    ).then(
-        preprocess_image,
-        inputs=[generated_image],
-        outputs=[generated_image],
     ).then(
         image_to_3d,
         inputs=[
@@ -272,6 +267,11 @@ with gr.Blocks() as demo:
         outputs=[extract_glb_btn, extract_gs_btn],
     )
 
+    video_output.clear(
+        lambda: tuple([gr.Button(interactive=False), gr.Button(interactive=False)]),
+        outputs=[extract_glb_btn, extract_gs_btn],
+    )
+    
     # Extraer GLB
     extract_glb_btn.click(
         extract_glb,
@@ -291,6 +291,11 @@ with gr.Blocks() as demo:
         lambda: gr.Button(interactive=True),
         outputs=[download_gs],
     )
+
+    model_output.clear(
+        lambda: gr.Button(interactive=False),
+        outputs=[download_glb],
+    )
     
 # Initialize both pipelines
 if __name__ == "__main__":