vggt

app.py

@@ -117,9 +117,10 @@ def handle_video_upload(video):
             try:
                 print("🔧 Calling backend API for video upload...")
                 
-                # Use call_function method
+                # Use call_function method - function 0 is the main processing function
                 if 0 in backend_api.fns:
-                    print("🔧 Using backend call_function method")
+                    print("🔧 Using backend call_function method (main processing)")
+                    # Call the main processing function with video and empty points
                     result = backend_api.call_function(0, [video, [], 50, 756, 3])
                     
                     print(f"✅ Backend video upload API call successful!")
@@ -174,25 +175,28 @@ def select_point(original_img: str, sel_pix: list, point_type: str, evt: gr.Sele
             try:
                 print(f"🔧 Calling backend select point API: x={evt.index[0]}, y={evt.index[1]}, type={point_type}")
                 
-                # Use call_function method
-                if 1 in backend_api.fns:
-                    print("🔧 Using backend call_function method for point selection")
-                    result = backend_api.call_function(1, [original_img, sel_pix, point_type, evt.index[0], evt.index[1]])
+                # Use call_function method - function 0 is the main processing function
+                if 0 in backend_api.fns:
+                    print("🔧 Using backend call_function method (main processing for point selection)")
+                    # Create points list with the new point
+                    points = [(evt.index[0], evt.index[1], point_type)]
+                    # Call the main processing function with video and points
+                    result = backend_api.call_function(0, [None, points, 50, 756, 3])
                     
                     print(f"✅ Backend select point API call successful!")
                     print(f"🔧 Result type: {type(result)}")
                     
-                    # Parse the result - expect (display_image, new_sel_pix)
-                    if isinstance(result, tuple) and len(result) == 2:
-                        display_image_b64, new_sel_pix = result
-                        display_image = base64_to_image(display_image_b64)
-                        return display_image, new_sel_pix
+                    # Parse the result - expect a dict with success status
+                    if isinstance(result, dict) and result.get("success"):
+                        # For now, use local processing for visualization
+                        # Fallback to local processing
+                        pass
                     else:
-                        print("Backend returned unexpected format, using local fallback")
+                        print("Backend processing failed, using local fallback")
                         # Fallback to local processing
                         pass
                 else:
-                    print("Backend function 1 (select point) not found")
+                    print("Backend function 0 not found")
                     # Fallback to local processing
                     pass
             except Exception as e:
@@ -260,25 +264,26 @@ def reset_points(original_img: str, sel_pix):
             try:
                 print("🔧 Calling backend reset points API")
                 
-                # Use call_function method
-                if 2 in backend_api.fns:
-                    print("🔧 Using backend call_function method for reset")
-                    result = backend_api.call_function(2, [original_img, sel_pix])
+                # Use call_function method - function 0 is the main processing function
+                if 0 in backend_api.fns:
+                    print("🔧 Using backend call_function method (main processing for reset)")
+                    # Call the main processing function with video and empty points
+                    result = backend_api.call_function(0, [None, [], 50, 756, 3])
                     
                     print(f"✅ Backend reset points API call successful!")
                     print(f"🔧 Result type: {type(result)}")
                     
-                    # Parse the result - expect (display_image, new_sel_pix)
-                    if isinstance(result, tuple) and len(result) == 2:
-                        display_image_b64, new_sel_pix = result
-                        display_image = base64_to_image(display_image_b64)
-                        return display_image, new_sel_pix
+                    # Parse the result - expect a dict with success status
+                    if isinstance(result, dict) and result.get("success"):
+                        # For now, use local processing for visualization
+                        # Fallback to local processing
+                        pass
                     else:
-                        print("Backend returned unexpected format, using local fallback")
+                        print("Backend processing failed, using local fallback")
                         # Fallback to local processing
                         pass
                 else:
-                    print("Backend function 2 (reset points) not found")
+                    print("Backend function 0 not found")
                     # Fallback to local processing
                     pass
             except Exception as e:
@@ -320,10 +325,10 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                 print(f"🔧 Original image state type: {type(original_image_state)}")
                 print(f"🔧 Original image state preview: {str(original_image_state)[:100]}...")
                 
-                # Use call_function method
+                # Use call_function method - function 0 is the main processing function
                 if 0 in backend_api.fns:
-                    print("🔧 Using backend call_function method")
-                    # For now, we'll use empty points since we're in local mode
+                    print("🔧 Using backend call_function method (main processing)")
+                    # Call the main processing function with video and empty points
                     result = backend_api.call_function(0, [None, [], grid_size, vo_points, fps])
                     
                     print(f"✅ Backend API call successful!")