update app.py

app.py

@@ -1,159 +1,57 @@
 import gradio as gr
 import os
-import struct
-import numpy as np
-import spaces  # <--- 1. 新增這個 import
 
-# --- 1. Configuration ---
+# --- Configuration ---
+# 指向我們剛剛上傳的 "assets_converted" 資料夾
 SCENES = [
     {
         "name": "Office 0", 
         "thumb": "assets/office0.PNG",           
-        "model": "assets/office0.ply",
+        "model": "assets_converted/office0_converted.ply", # 指向轉好的檔
         "tracking_img": "assets/office0_tracking.PNG" 
     },
     {
         "name": "Office 2",
         "thumb": "assets/office2.PNG",
-        "model": "assets/office2.ply",
+        "model": "assets_converted/office2_converted.ply",
         "tracking_img": "assets/office2_tracking.PNG"
     },
     {
         "name": "Room 0",
         "thumb": "assets/room0.PNG",
-        "model": "assets/room0.ply",
+        "model": "assets_converted/room0_converted.ply",
         "tracking_img": "assets/room0_tracking.PNG"
     }
 ]
 
 gallery_items = [(item["thumb"], item["name"]) for item in SCENES]
 
-# --- 2. 讀取並轉換 Gaussian Splatting PLY ---
-def read_gaussian_ply(filepath):
-    """讀取 Gaussian Splatting PLY 並提取位置和顏色"""
-    
-    with open(filepath, 'rb') as f:
-        # 讀取 header
-        header = []
-        while True:
-            line = f.readline().decode('utf-8').strip()
-            header.append(line)
-            if line == 'end_header':
-                break
-        
-        # 解析 vertex 數量
-        vertex_count = 0
-        for line in header:
-            if line.startswith('element vertex'):
-                vertex_count = int(line.split()[-1])
-                break
-        
-        print(f"Reading {vertex_count:,} Gaussian splats from {os.path.basename(filepath)}...")
-        
-        # 讀取二進制數據
-        bytes_per_vertex = 17 * 4  # 17 floats * 4 bytes
-        data = f.read(vertex_count * bytes_per_vertex)
-        
-        # 解析數據
-        vertices = []
-        colors = []
-        
-        for i in range(vertex_count):
-            offset = i * bytes_per_vertex
-            vertex_data = struct.unpack('17f', data[offset:offset + bytes_per_vertex])
-            
-            # 提取位置 (x, y, z)
-            x, y, z = vertex_data[0], vertex_data[1], vertex_data[2]
-            vertices.append([x, y, z])
-            
-            # 提取顏色 (f_dc_0, f_dc_1, f_dc_2) 並轉換到 [0, 255]
-            SH_C0 = 0.28209479177387814
-            r = max(0, min(255, int((0.5 + SH_C0 * vertex_data[6]) * 255)))
-            g = max(0, min(255, int((0.5 + SH_C0 * vertex_data[7]) * 255)))
-            b = max(0, min(255, int((0.5 + SH_C0 * vertex_data[8]) * 255)))
-            colors.append([r, g, b])
-        
-        return np.array(vertices, dtype=np.float32), np.array(colors, dtype=np.uint8)
-
-# --- 3. 轉換為標準 PLY ---
-def convert_to_standard_ply(input_path, output_path):
-    """將 Gaussian Splatting PLY 轉換為標準帶顏色的 PLY"""
-    
-    vertices, colors = read_gaussian_ply(input_path)
-    
-    # 寫入 ASCII PLY
-    with open(output_path, 'w') as f:
-        f.write('ply\n')
-        f.write('format ascii 1.0\n')
-        f.write(f'element vertex {len(vertices)}\n')
-        f.write('property float x\n')
-        f.write('property float y\n')
-        f.write('property float z\n')
-        f.write('property uchar red\n')
-        f.write('property uchar green\n')
-        f.write('property uchar blue\n')
-        f.write('end_header\n')
-        
-        for vertex, color in zip(vertices, colors):
-            f.write(f'{vertex[0]} {vertex[1]} {vertex[2]} {color[0]} {color[1]} {color[2]}\n')
-    
-    print(f"✅ Converted to: {output_path}")
-
-# --- 4. 核心處理函式 (加上 @spaces.GPU) ---
-# 注意：即使這只是檔案轉換，加上這個標籤可以解決 ZeroGPU 的檢查錯誤
-# 並且將運算移到計算節點上執行
-@spaces.GPU(duration=120) 
-def process_and_load_scene(evt: gr.SelectData):
+def load_scene(evt: gr.SelectData):
+    """
+    因為檔案已經是標準格式，直接回傳路徑即可。
+    不需要 GPU，不需要運算。
+    """
     index = evt.index
     scene = SCENES[index]
     
