backend

app.py

@@ -533,8 +533,8 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                     track_video_content = result.get("track_video_content", None)
                     track_video_filename = result.get("track_video_filename", "tracked_video.mp4")
                     
-                    # ✅ 关键修复：直接返回backend的HTML内容，不需要iframe包装
-                    # Backend返回的viz_html已经是完整的HTML，包含了嵌入的base64数据
+                    # ✅ 关键修复：Backend返回的是完整HTML页面，需要保存为文件并用iframe显示
+                    # 不能直接返回HTML内容，否则会覆盖整个Frontend页面
                     print(f"✅ Received HTML content from backend (length: {len(viz_html)})")
                     
                     # 🔧 Debug: 检查HTML内容是否包含关键元素
@@ -552,10 +552,40 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                             print("⚠️ Warning: HTML doesn't contain embedded base64 data!")
                         if not has_canvas:
                             print("⚠️ Warning: HTML doesn't contain canvas element!")
+                        
+                        # 💾 保存HTML到临时文件
+                        temp_viz_dir = "temp_frontend_viz"
+                        os.makedirs(temp_viz_dir, exist_ok=True)
+                        
+                        # 生成唯一文件名
+                        timestamp = str(int(time.time() * 1000))
+                        html_filename = f"viz_{timestamp}.html"
+                        html_path = os.path.join(temp_viz_dir, html_filename)
+                        
+                        # 保存HTML文件
+                        with open(html_path, 'w', encoding='utf-8') as f:
+                            f.write(viz_html)
+                        
+                        print(f"✅ Saved HTML visualization to: {html_path}")
+                        
+                        # 🖼️ 创建iframe HTML来显示保存的文件
+                        iframe_html = f"""
+                        <div style='border: 3px solid #667eea; border-radius: 10px; overflow: hidden; box-shadow: 0 8px 32px rgba(102, 126, 234, 0.3);'>
+                            <iframe id="viz_iframe" src="/gradio_api/file={html_path}" width="100%" height="950px" style="border:none;"></iframe>
+                        </div>
+                        """
+                        
                     else:
                         print("❌ Warning: Empty HTML content received from backend!")
+                        iframe_html = """
+                        <div style='border: 3px solid #ff6b6b; border-radius: 10px; padding: 20px; background-color: #fff5f5; text-align: center;'>
+                            <h3 style='color: #d63031;'>❌ Empty Visualization Data</h3>
+                            <p style='color: #2d3436;'>Backend returned empty HTML content.</p>
+                        </div>
+                        """
                     
                     # If we have base64 encoded video content, save it as a temporary file
+                    local_video_path = None
                     if track_video_content:
                         try:
                             # Create a temporary file for the video
@@ -574,17 +604,14 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                             print(f"✅ Successfully saved tracking video to: {local_video_path}")
                             print(f"🔧 Video file size: {len(video_bytes)} bytes")
                             
-                            # Use the local video path for display
-                            return viz_html, local_video_path
-                            
                         except Exception as e:
                             print(f"❌ Failed to process tracking video: {e}")
-                            # Return HTML only if video processing fails
-                            return viz_html, None
+                            local_video_path = None
                     else:
-                        # No video content, return HTML only
                         print("⚠️ No tracking video content received from backend")
-                        return viz_html, None
+                    
+                    # 返回iframe HTML和视频路径
+                    return iframe_html, local_video_path
                 else:
                     error_msg = result.get("error", "Unknown error") if isinstance(result, dict) else "Backend processing failed"
                     print(f"❌ Backend processing failed: {error_msg}")