Make Gradio tabs compatible with Space SDK

app.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from functools import lru_cache
+import inspect
 import os
 from pathlib import Path
 from typing import Optional
@@ -41,6 +42,20 @@ def _ensure_hf_cache_dirs() -> None:
 
 _ensure_hf_cache_dirs()
 
+
+def _ensure_valid_omp_num_threads() -> None:
+    value = os.environ.get("OMP_NUM_THREADS", "").strip()
+    try:
+        if int(value) > 0:
+            return
+    except (TypeError, ValueError):
+        pass
+
+    os.environ["OMP_NUM_THREADS"] = "1"
+
+
+_ensure_valid_omp_num_threads()
+
 import cv2
 import gradio as gr
 import numpy as np
@@ -122,6 +137,19 @@ DEPTH_VIEW_TAB_ID = "pcd-viewer-tab"
 GS_VIEW_TAB_ID = "gs-viewer-tab"
 gr.set_static_paths(paths=[str(GS_VIEWER_ROOT)])
 
+
+def _supports_component_kwarg(component_init, kwarg: str) -> bool:
+    try:
+        return kwarg in inspect.signature(component_init).parameters
+    except (TypeError, ValueError):
+        return False
+
+
+TAB_RENDER_KWARGS = {"render_children": True} if _supports_component_kwarg(gr.Tab.__init__, "render_children") else {}
+TABS_RENDER_KWARGS = (
+    {"render_children": True} if _supports_component_kwarg(gr.Tabs.__init__, "render_children") else {}
+)
+
 CSS = """
 #top-workspace {
     align-items: stretch;
@@ -798,8 +826,8 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=8, min_width=640, elem_id="outputs-column"):
             status_output = gr.Markdown()
 
-            with gr.Tabs(selected=DEPTH_VIEW_TAB_ID, elem_id="primary-view-tabs") as primary_view_tabs:
-                with gr.Tab("PCD Viewer", id=DEPTH_VIEW_TAB_ID, render_children=True):
+            with gr.Tabs(selected=DEPTH_VIEW_TAB_ID, elem_id="primary-view-tabs", **TABS_RENDER_KWARGS) as primary_view_tabs:
+                with gr.Tab("PCD Viewer", id=DEPTH_VIEW_TAB_ID, **TAB_RENDER_KWARGS):
                     depth_model_3d = gr.Model3D(
                         label="Point Cloud Viewer",
                         display_mode="solid",
@@ -807,11 +835,11 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                         height=700,
                         elem_id="depth-model3d-viewer",
                     )
-                with gr.Tab("GS Viewer", id=GS_VIEW_TAB_ID, render_children=True):
+                with gr.Tab("GS Viewer", id=GS_VIEW_TAB_ID, **TAB_RENDER_KWARGS):
                     gs_viewer_html = gr.HTML(elem_id="gs-viewer-html")
 
-            with gr.Tabs(elem_id="secondary-output-tabs"):
-                with gr.Tab("Depth Analysis", render_children=True):
+            with gr.Tabs(elem_id="secondary-output-tabs", **TABS_RENDER_KWARGS):
+                with gr.Tab("Depth Analysis", **TAB_RENDER_KWARGS):
                     depth_comparison = gr.Image(
                         label="RGB vs Depth",
                         type="filepath",
@@ -831,7 +859,7 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
                             height=260,
                             elem_id="depth-preview",
                         )
-                with gr.Tab("Downloads", render_children=True):
+                with gr.Tab("Downloads", **TAB_RENDER_KWARGS):
                     with gr.Row():
                         depth_download_files = gr.File(label="Depth Files", type="filepath")
                         gs_download_files = gr.File(label="GS Files", type="filepath")