preprocess image

app.py

@@ -58,11 +58,14 @@ def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     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)
 
+def preprocess_image(image: Image.Image) -> Image.Image:
+    processed_image = pipeline.preprocess_image(image)
+    return processed_image
+
 
 
 @spaces.GPU()
@@ -449,10 +452,12 @@ def create_interface():
                 max_length=2048,
             )
             visual_button = gr.Button("Create visual with Flux")
-
             generated_image = gr.Image(show_label=False)
-            gen3d_button = gr.Button("Create 3D visual with Trellis")
 
+            preprocessed_button = gr.Button("Create 3D visual with Trellis")
+            preprocessed_image = gr.Image(show_label=False)
+
+            gen3d_button = gr.Button("Create 3D visual with Trellis")
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
             # model_output = LitModel3D(label="Extracted GLB/Gaussian", exposure=10.0, height=300)
             
@@ -550,10 +555,17 @@ def create_interface():
             outputs=[generated_image, message_box]
         )
 
+        gr.on(
+            triggers=[preprocessed_button.click],
+            fn=preprocess_image,
+            inputs=[generated_image],
+            outputs=[preprocessed_image, message_box]
+        )
+
         gr.on(
             triggers=[gen3d_button.click],
             fn=image_to_3d,
-            inputs=[generated_image, trellis_seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
+            inputs=[preprocessed_image, trellis_seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps],
             outputs=[output_state, video_output],
         ).then(
             # Update button states after successful 3D generation