Update index.html

index.html

@@ -688,6 +688,7 @@
   let inputImageBase64 = null;
   let inputImageResolution = null;
   let currentGeneratedSplat = null; // 跟踪当前生成的场景
+  let currentDownloadedBlob = null; // 保存已下载的文件blob
 
   // UI Elements
   const loadingElement = document.getElementById('loading');
@@ -882,6 +883,10 @@
 
           const blob = new Blob(chunks);
           const url = URL.createObjectURL(blob);
+          
+          // 保存blob供后续下载使用
+          currentDownloadedBlob = blob;
+          
           // Continue to load the splat
           updateStatus('Loading generated scene...', cameraParams.length);
           
@@ -891,6 +896,11 @@
           updateStatus('Scene generated successfully!', cameraParams.length);
           // Show generation time and total file size (MB)
           showGenerationInfo(latestGenerationTime || 0, total || blob.size);
+          
+          // 启用下载按钮
+          if (window.downloadController) {
+            window.downloadController.enable();
+          }
           // Notify backend to delete the server file after client has downloaded it
           try {
             if (info.file_id) {
@@ -971,7 +981,7 @@
   // GUI Options - declare early
   const guiOptions = {
     // 后端地址，默认为本页面ip
-    BackendAddress: `https://imlixinyang-flashworld-demo.hf.space`,
+    BackendAddress: `${window.location.protocol}//${window.location.hostname}:7860`,
     HF_TOKEN: "",
     FOV: 60,
     LoadFromJson: () => { 
@@ -1130,6 +1140,27 @@
     // Playback scrub value (0..1)
     playbackT: 0,
     
+    downloadGeneratedFile: () => {
+      if (!currentDownloadedBlob) {
+        updateStatus('No generated file available to download', cameraParams.length);
+        return;
+      }
+      
+      updateStatus(`Downloading SPZ file (${(currentDownloadedBlob.size / 1024 / 1024).toFixed(2)} MB)...`, cameraParams.length);
+      
+      // 直接使用已保存的blob创建下载链接
+      const url = URL.createObjectURL(currentDownloadedBlob);
+      const a = document.createElement('a');
+      a.href = url;
+      a.download = `${Date.now()}.spz`;
+      document.body.appendChild(a);
+      a.click();
+      document.body.removeChild(a);
+      URL.revokeObjectURL(url);
+      
+      updateStatus('SPZ file downloaded successfully!', cameraParams.length);
+    },
+    
     generate: () => {
       // 检查是否有足够的相机
       if (cameraParams.length < 2) {
@@ -1147,6 +1178,14 @@
         console.log('Previous generated scene removed');
       }
       
+      // 清除之前的下载文件
+      currentDownloadedBlob = null;
+      
+      // 禁用下载按钮
+      if (window.downloadController) {
+        window.downloadController.disable();
+      }
+      
       // 初始化进度条信息
       const generationTimeElement = document.getElementById('generation-time');
       const fileSizeElement = document.getElementById('file-size');
@@ -1295,6 +1334,9 @@
                 });
               }
               return pump().then(blob => {
+                // 保存blob供后续下载使用
+                currentDownloadedBlob = blob;
+                
                 const url = URL.createObjectURL(blob);
             return { url, __deleteAfterDownloadFileId: (typeof responseData !== 'undefined' ? responseData.file_id : null) };
               });
@@ -1313,6 +1355,12 @@
           currentGeneratedSplat = GeneratedSplat;
           console.log('3D scene loaded successfully!');
           updateStatus('Scene generated successfully!', cameraParams.length);
+          
+          // 启用下载按钮
+          if (window.downloadController) {
+            window.downloadController.enable();
+          }
+          
           hideDownloadProgress();
           showLoading(false);
 
@@ -2054,8 +2102,13 @@
       // Step 4: Generate Your Scene
       const step4Folder = gui.addFolder('4. Generate Scene');
       step4Folder.add(guiOptions, "generate").name("Generate!");
+      const downloadController = step4Folder.add(guiOptions, "downloadGeneratedFile").name("Download SPZ File");
+      downloadController.disable(); // Initially disabled
       step4Folder.open();
       
+      // Store download controller globally for enabling/disabling
+      window.downloadController = downloadController;
+      
       // Step 5: Trajectory Playback (Scrubber)
       const step5Folder = gui.addFolder('5. Trajectory Playback');
       step5Folder.add(guiOptions, 'playbackT', 0, 1, 0.001).name('Scrub (0-1)').onChange((value) => {