Implement visualization directory management in app.py and navigation.py. Added clear_visualization_directory function to prevent users from seeing each other's generated images. Updated IMAGE_PATHS with new sample images and removed unused imports.

app.py

@@ -7,7 +7,7 @@ subprocess.check_call([
 ])
 
 from typing import List, Literal
-from pathlib import Path
+# from pathlib import Path
 from functools import partial
 import spaces
 import gradio as gr
@@ -16,12 +16,14 @@ import torch
 from omegaconf import OmegaConf
 from modeling.pipeline import VMemPipeline
 from diffusers.utils import export_to_video
-from scipy.spatial.transform import Rotation, Slerp
+# from scipy.spatial.transform import Rotation, Slerp
 from navigation import Navigator
-from PIL import Image
-from utils import tensor_to_pil, encode_vae_image, encode_image, get_default_intrinsics, load_img_and_K, transform_img_and_K
+# from PIL import Image
+from utils import tensor_to_pil, get_default_intrinsics, load_img_and_K, transform_img_and_K
 import os
-import glob
+# import glob
+import shutil
+
 
 
 
@@ -46,6 +48,28 @@ IMAGE_PATHS = ['test_samples/oxford.jpg',
                'test_samples/jesus.jpg',]
 
 
+def clear_visualization_directory():
+    """
+    Clear all contents from the visualization directory to prevent users from seeing
+    each other's generated images.
+    """
+    viz_dir = "./visualization"
+    try:
+        if os.path.exists(viz_dir):
+            shutil.rmtree(viz_dir)
+        os.makedirs(viz_dir, exist_ok=True)
+        print(f"Cleared visualization directory: {viz_dir}")
+    except Exception as e:
+        print(f"Warning: Could not clear visualization directory: {e}")
+
+
+IMAGE_PATHS = ['test_samples/oxford.jpg', 
+               'test_samples/open_door.jpg', 
+               'test_samples/living_room.jpg', 
+               'test_samples/arc_de_tromphe.jpeg',
+               'test_samples/changi.jpg', 
+               'test_samples/jesus.jpg',]
+
 # If no images found, create placeholders
 if not IMAGE_PATHS:
     def create_placeholder_images(num_samples=5, height=HEIGHT, width=WIDTH):
@@ -355,6 +379,9 @@ def render_demonstrate(
                             gr.Warning("Please upload an image first")
                             return "Selection", None, None, None
                         try:
+                            # Clear visualization directory to prevent users from seeing each other's generated images
+                            clear_visualization_directory()
+                            
                             # Load image and prepare for navigation
                             result = load_image_for_navigation(image_path)
                             
@@ -416,6 +443,9 @@ def render_demonstrate(
                 
                 def start_navigation(evt: gr.SelectData):
                     try:
+                        # Clear visualization directory to prevent users from seeing each other's generated images
+                        clear_visualization_directory()
+                        
                         # Get the selected image path
                         selected_path = IMAGE_PATHS[evt.index]
                         
@@ -824,7 +854,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="blue")) as demo:
                 )
                 gr.Button(
                     value="Paper",
-                    link="https://arxiv.org/abs/2506.18903",
+                    link="http://arxiv.org/abs/2506.18903",
                     icon="https://simpleicons.org/icons/arxiv.svg",
                     elem_classes=["header-button"],
                     size="md",

navigation.py

@@ -7,7 +7,7 @@ import json
 from typing import List, Optional, Tuple
 import scipy.spatial.transform as spt
 from omegaconf import OmegaConf
-
+import shutil
 
 from modeling.pipeline import VMemPipeline
 from utils import load_img_and_K, transform_img_and_K, get_default_intrinsics
@@ -65,6 +65,11 @@ class Navigator:
             "transform_matrix": initial_pose.tolist() if isinstance(initial_pose, np.ndarray) else initial_pose
         })
         
+        # clean the visualization folder
+        if os.path.exists("visualization"):
+            shutil.rmtree("visualization")
+        os.makedirs("visualization", exist_ok=True)
+        
         return initial_frame
         
     def initialize_pipeline(self, image_tensor, initial_pose, initial_K):
@@ -428,3 +433,4 @@ class Navigator:
     
    
 
+