backend

app.py

@@ -496,7 +496,7 @@ def reset_points(original_img: str, sel_pix):
 def launch_viz(grid_size, vo_points, fps, original_image_state):
     """Launch visualization with user-specific temp directory"""
     if original_image_state is None:
-        return None, None
+        return None, None, None
     
     try:
         if BACKEND_AVAILABLE and backend_client:
@@ -589,8 +589,8 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                     else:
                         print("⚠️ No tracking video content received from backend")
                     
-                    # 返回iframe HTML和视频路径
-                    return iframe_html, local_video_path
+                    # 返回iframe HTML、视频路径和HTML文件路径（用于下载）
+                    return iframe_html, local_video_path, random_path
                 else:
                     error_msg = result.get("error", "Unknown error") if isinstance(result, dict) else "Backend processing failed"
                     print(f"❌ Backend processing failed: {error_msg}")
@@ -663,11 +663,11 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
             </div>
         </div>
         """
-        return error_message, None
+        return error_message, None, None
         
     except Exception as e:
         print(f"Error in launch_viz: {e}")
-        return None, None
+        return None, None, None
 
 def clear_all():
     """Clear all buffers and temporary files"""
@@ -676,6 +676,14 @@ def clear_all():
             gr.update(value=756), 
             gr.update(value=3))
 
+def clear_all_with_download():
+    """Clear all buffers including download component"""
+    return (None, None, [], 
+            gr.update(value=50), 
+            gr.update(value=756), 
+            gr.update(value=3),
+            None)  # HTML download component
+
 def update_tracker_model(model_name):
     """Update tracker model (placeholder function)"""
     return
@@ -1091,6 +1099,13 @@ with gr.Blocks(
                     interactive=False,
                     height=300
                 )
+                
+                # HTML文件下载组件
+                html_download = gr.File(
+                    label="📥 Download 3D Visualization HTML",
+                    interactive=False,
+                    visible=True
+                )
     
     # 3D Visualization - Make it larger and more prominent
     with gr.Row():
@@ -1181,14 +1196,14 @@ with gr.Blocks(
     )
     
     clear_all_btn.click(
-        fn=clear_all,
-        outputs=[video_input, interactive_frame, selected_points, grid_size, vo_points, fps]
+        fn=clear_all_with_download,
+        outputs=[video_input, interactive_frame, selected_points, grid_size, vo_points, fps, html_download]
     )
     
     launch_btn.click(
         fn=launch_viz,
         inputs=[grid_size, vo_points, fps, original_image_state],
-        outputs=[viz_html, tracking_result_video]
+        outputs=[viz_html, tracking_result_video, html_download]
     )
 
     # Acknowledgment section for TAPIR3D - moved to the end