-    converted_dir = "converted"
-    os.makedirs(converted_dir, exist_ok=True)
-    
-    input_ply = scene["model"]
-    output_ply = f"{converted_dir}/{os.path.basename(input_ply)}"
-    
-    # 檢查是否需要轉換 (Lazy Loading)
-    if not os.path.exists(output_ply):
-        print(f"\n🔄 Converting {scene['name']} on GPU Node...")
-        try:
-            convert_to_standard_ply(input_ply, output_ply)
-        except Exception as e:
-            return (
-                None,
-                scene["tracking_img"],
-                f"**Error converting {scene['name']}: {str(e)}**"
-            )
-    else:
-        print(f"✅ Using cached model: {output_ply}")
-
-    # 檢查檔案大小
-    if os.path.exists(output_ply):
-        file_size = os.path.getsize(output_ply) / (1024 * 1024)
+    if not os.path.exists(scene["model"]):
         return (
-            output_ply,
+            None,
             scene["tracking_img"],
-            f"**{scene['name']}** ({file_size:.1f} MB) ✓ Loaded"
+            f"**Error: Model file not found: {scene['model']}**"
         )
-    else:
-        return (
-            None, 
-            scene["tracking_img"], 
-            "**Error: File not found after conversion**"
-        )
-
-# --- 5. UI ---
-print("\n" + "="*60)
-print("🚀 RGS-SLAM Viewer - Initializing UI...")
-print("="*60 + "\n")
+    
+    file_size = os.path.getsize(scene["model"]) / (1024 * 1024)
+    return (
+        scene["model"],
+        scene["tracking_img"],
+        f"**{scene['name']}** ({file_size:.1f} MB) ✓ Loaded"
+    )
 
+# --- UI ---
 with gr.Blocks(title="RGS-SLAM Demo") as demo:
     
     gr.Markdown("# 🚀 RGS-SLAM Gaussian Splatting Viewer")
-    gr.Markdown(
-        "**3D Gaussian Splatting** reconstruction results with camera trajectories. "
-        "Select a scene below to load the model."
-    )
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -165,47 +63,24 @@ with gr.Blocks(title="RGS-SLAM Demo") as demo:
                 height=400,
                 object_fit="contain"
             )
-            
-            gr.Markdown("""
-            ### 🎮 Controls
-            - **Rotate**: Left mouse drag
-            - **Zoom**: Mouse wheel
-            - **Pan**: Right mouse drag
-            """)
         
         with gr.Column(scale=3):
-            status = gr.Markdown("**👈 Select a scene to begin (Conversion happens on click)**")
+            status = gr.Markdown("**👈 Select a scene to load**")
             
             with gr.Tab("🎯 3D Reconstruction"):
-                gr.Markdown("### Point Cloud with Colors")
                 model_viewer = gr.Model3D(
-                    label="Gaussian Splat Visualization",
+                    label="3D Model",
                     clear_color=[0.1, 0.1, 0.1, 1.0],
                     height=600
                 )
-                gr.Markdown("*Colors extracted from Gaussian Splatting's spherical harmonics (f_dc)*")
             
             with gr.Tab("📍 Camera Trajectory"):
-                gr.Markdown("### SLAM Tracking Results")
-                trajectory_img = gr.Image(
-                    label="Camera Path",
-                    height=600
-                )
+                trajectory_img = gr.Image(label="Camera Path", height=600)
     
-    # Event binding (直接綁定到有 @spaces.GPU 的函式)
     scene_gallery.select(
-        fn=process_and_load_scene,
+        fn=load_scene,
         outputs=[model_viewer, trajectory_img, status]
     )
-    
-    gr.Markdown("---")
-    gr.Markdown(
-        "**Note:** Gaussian Splatting PLY files contain spherical harmonics coefficients. "
-        "Colors are extracted from the DC component (f_dc_0, f_dc_1, f_dc_2) for visualization."
-    )
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+    demo.launch()