vggt

app.py

@@ -20,9 +20,18 @@ def initialize_backend():
     try:
         print(f"Attempting to connect to backend: {BACKEND_SPACE_URL}")
         backend_api = gr.load(f"spaces/{BACKEND_SPACE_URL}", token=hf_token)
-        BACKEND_AVAILABLE = True
-        print("✅ Backend connection successful!")
-        return True
+        
+        # Test if the API object has the expected methods
+        if hasattr(backend_api, 'upload_video_api') and hasattr(backend_api, 'run_tracker_api'):
+            BACKEND_AVAILABLE = True
+            print("✅ Backend connection successful!")
+            print("✅ Backend API methods are available")
+            return True
+        else:
+            print("❌ Backend API methods not found")
+            BACKEND_AVAILABLE = False
+            return False
+            
     except Exception as e:
         print(f"❌ Backend connection failed: {e}")
         print("⚠️  Running in standalone mode (some features may be limited)")
@@ -207,17 +216,28 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
         if BACKEND_AVAILABLE and backend_api:
             # 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]}...")
+                
                 viz_iframe_html, track_video_path = backend_api.run_tracker_api(
                     grid_size, vo_points, fps, original_image_state
                 )
+                
+                print(f"✅ Backend API call successful!")
+                print(f"🔧 Viz HTML type: {type(viz_iframe_html)}")
+                print(f"🔧 Track video path: {track_video_path}")
+                
                 return viz_iframe_html, track_video_path
             except Exception as e:
-                print(f"Backend API call failed: {e}")
+                print(f"❌ Backend API call failed: {e}")
+                print(f"🔧 Error type: {type(e)}")
+                print(f"🔧 Error details: {str(e)}")
                 # Fallback to local processing
                 pass
         
         # Fallback: show message that backend is required
-        error_message = """
+        error_message = f"""
         <div style='border: 3px solid #ff6b6b; border-radius: 10px; padding: 20px; background-color: #fff5f5;'>
             <h3 style='color: #d63031; margin-bottom: 15px;'>⚠️ Backend Connection Required</h3>
             <p style='color: #2d3436; line-height: 1.6;'>
@@ -229,6 +249,12 @@ def launch_viz(grid_size, vo_points, fps, original_image_state):
                 <li>The BACKEND_SPACE_URL is correctly configured</li>
                 <li>You have proper access permissions to the backend Space</li>
             </ul>
+            <div style='background-color: #f8f9fa; border-radius: 5px; padding: 10px; margin-top: 10px;'>
+                <p style='color: #2d3436; font-weight: bold; margin: 0 0 5px 0;'>Debug Information:</p>
+                <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>
+            </div>
             <p style='color: #2d3436; font-weight: bold; margin-top: 15px;'>
                 Current Status: Backend unavailable - Running in limited mode
             </p>
@@ -258,10 +284,60 @@ def handle_video_change(video):
     
     return original_image_state, display_image, selected_points, grid_size_val, vo_points_val, fps_val
 
+def test_backend_connection():
+    """Test if backend is actually working"""
+    global BACKEND_AVAILABLE
+    if not backend_api:
+        return False
+    
+    try:
+        # Try a simple API call to test connection
+        print("Testing backend connection with a simple call...")
+        # We'll test with a dummy call or check if the API object is properly loaded
+        if hasattr(backend_api, 'upload_video_api'):
+            print("✅ Backend API methods are available")
+            return True
+        else:
+            print("❌ Backend API methods not found")
+            BACKEND_AVAILABLE = False
+            return False
+    except Exception as e:
+        print(f"❌ Backend connection test failed: {e}")
+        BACKEND_AVAILABLE = False
+        return False
+
+def test_backend_api():
+    """Test specific backend API functions"""
+    if not BACKEND_AVAILABLE or not backend_api:
+        print("❌ Backend not available for testing")
+        return False
+    
+    try:
+        print("🧪 Testing backend API functions...")
+        
+        # Test if methods exist
+        methods_to_test = ['upload_video_api', 'select_point_api', 'reset_points_api', 'run_tracker_api']
+        for method in methods_to_test:
+            if hasattr(backend_api, method):
+                print(f"✅ {method} is available")
+            else:
+                print(f"❌ {method} is not available")
+        
+        return True
+    except Exception as e:
+        print(f"❌ Backend API test failed: {e}")
+        return False
+
 # Initialize backend connection
 print("🔧 Initializing backend connection...")
 initialize_backend()
 
+# Test the connection
+test_backend_connection()
+
+# Test specific API functions
+test_backend_api()
+
 # Build UI
 with gr.Blocks(css="""
     #advanced_settings .wrap {
@@ -321,7 +397,7 @@ with gr.Blocks(css="""
     original_image_state = gr.State()  # Store original image in state
     
     with gr.Row():
-        # Show backend status
+        # Show backend status with more detailed information
         status_color = "#28a745" if BACKEND_AVAILABLE else "#dc3545"
         status_text = "Connected" if BACKEND_AVAILABLE else "Disconnected"
         status_icon = "✅" if BACKEND_AVAILABLE else "❌"
@@ -343,6 +419,9 @@ with gr.Blocks(css="""
             <p style='font-size: 14px; color: #666; margin: 5px 0 0 0;'>
                 {BACKEND_SPACE_URL}
             </p>
+            <p style='font-size: 12px; color: #888; margin: 5px 0 0 0;'>
+                {'API methods available' if BACKEND_AVAILABLE else 'Connection failed - using local mode'}
+            </p>
         </div>
         </div>
         """)
@@ -399,10 +478,14 @@ with gr.Blocks(css="""
                 examples_per_page=20  # Show all examples on one page to enable scrolling
             )
             
-            # Initialize with the template interface showing "Interactive 3D Tracking"
+            # Initialize with a placeholder interface instead of static file
             viz_iframe = gr.HTML("""
-                                <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=_viz/viz_template.html" width="100%" height="950px" style="border:none;"></iframe>
+                                <div style='border: 3px solid #667eea; border-radius: 10px; overflow: hidden; box-shadow: 0 8px 32px rgba(102, 126, 234, 0.3); background: #f8f9fa; display: flex; align-items: center; justify-content: center; height: 950px;'>
+                                    <div style='text-align: center; color: #666;'>
+                                        <h3 style='margin-bottom: 20px; color: #667eea;'>🎮 Interactive 3D Tracking</h3>
+                                        <p style='font-size: 16px; margin-bottom: 10px;'>Upload a video and select points to start tracking</p>
+                                        <p style='font-size: 14px; color: #999;'>Powered by SpaTrackV2</p>
+                                    </div>
                                 </div>
                                 """)