add app script

app.py

@@ -0,0 +1,57 @@
+import time
+import gradio as gr
+import os
+
+import numpy as np
+from PIL import Image
+import math
+from collections import defaultdict
+import os
+from huggingface_hub import HfApi
+
+
+def get_dataset_classes():
+    hf_api = HfApi()
+    info = hf_api.dataset_info(repo_id="simenv-explorer/public-test")
+    dataset_classes = defaultdict(list)
+
+    for file in info.siblings:
+        if ".glb" in file.rfilename:
+            class_name = file.rfilename.split("/")[0]
+            dataset_classes[class_name].append(file.rfilename)
+
+    print(dataset_classes)
+    return dataset_classes
+
+
+dataset_dict = get_dataset_classes()
+dataset_classes = list(dataset_dict.keys())
+default_models = dataset_dict[dataset_classes[0]]
+
+
+def load_mesh(mesh_file_name):
+    return mesh_file_name, mesh_file_name
+
+
+def update(model_name):
+    # wget the glb file from the datasets repo
+    print(model_name)
+    return f"{model_name}/0.glb"
+
+
+def update_model_list(choice_class):
+    print(f"inp1 changed {choice_class}")
+    return {"choices": dataset_dict[choice_class]}
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            inp = gr.Dropdown(choices=dataset_classes, interactive=True, label="3D Model Class", value=dataset_classes[0])
+            out1 = gr.Dropdown(choices=default_models, interactive=True, label="3D Model", value=default_models[0])
+        out2 = gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model")
+    
+    inp.change(fn=update, inputs=inp, outputs=out2)
+
+
+demo.launch()