update

app.py

@@ -34,7 +34,7 @@ import gradio as gr
 import numpy as np
 import pandas as pd
 import trimesh
-from app_style import custom_theme, lighting_css
+from app_style import custom_theme
 from embodied_gen.utils.tags import VERSION
 
 try:
@@ -124,6 +124,7 @@ os.makedirs(TMP_DIR, exist_ok=True)
 css = """
 .gradio-container .gradio-group { box-shadow: 0 2px 4px rgba(0,0,0,0.05) !important; }
 #asset-gallery { border: 1px solid #E5E7EB; border-radius: 8px; padding: 8px; background-color: #F9FAFB; }
+#download-row button { white-space: nowrap; }
 """
 
 lighting_css = """
@@ -724,13 +725,33 @@ def download_mjcf(asset_dir: str, req: gr.Request) -> str | None:
     return zip_path
 
 
+def download_affordance(asset_dir: str, req: gr.Request) -> str | None:
+    """Package the ``affordance/`` folder."""
+    if not asset_dir or not os.path.isdir(asset_dir):
+        gr.Warning("Please select an asset first.")
+        return None
+
+    gr.Info("⏳ Preparing affordance asset for download, please wait...")
+    ensure_hf_files([f"{_rel_dir(asset_dir)}/affordance/**"])
+    affordance_dir = os.path.join(asset_dir, "affordance")
+    if not os.path.isdir(affordance_dir):
+        gr.Warning("No affordance folder available for this asset.")
+        return None
+
+    zip_path, uid, zip_name = _fmt_zip_path(asset_dir, "affordance", req)
+    _zip_items(zip_path, [(affordance_dir, "affordance")])
+    gr.Info(f"✅ {zip_name} is ready.")
+    return zip_path
+
+
 # Download buttons keyed by format; order matches the outputs list used when
 # locking/unlocking them together.
-_DL_KEYS = ("urdf", "usd", "mjcf")
+_DL_KEYS = ("urdf", "usd", "mjcf", "affordance")
 _DL_LABELS = {
     "urdf": "⬇️ Download URDF",
     "usd": "⬇️ Download USD",
     "mjcf": "⬇️ Download MJCF",
+    "affordance": "⬇️ Affordance",
 }
 
 
@@ -780,7 +801,7 @@ with gr.Blocks(
     css=css,
     title="3D Asset Library",
 ) as demo:
-    gr.HTML(lighting_css, visible=False)
+    # gr.HTML(lighting_css, visible=False)
     gr.Markdown(
         """
         ## 🏛️ ***EmbodiedGen***: 3D Asset Gallery Explorer
@@ -931,7 +952,7 @@ with gr.Blocks(
                     urdf_file = gr.Textbox(
                         label="URDF File Path", interactive=False, lines=2
                     )
-                with gr.Row():
+                with gr.Row(elem_id="download-row"):
                     # DownloadButtons that build their zip into their own value
                     # on click; a chained JS handler then triggers the browser
                     # download from that value (one-click). If the JS ever
@@ -952,6 +973,11 @@ with gr.Blocks(
                         variant="primary",
                         interactive=False,
                     )
+                    affordance_dl_btn = gr.DownloadButton(
+                        label="⬇️ Affordance",
+                        variant="primary",
+                        interactive=False,
+                    )
 
     search_button.click(
         fn=search_assets,
@@ -1030,7 +1056,7 @@ with gr.Blocks(
         ],
     ).success(
         fn=_reset_downloads,
-        outputs=[urdf_dl_btn, usd_dl_btn, mjcf_dl_btn],
+        outputs=[urdf_dl_btn, usd_dl_btn, mjcf_dl_btn, affordance_dl_btn],
     )
 
     # After the zip is built into the button's own value, pass that file value
@@ -1050,7 +1076,7 @@ with gr.Blocks(
     }
     """
 
-    dl_btns = [urdf_dl_btn, usd_dl_btn, mjcf_dl_btn]
+    dl_btns = [urdf_dl_btn, usd_dl_btn, mjcf_dl_btn, affordance_dl_btn]
 
     # Each flow: lock all buttons (prevent spam clicks) -> build the zip ->
     # JS-trigger the browser download -> unlock. Selecting another asset
@@ -1077,6 +1103,16 @@ with gr.Blocks(
         fn=_unlock_downloads, outputs=dl_btns
     )
 
+    affordance_dl_btn.click(
+        fn=lambda: _lock_downloads("affordance"), outputs=dl_btns
+    ).then(
+        fn=download_affordance,
+        inputs=[asset_folder],
+        outputs=[affordance_dl_btn],
+    ).then(
+        fn=lambda *a: None, inputs=[affordance_dl_btn], js=download_js
+    ).then(fn=_unlock_downloads, outputs=dl_btns)
+
     demo.load(start_session)
     demo.unload(end_session)
 

embodied_gen/utils/gpt_clients.py

@@ -255,7 +255,10 @@ class GPTclient:
             if is_gpt5:
                 # GPT-5.x rejects custom temperature/top_p/penalty/stop and
                 # uses `max_completion_tokens` instead of `max_tokens`.
-                if "max_tokens" in params and "max_completion_tokens" not in params:
+                if (
+                    "max_tokens" in params
+                    and "max_completion_tokens" not in params
+                ):
                     params["max_completion_tokens"] = params.pop("max_tokens")
                 for k in (
                     "temperature",

embodied_gen/utils/gpt_config.yaml

@@ -1,5 +1,5 @@
 # config.yaml
-agent_type: "gpt-5.4" # gpt-4o, gpt-5.4 or qwen2.5-vl
+agent_type: "gpt-5.4" # gpt-4o, gpt-5.4 or gemma-4-31b
 
 gpt-4o:
   endpoint: https://xxx.openai.azure.com
@@ -13,8 +13,8 @@ gpt-5.4:
   api_version: 2024-12-01-preview
   model_name: gpt-5.4
 
-qwen2.5-vl:
+gemma-4-31b:
   endpoint: https://openrouter.ai/api/v1
   api_key: sk-or-v1-xxx
   api_version: null
-  model_name: qwen/qwen2.5-vl-72b-instruct:free
+  model_name: google/gemma-4-31b-it:free

embodied_gen/utils/tags.py

@@ -1 +1 @@
-VERSION = "v0.1.8"
+VERSION = "v2.0.0"