Update app.py

.gitattributes

@@ -34,6 +34,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 images/logo.png filter=lfs diff=lfs merge=lfs -text
+images/beeuty_545jlbh1_v12_alpha96_300dpi.png filter=lfs diff=lfs merge=lfs -text
+images/beeuty_545jlbh1_v12_alpha96_300dpi_depth.png filter=lfs diff=lfs merge=lfs -text
 *.glb filter=lfs diff=lfs merge=lfs -text
 *.gltf filter=lfs diff=lfs merge=lfs -text
 *.ply filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-def load_data(query_params):
+def load_data(query_params, model_3d):
     model_url = query_params.get("3d", None) if query_params else None
     hm_url = query_params.get("hm", None) if query_params else None
     img_url = query_params.get("image", None) if query_params else None
@@ -9,23 +9,23 @@ def load_data(query_params):
         slider_images.append(hm_url)
     if img_url:
         slider_images.append(img_url)
-    if slider_images == []:
-        slider_images = ["images/logo.png"]
+    if not slider_images:
+        slider_images = ["images/logo.png", "images/background.png"]
     return model_url, slider_images
 
-gr.set_static_paths(paths=["images/","models/","assets/"])
+gr.set_static_paths(paths=["images/", "models/", "assets/"])
 with gr.Blocks(theme="Surn/Beeuty") as demo:
     gr.Markdown("# 3D Model Viewer")
     
     with gr.Row():
         model_3d = gr.Model3D(
             label="3D Model",
-            value=None,  # Will be updated by the load function if a URL is provided.
+            value=None,
             height=400
         )
         image_slider = gr.ImageSlider(
             label="Images",
-            value=[],  # Images will be populated automatically from the query strings.
+            value=["images/logo.png", "images/background.png"],
             height=400
         )
         
@@ -34,11 +34,11 @@ with gr.Blocks(theme="Surn/Beeuty") as demo:
             "Upload",
             file_types=[".glb", ".gltf", ".obj", ".png", ".jpg", ".ply"]
         )
-    
-    # Load query parameters from the URL and update the 3D model and image slider accordingly.
+        
+    # Use JavaScript to pass the query parameters to your callback.
     demo.load(
         load_data,
-        inputs=None,
+        inputs=[gr.JSON(), model_3d],
         outputs=[model_3d, image_slider],
         js="""() => {
             const params = Object.fromEntries(new URLSearchParams(window.location.search));
@@ -47,4 +47,7 @@ with gr.Blocks(theme="Surn/Beeuty") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=["assets","assets/","./assets","images/","./images", 'e:/TMP', 'models/'], favicon_path="./assets/favicon.ico")
+    demo.launch(
+        allowed_paths=["assets", "assets/", "./assets", "images/", "./images", 'e:/TMP', 'models/'],
+        favicon_path="./assets/favicon.ico"
+    )