update

app.py

@@ -14,6 +14,7 @@
 # implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
+
 import os
 
 gradio_tmp_dir = os.path.join(
@@ -26,12 +27,11 @@ import shutil
 import uuid
 import xml.etree.ElementTree as ET
 from pathlib import Path
-from typing import Any, Dict, Tuple
 
 import gradio as gr
 import pandas as pd
-import yaml
 from app_style import custom_theme, lighting_css
+from embodied_gen.utils.tags import VERSION
 
 try:
     from embodied_gen.utils.gpt_clients import GPT_CLIENT as gpt_client
@@ -47,14 +47,8 @@ except Exception as e:
 
 
 # --- Configuration & Data Loading ---
-VERSION = "v0.1.5"
 RUNNING_MODE = "hf_remote"  # local or hf_remote
 CSV_FILE = "dataset_index.csv"
-import spaces
-@spaces.GPU
-def fake_gpu_init():
-    pass
-fake_gpu_init()
 
 if RUNNING_MODE == "local":
     DATA_ROOT = "/horizon-bucket/robot_lab/datasets/embodiedgen/assets"
@@ -260,8 +254,7 @@ def search_assets(query: str, top_k: int):
     return items, gr.update(interactive=True), top_assets
 
 
-# --- Mesh extraction ---
-def _extract_mesh_paths(row) -> Tuple[str | None, str | None, str]:
+def _extract_mesh_paths(row) -> tuple[str | None, str | None, str]:
     desc = row["description"]
     urdf_path = os.path.join(DATA_ROOT, row["urdf_path"])
     asset_dir = os.path.join(DATA_ROOT, row["asset_dir"])
@@ -731,4 +724,9 @@ with gr.Blocks(
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        # server_port=8088,
+        # allowed_paths=[
+            "/horizon-bucket/robot_lab/datasets/embodiedgen/assets"
+        ],
+    )

app_style.py

@@ -1,10 +1,26 @@
+# Project EmbodiedGen
+#
+# Copyright (c) 2025 Horizon Robotics. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied. See the License for the specific language governing
+# permissions and limitations under the License.
+
 from gradio.themes import Soft
 from gradio.themes.utils.colors import gray, neutral, slate, stone, teal, zinc
 
 lighting_css = """
 <style>
 #lighter_mesh canvas {
-    filter: brightness(1.9) !important;
+    filter: brightness(2.3) !important;
 }
 </style>
 """