Update app.py

app.py

@@ -113,6 +113,12 @@ def preprocess_image(image: Image.Image) -> Tuple[str, Image.Image]:
         return "", None
         
     try:
+        # webp 이미지를 RGB로 변환
+        if isinstance(image, str) and image.endswith('.webp'):
+            image = Image.open(image).convert('RGB')
+        elif isinstance(image, Image.Image):
+            image = image.convert('RGB')
+            
         trial_id = str(uuid.uuid4())
         processed_image = g.trellis_pipeline.preprocess_image(image)
         if processed_image is not None:
@@ -253,10 +259,17 @@ def text_to_image(prompt: str, height: int, width: int, steps: int, scales: floa
         print(f"Error in image generation: {str(e)}")
         return None
 
-# Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""## Craft3D""")
     
+    # Examples 이미지 로드
+    example_dir = "assets/example_image/"
+    example_images = []
+    if os.path.exists(example_dir):
+        for file in os.listdir(example_dir):
+            if file.endswith('.webp'):
+                example_images.append(os.path.join(example_dir, file))
+    
     with gr.Row():
         with gr.Column():
             text_prompt = gr.Textbox(
@@ -265,6 +278,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 lines=3
             )
             
+            # Examples 갤러리 추가
+            if example_images:
+                gr.Gallery(
+                    value=example_images,
+                    label="Example Images",
+                    show_label=True,
+                    elem_id="gallery",
+                    columns=[3],
+                    rows=[2],
+                    height=200,
+                    allow_preview=True
+                )
+                
+            
             with gr.Accordion("Image Generation Settings", open=False):
                 with gr.Row():
                     height = gr.Slider(
@@ -316,6 +343,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 slat_sampling_steps = gr.Slider(1, 50, label="Latent Sampling Steps", value=12, step=1)
 
             generate_3d_btn = gr.Button("Generate 3D")
+
+            gr.Examples(
+                examples=example_images,
+                inputs=image_prompt,
+                outputs=[trial_id, image_prompt],
+                fn=preprocess_image,
+                cache_examples=True,
+                label="Click on an example to load it"
+            )
+
             
             with gr.Accordion("GLB Extraction Settings", open=False):
                 mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)