Update app.py

app.py

@@ -122,7 +122,9 @@ def preprocess_image(image: Image.Image) -> Tuple[str, Image.Image]:
         trial_id = str(uuid.uuid4())
         processed_image = g.trellis_pipeline.preprocess_image(image)
         if processed_image is not None:
-            processed_image.save(f"{TMP_DIR}/{trial_id}.png")
+            save_path = f"{TMP_DIR}/{trial_id}.png"
+            processed_image.save(save_path)
+            print(f"Saved processed image to: {save_path}")
             return trial_id, processed_image
         else:
             print("Error: Processed image is None")
@@ -173,7 +175,9 @@ def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
 
 @spaces.GPU
 def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_strength: float, ss_sampling_steps: int, slat_guidance_strength: float, slat_sampling_steps: int) -> Tuple[dict, str]:
-    if not trial_id:  # trial_id 검증 추가
+    print(f"Starting image_to_3d with trial_id: {trial_id}")
+    
+    if not trial_id or trial_id.strip() == "":
         print("Error: No trial_id provided")
         return None, None
         
@@ -182,11 +186,15 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
             seed = np.random.randint(0, MAX_SEED)
             
         image_path = f"{TMP_DIR}/{trial_id}.png"
+        print(f"Looking for image at: {image_path}")
+        
         if not os.path.exists(image_path):
             print(f"Error: Image file not found at {image_path}")
             return None, None
             
         image = Image.open(image_path)
+        print(f"Successfully loaded image with size: {image.size}")
+        
         
         outputs = g.trellis_pipeline.run(
             image,
@@ -295,34 +303,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 lines=3
             )
             
-            # 이미지 프롬프트를 갤러리 전에 정의
+            # 이미지 프롬프트
             image_prompt = gr.Image(label="Image Prompt", image_mode="RGBA", type="pil", height=300)
             
-            # Examples 갤러리를 image_prompt 아래로 이동
-            if example_images:
-                with gr.Row():
-                    gallery = gr.Gallery(
-                        value=example_images,
-                        label="Example Images",
-                        show_label=True,
-                        elem_id="gallery",
-                        columns=5,
-                        rows=5,
-                        height=200,
-                        allow_preview=True
-                    )
-                    
-                    # Gallery 클릭 이벤트 추가
-                    def load_example(evt: gr.SelectData):
-                        return example_images[evt.index]
-                    
-                    gallery.select(
-                        load_example,
-                        None,
-                        image_prompt,
-                        show_progress=True
-                    )
-            
             with gr.Accordion("Image Generation Settings", open=False):
                 with gr.Row():
                     height = gr.Slider(
@@ -383,10 +366,37 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
             model_output = LitModel3D(label="Extracted GLB", exposure=20.0, height=300)
             download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
-            
+    
     trial_id = gr.Textbox(visible=False)
     output_buf = gr.State()
 
+    # Examples 갤러리를 맨 아래로 이동
+    if example_images:
+        gr.Markdown("""### Example Images""")
+        with gr.Row():
+            gallery = gr.Gallery(
+                value=example_images,
+                label="Click an image to use it",
+                show_label=True,
+                elem_id="gallery",
+                columns=5,
+                rows=5,
+                height=600,
+                allow_preview=True
+            )
+            
+            def load_example(evt: gr.SelectData):
+                selected_image = Image.open(example_images[evt.index])
+                trial_id_val, processed_image = preprocess_image(selected_image)
+                return selected_image, trial_id_val
+            
+            gallery.select(
+                load_example,
+                None,
+                [image_prompt, trial_id],
+                show_progress=True
+            )
+
     # Handlers
     generate_image_btn.click(
         text_to_image,