update

app.py

@@ -1,66 +1,97 @@
 import gradio as gr
 import os
 
-# --- Configuration ---
-# 假設你的 .splat 檔案已經放在 "assets_converted" 資料夾中
-# 建議將檔案命名為 office0.splat, office2.splat, room0.splat 以保持簡潔
+# 獲取當前絕對路徑，確保讀取檔案無誤
+BASE_DIR = os.path.abspath(os.getcwd())
+
 SCENES = [
     {
         "name": "Office 0", 
         "thumb": "assets/office0.PNG",            
-        "model": "assets/office0.splat", # <--- 修改處：改成 .splat 檔案
+        "model": "assets/office0.ply",
         "tracking_img": "assets/office0_tracking.PNG" 
     },
     {
         "name": "Office 2",
         "thumb": "assets/office2.PNG",
-        "model": "assets/office2.splat", # <--- 修改處
+        "model": "assets/office2.ply",
         "tracking_img": "assets/office2_tracking.PNG"
     },
     {
         "name": "Room 0",
         "thumb": "assets/room0.PNG",
-        "model": "assets/room0.splat",   # <--- 修改處
+        "model": "assets/room0.ply",
         "tracking_img": "assets/room0_tracking.PNG"
     }
 ]
 
-# 建立 Gallery 用清單
 gallery_items = [(item["thumb"], item["name"]) for item in SCENES]
 
-def load_scene(evt: gr.SelectData):
+def load_scene(evt: gr.SelectData, request: gr.Request):
     """
-    載入選定的場景。
-    Gradio 的 Model3D 元件會根據副檔名 (.splat) 自動切換渲染模式。
+    載入選定的場景並生成自動播放的 iframe 連結
+    request: 用來獲取當前 Gradio 的網址 (localhost 或 share link)
     """
     index = evt.index
     scene = SCENES[index]
     
-    # 檢查檔案是否存在，避免報錯讓使用者困惑
-    if not os.path.exists(scene["model"]):
+    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'>⚠️ File not found</div>",
             None,
-            scene["tracking_img"] if os.path.exists(scene["tracking_img"]) else None,
-            f"**⚠️ Error: File not found: {scene['model']}**\n請確認 .splat 檔案已上傳至正確路徑。"
+            f"**⚠️ Error:** File not found: {scene['model']}"
         )
     
-    # 計算檔案大小供顯示
-    file_size = os.path.getsize(scene["model"]) / (1024 * 1024)
+    # 1. 取得你目前的 Gradio 伺服器網址 (例如 https://xxxx.gradio.live 或 http://127.0.0.1:7860)
+    # 注意：如果是在本地跑且沒有 share=True，這個網址通常是 http
+    if request:
+        host = request.headers.get("host") # e.g. "7213...gradio.live"
+        # 判斷是 http 還是 https (Gradio Share 連結通常是 https)
+        protocol = "https" if "gradio.live" in host else "http"
+        base_url = f"{protocol}://{host}"
+    else:
+        base_url = "http://127.0.0.1:7860"
+
+    # 2. 組合檔案的下載連結 (Gradio 的 /file/ 路徑)
+    # 注意 Windows路徑的反斜線可能需要處理，但在 Gradio file serving 通常沒問題
+    file_url = f"{base_url}/file={full_model_path}"
+    
+    # 3. 組合外部 Viewer 的連結 (利用 ?url= 參數)
+    viewer_url = f"https://antimatter15.com/splat/?url={file_url}"
     
+    print(f"🔗 Generated Link: {viewer_url}") # Debug 用，可以在終端機看到連結
+
+    # 4. 生成 HTML
+    html = f"""
+    <iframe 
+        src="{viewer_url}" 
+        width="100%" 
+        height="600px" 
+        frameborder="0"
+        allow="camera; gpu"
+        style="border-radius: 8px; background: #000;"
+    ></iframe>
+    """
+    
+    # 提示訊息
+    status_msg = f"**{scene['name']}** - Auto-loading via External Viewer..."
+    if protocol == "http":
+        status_msg += "\n\n⚠️ **Note:** If screen is black, use `share=True` (see explanation)."
+
     return (
-        scene["model"],
+        html,
         scene["tracking_img"],
-        f"**{scene['name']}** ({file_size:.1f} MB) ✓ Loaded (.splat)"
+        status_msg
     )
 
 # --- UI ---
 with gr.Blocks(title="RGS-SLAM Demo") as demo:
     
     gr.Markdown("# 🚀 RGS-SLAM Gaussian Splatting Viewer")
-    gr.Markdown("此檢視器使用 `.splat` 格式，載入速度更快且相容性更高。")
     
     with gr.Row():
-        # 左側選單
         with gr.Column(scale=1):
             gr.Markdown("### 📂 Select Scene")
             scene_gallery = gr.Gallery(
@@ -69,30 +100,28 @@ with gr.Blocks(title="RGS-SLAM Demo") as demo:
                 columns=1,
                 height=400,
                 object_fit="contain",
-                allow_preview=False # 關閉點擊預覽圖片，強制觸發選擇事件
+                allow_preview=False
             )
         
-        # 右側顯示區
         with gr.Column(scale=3):
-            status = gr.Markdown("**👈 Select a scene from the left to load**")
+            status = gr.Markdown("**👈 Select a scene to auto-load**")
             
-            with gr.Tab("🎯 3D Reconstruction"):
-                # Model3D 支援 .splat, .ply, .glb 等格式
-                model_viewer = gr.Model3D(
-                    label="3D Gaussian Splat",
-                    clear_color=[0.0, 0.0, 0.0, 1.0], # 建議 splat 使用黑色背景
-                    height=600,
-                    interactive=True
+            with gr.Tab("🎯 3D Viewer"):
+                # 初始狀態為空
+                viewer_frame = gr.HTML(
+                    value="<div style='text-align: center; padding: 100px; color: #666;'>Select a scene</div>"
                 )
             
             with gr.Tab("📍 Camera Trajectory"):
                 trajectory_img = gr.Image(label="Camera Path", height=600)
     
-    # 事件綁定
     scene_gallery.select(
         fn=load_scene,
-        outputs=[model_viewer, trajectory_img, status]
+        inputs=None, # inputs 不需要顯式傳入，gr.SelectData 會自動處理
+        outputs=[viewer_frame, trajectory_img, status]
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # 關鍵：allowed_paths 允許 Gradio 讀取當前資料夾的檔案
+    # 建議開啟 share=True 以解決 HTTPS 跨域問題
+    demo.launch(share=True, allowed_paths=[BASE_DIR])