vggt

app.py

@@ -25,15 +25,15 @@ def initialize_backend():
         print(f"🔧 Backend API object type: {type(backend_api)}")
         print(f"🔧 Backend API object attributes: {dir(backend_api)}")
         
-        # gr.load() typically exposes the Interface's fn function directly
-        # So we should look for the main function name, not the wrapper names
-        if hasattr(backend_api, 'process_video_with_points'):
+        # Check if it's a Blocks object and has fns
+        if hasattr(backend_api, 'fns') and backend_api.fns:
+            print(f"🔧 Available functions in fns: {list(backend_api.fns.keys())}")
             BACKEND_AVAILABLE = True
             print("✅ Backend connection successful!")
-            print("✅ Backend API methods are available")
+            print("✅ Backend API functions are available")
             return True
         else:
-            print("❌ Backend API methods not found")
+            print("❌ Backend API functions not found")
             print(f"🔧 Available methods: {[attr for attr in dir(backend_api) if not attr.startswith('_')]}")
             BACKEND_AVAILABLE = False
             return False
@@ -97,19 +97,24 @@ def handle_video_upload(video):
         return None, None, [], 50, 756, 3
     
     try:
-        if BACKEND_AVAILABLE and backend_api:
+        if BACKEND_AVAILABLE and backend_api and hasattr(backend_api, 'fns'):
             # Try to use backend API
             try:
-                # Use the main function directly since gr.load() exposes the Interface's fn
-                result = backend_api.process_video_with_points(video, [], 50, 756, 3)
-                # Parse the result to extract what we need
-                if isinstance(result, dict) and result.get("success"):
-                    # For now, just extract the first frame locally
-                    display_image = extract_first_frame(video)
-                    original_image_state = json.dumps({"video_path": video, "frame": "backend_processing"})
-                    return original_image_state, display_image, [], 50, 756, 3
+                # Use the Blocks fns to call the function
+                if 'process_video_with_points' in backend_api.fns:
+                    result = backend_api.fns['process_video_with_points'](video, [], 50, 756, 3)
+                    # Parse the result to extract what we need
+                    if isinstance(result, dict) and result.get("success"):
+                        # For now, just extract the first frame locally
+                        display_image = extract_first_frame(video)
+                        original_image_state = json.dumps({"video_path": video, "frame": "backend_processing"})
+                        return original_image_state, display_image, [], 50, 756, 3
+                    else:
+                        print("Backend processing failed, using local fallback")
+                        # Fallback to local processing
+                        pass
                 else:
-                    print("Backend processing failed, using local fallback")
+                    print("process_video_with_points function not found in backend")
                     # Fallback to local processing
                     pass
             except Exception as e:
@@ -228,30 +233,35 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
         return None, None
     
     try:
-        if BACKEND_AVAILABLE and backend_api:
+        if BACKEND_AVAILABLE and backend_api and hasattr(backend_api, 'fns'):
             # Try to use backend API
             try:
                 print(f"🔧 Calling backend API with parameters: grid_size={grid_size}, vo_points={vo_points}, fps={fps}")
                 print(f"🔧 Original image state type: {type(original_image_state)}")
                 print(f"🔧 Original image state preview: {str(original_image_state)[:100]}...")
                 
-                # Use the main function with points from the state
-                # For now, we'll use empty points since we're in local mode
-                result = backend_api.process_video_with_points(
-                    None, [], grid_size, vo_points, fps
-                )
-                
-                print(f"✅ Backend API call successful!")
-                print(f"🔧 Result type: {type(result)}")
-                print(f"🔧 Result: {result}")
-                
-                # Parse the result
-                if isinstance(result, dict) and result.get("success"):
-                    viz_html = result.get("viz_html_path", "")
-                    track_video_path = result.get("track_video_path", "")
-                    return viz_html, track_video_path
+                # Use the Blocks fns to call the function
+                if 'process_video_with_points' in backend_api.fns:
+                    # For now, we'll use empty points since we're in local mode
+                    result = backend_api.fns['process_video_with_points'](
+                        None, [], grid_size, vo_points, fps
+                    )
+                    
+                    print(f"✅ Backend API call successful!")
+                    print(f"🔧 Result type: {type(result)}")
+                    print(f"🔧 Result: {result}")
+                    
+                    # Parse the result
+                    if isinstance(result, dict) and result.get("success"):
+                        viz_html = result.get("viz_html_path", "")
+                        track_video_path = result.get("track_video_path", "")
+                        return viz_html, track_video_path
+                    else:
+                        print("Backend processing failed, showing error message")
+                        # Fallback to error message
+                        pass
                 else:
-                    print("Backend processing failed, showing error message")
+                    print("process_video_with_points function not found in backend")
                     # Fallback to error message
                     pass
             except Exception as e:
@@ -279,6 +289,7 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                 <p style='color: #666; font-size: 12px; margin: 0;'>Backend Available: {BACKEND_AVAILABLE}</p>
                 <p style='color: #666; font-size: 12px; margin: 0;'>Backend API Object: {backend_api is not None}</p>
                 <p style='color: #666; font-size: 12px; margin: 0;'>Backend URL: {BACKEND_SPACE_URL}</p>
+                <p style='color: #666; font-size: 12px; margin: 0;'>Available Functions: {list(backend_api.fns.keys()) if backend_api and hasattr(backend_api, 'fns') else 'None'}</p>
             </div>
             <p style='color: #2d3436; font-weight: bold; margin-top: 15px;'>
                 Current Status: Backend unavailable - Running in limited mode