Update app.py

app.py

@@ -1,71 +1,104 @@
+# app.py —— 放在檔案最上方的環境設定（✦✦✦ 這段要在所有 import 之前 ✦✦✦）
 import os
-os.environ["CUDA_VISIBLE_DEVICES"] = ""         # 強制 CPU
-os.environ["WORLDGEN_DISABLE_EVAL"] = "1"       # 若程式支援，避免 KNN
 
-import gradio as gr
-from worldgen import WorldGen
+# 1) 禁用 CUDA（強制在 CPU 上跑，避免 CUDA 警告）
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
-wg = WorldGen(mode="t2s", device="cpu", low_vram=True)
+# 2) 關閉評估（若 WorldGen 支援此旗標，可避免 KNN/自訂外掛）
+os.environ["WORLDGEN_DISABLE_EVAL"] = "1"
 
-gr.Markdown("""# WorldGen — 文字 -> 3D 場景- 僅啟用 Text -> 3D。- Free CPU 會很慢，建議升級 GPU（T4/A10G）。""")
-# 其餘邏輯同前；記得回傳後用延時 Thread 清理臨時資料夾，避免 50GB 被用滿
+# 3) 指定 Gradio 暫存目錄（選用；集中暫存防止亂堆）
+os.environ["GRADIO_TEMP_DIR"] = "/tmp/gradio"
 
-import shutil, tempfile
+
+# ===== 下面才開始 import 其他套件 =====
+import shutil
+import tempfile
+import threading
 from pathlib import Path
-import gradio as gr, torch
+
+import gradio as gr
+import torch
 from worldgen import WorldGen
 
+
+# 初始化 WorldGen（✦ CPU + low_vram ✦）
+wg = WorldGen(mode="t2s", device="cpu", low_vram=True)
+
+
 def _zip_directory(dir_path: Path) -> str:
-    zip_path = dir_path.with_suffix('.zip')
+    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))
+    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)
-    for meth in ('save','export','write','to_disk'):
+
+    # 嘗試通用保存方法（不同版本 WorldGen 返回物件可能不同）
+    for meth in ("save", "export", "write", "to_disk"):
         if hasattr(generated, meth):
-            getattr(generated, meth)(str(out_dir)); return out_dir
-    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
-    (out_dir / 'README.txt').write_text('未自動偵測到保存方法。請依 WorldGen repo 更新保存流程。')
-    return out_dir
+            fp = out_dir / ("scene.glb" if "glb" in meth else "scene.ply")
+            getattr(generated, meth)(str(fp))
+            return out_dir
 
-_worldgen=None
+    # 最後手段：提示需依你使用的 WorldGen 版本更新保存邏輯
+    (out_dir / "README.txt").write_text(
+        "未自動偵測到保存方法。請依 WorldGen 版本的 demo/README 更新保存流程。"
+    )
+    return out_dir
 
-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):
+def generate_from_text(prompt: str):
     if not prompt or not prompt.strip():
-        return None, '請輸入場景描述（英文更穩定）。'
-    wg=_get_worldgen()
+        return None, "請輸入場景描述（英文更穩定）。"
+
     try:
         generated = wg.generate_world(prompt.strip())
     except Exception as e:
-        return None, f'生成失敗：{e}'
+        return None, f"生成失敗：{e}"
+
+    # 臨時目錄寫入與壓縮
     tmpdir = Path(tempfile.mkdtemp())
-    out_dir = tmpdir / 'worldgen_output'
+    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 場景")
+    # ✦ 延時清理臨時目錄，避免磁碟爆掉（50GB 上限）
+    def _cleanup(p: Path):
+        try:
+            shutil.rmtree(p, ignore_errors=True)
+        except Exception:
+            pass
+
+    threading.Timer(300, _cleanup, args=(tmpdir,)).start()  # 5 分鐘後清理
+
+    return zip_path, "✅ 生成完成！請下載 ZIP（含 3D 資產）。"
+
+
+# ===== Gradio 介面 =====
+with gr.Blocks(title="WorldGen • Text -> 3D Scene") as demo:
+    # ✦ 用「三重引號」寫多行 Markdown，避免未終止的字串錯誤
+    gr.Markdown("""# 🌍 WorldGen — 文字 -> 3D 場景- 僅啟用 Text -> 3D。- Free CPU 會很慢，建議升級 GPU（T4/A10G）。""")
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label='輸入英文場景描述', placeholder='a neon-lit cyberpunk street...')
-            btn = gr.Button('生成 3D 場景')
+            prompt = gr.Textbox(
+                label="輸入英文場景描述",
+                placeholder="a neon-lit cyberpunk street with rain reflections, highly detailed",
+                lines=3,
+            )
+            run_btn = gr.Button("🚀 生成 3D 場景")
         with gr.Column():
-            out_zip = gr.File(label='下載輸出（ZIP）')
+            out_zip = gr.File(label="下載輸出（ZIP）")
             status = gr.Markdown()
-    btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
 
-demo.launch()
+    run_btn.click(generate_from_text, inputs=[prompt], outputs=[out_zip, status])
+
+demo.launch()