backend

app.py

@@ -208,7 +208,10 @@ def extract_first_frame(video_path):
 def handle_video_upload(video):
     """Handle video upload and extract first frame"""
     if video is None:
-        return None, None, [], 50, 756, 3
+        return (None, None, [], 
+                gr.update(value=50), 
+                gr.update(value=756), 
+                gr.update(value=3))
     
     try:
         if BACKEND_AVAILABLE and backend_client:
@@ -249,9 +252,14 @@ def handle_video_upload(video):
                     
                     # Get video settings based on video name
                     video_name = get_video_name(video)
+                    print(f"🎬 Video path: '{video}' -> Video name: '{video_name}'")
                     grid_size_val, vo_points_val, fps_val = get_video_settings(video_name)
+                    print(f"🎬 Video settings for '{video_name}': grid_size={grid_size_val}, vo_points={vo_points_val}, fps={fps_val}")
                     
-                    return original_image_state, display_image, selected_points, grid_size_val, vo_points_val, fps_val
+                    return (original_image_state, display_image, selected_points, 
+                            gr.update(value=grid_size_val), 
+                            gr.update(value=vo_points_val), 
+                            gr.update(value=fps_val))
                 else:
                     print("Backend processing failed, using local fallback")
                     # Fallback to local processing
@@ -300,13 +308,21 @@ def handle_video_upload(video):
         
         # Get video settings
         video_name = get_video_name(video)
+        print(f"🎬 Local fallback - Video path: '{video}' -> Video name: '{video_name}'")
         grid_size_val, vo_points_val, fps_val = get_video_settings(video_name)
+        print(f"🎬 Local fallback - Video settings for '{video_name}': grid_size={grid_size_val}, vo_points={vo_points_val}, fps={fps_val}")
         
-        return original_image_state, display_image, [], grid_size_val, vo_points_val, fps_val
+        return (original_image_state, display_image, [], 
+                gr.update(value=grid_size_val), 
+                gr.update(value=vo_points_val), 
+                gr.update(value=fps_val))
         
     except Exception as e:
         print(f"Error in handle_video_upload: {e}")
-        return None, None, [], 50, 756, 3
+        return (None, None, [], 
+                gr.update(value=50), 
+                gr.update(value=756), 
+                gr.update(value=3))
 
 def select_point(original_img: str, sel_pix: list, point_type: str, evt: gr.SelectData):
     """Handle point selection for SAM"""
@@ -533,56 +549,19 @@ 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显示
-                    # 不能直接返回HTML内容，否则会覆盖整个Frontend页面
-                    print(f"✅ Received HTML content from backend (length: {len(viz_html)})")
+                    # Save HTML to _viz directory (like local version)
+                    viz_dir = './_viz'
+                    os.makedirs(viz_dir, exist_ok=True)
+                    random_path = f'./_viz/_{time.time()}.html'
                     
-                    # 🔧 Debug: 检查HTML内容是否包含关键元素
-                    if viz_html:
-                        has_base64 = "window.embeddedBase64" in viz_html
-                        has_canvas = "<canvas" in viz_html
-                        has_script = "<script" in viz_html
-                        print(f"🔧 HTML Analysis:")
-                        print(f"   - Contains base64 data: {has_base64}")
-                        print(f"   - Contains canvas element: {has_canvas}")
-                        print(f"   - Contains script tags: {has_script}")
-                        print(f"   - HTML preview: {viz_html[:200]}...")
-                        
-                        if not has_base64:
-                            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 = "_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>
-                        """
+                    with open(random_path, 'w', encoding='utf-8') as f:
+                        f.write(viz_html)
+                    
+                    # Create iframe HTML to display the saved file
+                    iframe_html = f'<iframe id="viz_iframe" src="{random_path}" width="100%" height="600px" frameborder="0"></iframe>'
+                    
+                    print(f"💾 HTML saved to: {random_path}")
+                    print(f"📊 HTML content preview: {viz_html[:200]}...")
                     
                     # If we have base64 encoded video content, save it as a temporary file
                     local_video_path = None
@@ -692,7 +671,10 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
 
 def clear_all():
     """Clear all buffers and temporary files"""
-    return None, None, [], 50, 756, 3
+    return (None, None, [], 
+            gr.update(value=50), 
+            gr.update(value=756), 
+            gr.update(value=3))
 
 def update_tracker_model(model_name):
     """Update tracker model (placeholder function)"""