Update app.py

app.py

@@ -25,6 +25,7 @@ else:
         "runwayml/stable-diffusion-inpainting"
     ).to(device)
 
+
 DEFAULT_OBJ_FILE = "female.obj"
 DEFAULT_GLB_FILE = "vroid_girl1.glb"
 DEFAULT_VRM_FILE = "fischl.vrm"
@@ -229,6 +230,20 @@ def update_texture_display(prompt, texture_file, num_inference_steps):
         return Image.open(texture_file)
     return None
 
+def load_example(obj_file, texture_file):
+    """Loads and displays an example 3D object with texture."""
+    file_extension = obj_file.split('.')[-1].lower()
+    texture_image = None
+    if texture_file:
+        texture_image = Image.open(texture_file)
+
+    if file_extension == 'vrm':
+        return display_3d_object(obj_file, texture_image, 0.8, 0.5, "#D3D3D3", 1.0, 1.0, 'medium')  # Using default values for other parameters
+    else:
+        return display_3d_object(obj_file, texture_image, 0.8, 0.5, "#D3D3D3", 1.0, 1.0)  # Using default values for other parameters
+
+
+
 with gr.Blocks() as demo:
     gr.Markdown("## 3D Object Viewer with Custom Texture, UV Scale, Transparency, Color, and Adjustable Lighting")
 
@@ -288,17 +303,17 @@ with gr.Blocks() as demo:
     texture_file.change(fn=update_texture_display, inputs=[prompt_input, texture_file, num_inference_steps_slider], outputs=texture_preview)
 
     demo.load(fn=update_display, inputs=[obj_file, texture_file, uv_scale_slider, uv_quality_dropdown, light_intensity_slider, ambient_intensity_slider, transparency_slider, color_picker, num_inference_steps_slider], outputs=display)
-
-
-
-
     gr.Examples(
-        examples=[[DEFAULT_VRM_FILE, DEFAULT_TEXTURE],
-                  [DEFAULT_OBJ_FILE, None],
-                  [DEFAULT_GLB_FILE, None],
-                  [DEFAULT_VRM_FILE2, DEFAULT_TEXTURE2],
-                  [DEFAULT_VRM_FILE3, DEFAULT_TEXTURE3]],
+        examples=[
+            [DEFAULT_VRM_FILE, DEFAULT_TEXTURE],
+            [DEFAULT_OBJ_FILE, None],
+            [DEFAULT_GLB_FILE, None],
+            [DEFAULT_VRM_FILE2, DEFAULT_TEXTURE2],
+            [DEFAULT_VRM_FILE3, DEFAULT_TEXTURE3]
+        ],
         inputs=[obj_file, texture_file],
+        outputs=display,  # Specify the output component
+        fn=load_example,  # Specify the function to load the example
         label="Example Files"
     )