Upload 3 files

README.md

@@ -10,36 +10,11 @@ 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
+這個 Space 只做 **文字 -> 3D 場景**。
 
 ## 使用方式
-1. 在左側輸入英文場景描述（建議具體到風格、材質、光影）。
-   - 範例：`a neon-lit cyberpunk street with rain reflections, highly detailed, moody lighting`
+1. 在左側輸入英文場景描述。
 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
+3. 下載 ZIP 檔（內含 3D 資產）。

app.py

@@ -1,12 +1,6 @@
-import os
-import shutil
-import tempfile
+import shutil, tempfile
 from pathlib import Path
-
-import gradio as gr
-import torch
-
-# 來自 WorldGen 開源庫，支援 from worldgen import WorldGen 與 mode="t2s"
+import gradio as gr, torch
 from worldgen import WorldGen
 
 def _zip_directory(dir_path: Path) -> str:
@@ -18,47 +12,17 @@ def _zip_directory(dir_path: Path) -> str:
 
 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'):
+    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'):
+            getattr(generated, meth)(str(out_dir)); return out_dir
+    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 更新此函式以正確保存輸出。'
-    )
+            getattr(generated, meth)(str(fp)); return out_dir
+    (out_dir / 'README.txt').write_text('未自動偵測到保存方法。請依 WorldGen repo 更新保存流程。')
     return out_dir
 
-_worldgen = None
+_worldgen=None
 
 def _get_worldgen():
     global _worldgen
@@ -67,48 +31,31 @@ def _get_worldgen():
         _worldgen = WorldGen(mode='t2s', device=device, low_vram=True)
     return _worldgen
 
-def generate_from_text(prompt: str):
+def generate_from_text(prompt:str):
     if not prompt or not prompt.strip():
         return None, '請輸入場景描述（英文更穩定）。'
-
-    wg = _get_worldgen()
-
+    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）。'
-    )
+    gr.Markdown('# WorldGen — 文字 -> 3D 場景
+- 僅啟用 Text -> 3D。
+- Free CPU 會很慢，建議升級 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):
+        with gr.Column():
+            prompt = gr.Textbox(label='輸入英文場景描述', placeholder='a neon-lit cyberpunk street...')
+            btn = gr.Button('生成 3D 場景')
+        with gr.Column():
             out_zip = gr.File(label='下載輸出（ZIP）')
             status = gr.Markdown()
-
-    run_btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
+    btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
 
 demo.launch()

requirements.txt

@@ -1,8 +1,6 @@
 gradio==4.44.0
 torch==2.4.*
-
-# 視需要再打開：在 CPU 免費層常常不是必需
+# 如需再加：
 # torchvision==0.19.*
 
 git+https://github.com/ZiYang-xie/WorldGen.git
-``