Upload 3 files

README.md

@@ -1,12 +1,32 @@
----
-title: WorldgenTest
-emoji: 🔥
-colorFrom: yellow
-colorTo: pink
-sdk: gradio
-sdk_version: 5.49.1
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# WorldGen — Text -> 3D Scene (Hugging Face Spaces)
+
+這個 Space 只做 **文字 -> 3D 場景**，使用開源的 **WorldGen** 專案（作者：Ziyang Xie / Manifolds AI）。
+> 注意：Free CPU 會非常慢；建議到 Settings → Hardware 升級 GPU（T4/A10G）再生成，穩定很多。
+
+- 原始專案（作者 GitHub）：https://github.com/ZiYang-xie/WorldGen
+- 專案頁（示例與說明）：https://worldgen.github.io/
+- WorldGen 在 Hugging Face 的模型卡：https://huggingface.co/LeoXie/WorldGen
+
+## 使用方式
+1. 在左側輸入英文場景描述（建議具體到風格、材質、光影）。
+   - 範例：`a neon-lit cyberpunk street with rain reflections, highly detailed, moody lighting`
+2. 按「生成 3D 場景」。
+3. 右側會提供 ZIP 檔下載，內含 3D 資產（例如 GLB/PLY/3DGS 或中介檔），可用 Blender / Unity / Three.js 開啟。
+   - 不同版本 WorldGen 的輸出結構可能不同；若你只拿到 3DGS 或其他格式，請依作者 repo 的 README / demo.py 指南進行轉檔或可視化。
+
+## 執行環境
+- Hugging Face Spaces（Gradio）：本專案即為一個最小可運行的 Space。
+- 硬體建議：
+  - Free CPU：僅驗證流程，很容易超時。
+  - GPU（T4/A10G/A100）：推理較穩定快速。作者表示 A100 約 ~40 秒一個場景，low_vram 模式可再快一些。
+- 安裝：本 Space 會自動依 requirements.txt 安裝：
+  - gradio（UI）、torch/torchvision（推理）、worldgen（從作者 GitHub 安裝最新版本）。
+
+## 已知限制
+- 保存/匯出：WorldGen 不同版本的輸出物件可能有所差異（3DGS/mesh/中介資產）。app.py 已嘗試常見的 save()/export()/to_glb() 等方法；若仍無法輸出，請查看作者的 demo.py 與 README 並在 app.py 的 _save_scene(...) 中替換對應保存程式碼。
+- 線上預覽：為提高相容性，預設只提供 ZIP 下載。若你要直接在頁面預覽 .glb，可在 app.py 加入 <model-viewer> 或 three.js（用 gr.HTML）。
+
+## 參考
+- WorldGen GitHub（安裝與 WorldGen(mode="t2s") 用法）：https://github.com/ZiYang-xie/WorldGen
+- 作者回覆的推理耗時（A100 ~ 40s）：https://github.com/ZiYang-xie/WorldGen/issues/5
+- Hugging Face Spaces 硬體與計價（CPU 免費、GPU 按時計費）：https://huggingface.co/docs/hub/en/spaces-overview

app.py

@@ -0,0 +1,114 @@
+import os
+import shutil
+import tempfile
+from pathlib import Path
+
+import gradio as gr
+import torch
+
+# 來自 WorldGen 開源庫，支援 from worldgen import WorldGen 與 mode="t2s"
+from worldgen import WorldGen
+
+def _zip_directory(dir_path: Path) -> str:
+    zip_path = dir_path.with_suffix('.zip')
+    if zip_path.exists():
+        zip_path.unlink()
+    shutil.make_archive(str(dir_path), 'zip', root_dir=str(dir_path))
+    return str(zip_path)
+
+def _save_scene(generated, out_dir: Path) -> Path:
+    out_dir.mkdir(parents=True, exist_ok=True)
+
+    # 1) 物件方法保存
+    for meth in ('save', 'export', 'write', 'to_disk'):
+        if hasattr(generated, meth):
+            getattr(generated, meth)(str(out_dir))
+            return out_dir
+
+    # 2) 導出單一檔案（GLB/PLY）
+    for meth in ('save_as_glb', 'to_glb', 'save_as_ply', 'to_ply'):
+        if hasattr(generated, meth):
+            fp = out_dir / ('scene.glb' if 'glb' in meth else 'scene.ply')
+            getattr(generated, meth)(str(fp))
+            return out_dir
+
+    # 3) 直接給了路徑
+    if isinstance(generated, (str, Path)):
+        src = Path(generated)
+        if src.exists():
+            if src.is_dir():
+                shutil.copytree(src, out_dir, dirs_exist_ok=True)
+            else:
+                shutil.copy(src, out_dir / src.name)
+            return out_dir
+
+    if isinstance(generated, dict):
+        for v in generated.values():
+            p = Path(v)
+            if p.exists():
+                if p.is_dir():
+                    shutil.copytree(p, out_dir, dirs_exist_ok=True)
+                else:
+                    shutil.copy(p, out_dir / p.name)
+        return out_dir
+
+    # 4) 最後手段：放一個說明檔
+    (out_dir / 'README.txt').write_text(
+        '未自動偵測到保存方法。請依 WorldGen repo 的 demo.py/README 更新此函式以正確保存輸出。'
+    )
+    return out_dir
+
+_worldgen = None
+
+def _get_worldgen():
+    global _worldgen
+    if _worldgen is None:
+        device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        _worldgen = WorldGen(mode='t2s', device=device, low_vram=True)
+    return _worldgen
+
+def generate_from_text(prompt: str):
+    if not prompt or not prompt.strip():
+        return None, '請輸入場景描述（英文更穩定）。'
+
+    wg = _get_worldgen()
+
+    try:
+        generated = wg.generate_world(prompt.strip())
+    except Exception as e:
+        return None, f'生成失敗：{e}'
+
+    tmpdir = Path(tempfile.mkdtemp())
+    out_dir = tmpdir / 'worldgen_output'
+    out_dir = _save_scene(generated, out_dir)
+    zip_path = _zip_directory(out_dir)
+
+    return zip_path, '生成完成！請下載 ZIP（含 3D 資產）。'
+
+with gr.Blocks(title='WorldGen • Text -> 3D Scene') as demo:
+    gr.Markdown(
+        '# WorldGen — 文字 -> 3D 場景
+
+'
+        '- 僅啟用 Text -> 3D。
+
+'
+        '- Free CPU 會很慢，建議在 Settings → Hardware 升級 GPU（T4/A10G）。'
+    )
+    with gr.Row():
+        with gr.Column(scale=1):
+            prompt = gr.Textbox(
+                label='輸入英文場景描述',
+                placeholder=(
+                    'e.g. a neon-lit cyberpunk street with rain reflections, highly detailed'
+                ),
+                lines=3,
+            )
+            run_btn = gr.Button('生成 3D 場景')
+        with gr.Column(scale=1):
+            out_zip = gr.File(label='下載輸出（ZIP）')
+            status = gr.Markdown()
+
+    run_btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
+
+demo.launch()

requirements.txt

@@ -0,0 +1,6 @@
+gradio>=4.44.0
+
+torch
+torchvision
+
+git+https://github.com/ZiYang-xie/WorldGen.git