update

app.py

@@ -34,49 +34,57 @@ def load_scene(evt: gr.SelectData, request: gr.Request):
     index = evt.index
     scene = SCENES[index]
     
-    # 組合絕對路徑
-    full_model_path = os.path.join(BASE_DIR, scene["model"])
+    # 取得 Hugging Face Space 的 ID (例如: yfyangd/rgs-slam-demo)
+    space_id = os.environ.get("SPACE_ID")
     
-    # 1. 檢查檔案是否存在
-    if not os.path.exists(full_model_path):
-        return (
-            "<div style='text-align:center; padding:50px; color:red'>⚠️ File not found</div>",
-            None,
-            f"**⚠️ Error:** File not found: {full_model_path}"
-        )
+    # --- 關鍵修正邏輯 ---
     
-    # 2. 處理路徑 (Linux/HF Spaces 相容性)
-    # HF Spaces 是 Linux 環境，路徑通常是 /home/user/app/...
-    clean_path = full_model_path
-    
-    # 3. 判斷網址 (這是修正的關鍵！)
-    # 預設值
-    base_url = "http://127.0.0.1:7860" 
-    
-    if request:
-        host = request.headers.get("host") # 獲取網域，例如 yfyangd-rgs-slam-demo.hf.space
+    if space_id:
+        # 【情境 A：在 Hugging Face Spaces 線上執行】
+        # 我們改用 Hugging Face 的 "resolve" 網址。
+        # 這是直接指向 Git 倉庫的原始檔案，速度快且通常不會被 CORS 擋住。
+        # 格式: https://huggingface.co/spaces/{SPACE_ID}/resolve/main/{相對路徑}
+        
+        # 注意：這裡直接用 scene["model"] (例如 assets/office0.ply)，不需要絕對路徑
+        file_url = f"https://huggingface.co/spaces/{space_id}/resolve/main/{scene['model']}"
         
-        # --- 強制轉 HTTPS 的邏輯 ---
-        # 如果網址包含 'hf.space' (Hugging Face) 或 'gradio.live'，強制用 https
-        if "hf.space" in host or "gradio.live" in host:
-            base_url = f"https://{host}"
-        else:
-            base_url = f"http://{host}"
-            
-    # 4. 組合檔案連結
-    file_url = f"{base_url}/file={clean_path}"
+        # 有些 viewer 需要加上 download=true 才能正確觸發 header，但 antimatter 通常不用
+        print(f"🌍 Using HF Repo URL: {file_url}")
+        
+    else:
+        # 【情境 B：在 本機 (Localhost) 執行】
+        # 保持原本的邏輯，使用 localhost 的連結
+        full_model_path = os.path.join(BASE_DIR, scene["model"])
+        
+        if not os.path.exists(full_model_path):
+            return (
+                "<div style='text-align:center; padding:50px; color:red'>⚠️ File not found</div>",
+                None,
+                f"**⚠️ Error:** File not found: {full_model_path}"
+            )
+
+        # 處理 Windows 路徑
+        clean_path = full_model_path.replace("\\", "/")
+        
+        base_url = "http://127.0.0.1:7860"
+        if request:
+             host = request.headers.get("host")
+             if "gradio.live" in host:
+                 base_url = f"https://{host}"
+             else:
+                 base_url = f"http://{host}"
+        
+        file_url = f"{base_url}/file={clean_path}"
+        print(f"🏠 Using Local URL: {file_url}")
+
+    # --- 通用部分：編碼並生成 iframe ---
     
-    # 5. 編碼網址
-    # 這裡很關鍵：我們要把 file_url 變成外部 viewer 看得懂的參數
-    # 使用 quote 將特殊符號 (如空格、=、?) 轉碼，但保留 : 和 / 以維持網址結構
+    # 編碼 file_url (處理特殊字元)
     encoded_file_url = urllib.parse.quote(file_url, safe=':/')
     
-    # 6. 組合外部 Viewer 連結
+    # 組合外部 Viewer 連結
     viewer_url = f"https://antimatter15.com/splat/?url={encoded_file_url}"
     
-    print(f"🚀 Final URL (Should be HTTPS): {viewer_url}")
-
-    # 7. 生成 HTML
     html = f"""
     <iframe 
         src="{viewer_url}" 
@@ -88,7 +96,7 @@ def load_scene(evt: gr.SelectData, request: gr.Request):
     ></iframe>
     """
     
-    status_msg = f"**{scene['name']}** - Loaded from {base_url}"
+    status_msg = f"**{scene['name']}** - Loaded via {'HF CDN' if space_id else 'Local Server'}"
 
     return (
         html,