bugfix

app.py

@@ -314,19 +314,20 @@ def generate_tracking_cotracker(video_tensor, density=30):
     # 将结果返回
     return pred_tracks_with_depth.squeeze(0), pred_visibility.squeeze(0)
 
-@spaces.GPU(duration=280)
+@spaces.GPU(duration=350)
 def apply_tracking_unified(video_tensor, tracking_tensor, repaint_img_tensor, prompt, fps):
     """统一的应用跟踪函数"""
-    print("--- Entering apply_tracking_unified ---") # 添加打印语句
-    print(f"Prompt received: {prompt}")             # 添加打印语句
-    print(f"FPS received: {fps}")                   # 添加打印语句
-    print(f"Video tensor shape: {video_tensor.shape if video_tensor is not None else None}") # 添加打印语句
-    print(f"Tracking tensor shape: {tracking_tensor.shape if tracking_tensor is not None else None}") # 添加打印语句
-    print(f"Repaint tensor shape: {repaint_img_tensor.shape if repaint_img_tensor is not None else None}") # 添加打印语句
+    print("--- Entering apply_tracking_unified ---")
+    print(f"Prompt received: {prompt}")
+    print(f"FPS received: {fps}")
+    print(f"Video tensor shape: {video_tensor.shape if video_tensor is not None else None}")
+    print(f"Tracking tensor shape: {tracking_tensor.shape if tracking_tensor is not None else None}")
+    print(f"Repaint tensor shape: {repaint_img_tensor.shape if repaint_img_tensor is not None else None}")
     try:
         if video_tensor is None or tracking_tensor is None:
+            print("Error: Video tensor or tracking tensor is None.")
             return None
-            
+
         das = get_das_pipeline()
         output_path = das.apply_tracking(
             video_tensor=video_tensor,
@@ -335,15 +336,27 @@ def apply_tracking_unified(video_tensor, tracking_tensor, repaint_img_tensor, pr
             img_cond_tensor=repaint_img_tensor,
             prompt=prompt,
             checkpoint_path=DEFAULT_MODEL_PATH,
-            num_inference_steps=15
+            num_inference_steps=25
         )
-        
-        print(f"生成的视频路径: {output_path}")
-        
+
+        print(f"das.apply_tracking returned: {output_path}")
+
+        # --- 临时解决方案开始 ---
+        # 检查 das.apply_tracking 是否返回 None，并尝试使用日志中看到的固定路径
+        potential_fixed_path = os.path.join(project_root, OUTPUT_DIR, "result.mp4") # 构建预期的固定路径
+        print(f"Checking potential fixed path: {potential_fixed_path}")
+
+        if output_path is None and os.path.exists(potential_fixed_path):
+            print(f"Warning: das.apply_tracking returned None, but found file at {potential_fixed_path}. Using this path.")
+            output_path = potential_fixed_path
+        # --- 临时解决方案结束 ---
+
+        print(f"最终使用的视频路径: {output_path}")
+
         # 确保返回的是绝对路径
         if output_path and not os.path.isabs(output_path):
             output_path = os.path.abspath(output_path)
-            
+
         # 检查文件是否存在
         if output_path and os.path.exists(output_path):
             print(f"文件存在，大小: {os.path.getsize(output_path)} 字节")