add debug info

oss_utils.py

@@ -25,12 +25,21 @@ def list_oss_files(folder_path: str) -> List[str]:
     """列出OSS文件夹中的所有文件"""
     files = []
     try:
+        print(f"🔍 OSS DEBUG: Listing files with prefix: '{folder_path}'")
+        file_count = 0
         for obj in oss2.ObjectIterator(bucket, prefix=folder_path):
             if not obj.key.endswith('/'):  # 排除目录本身
                 files.append(obj.key)
+                file_count += 1
+                if file_count <= 5:  # 只输出前5个文件用于调试
+                    print(f"🔍 OSS DEBUG: Found file: {obj.key}")
+        
+        print(f"🔍 OSS DEBUG: Total files found: {len(files)}")
         return sorted(files, key=lambda x: os.path.splitext(x)[0]) 
     except Exception as e:
-        print(f"Error listing OSS files: {str(e)}")
+        print(f"❌ OSS DEBUG: Error listing OSS files: {str(e)}")
+        print(f"❌ OSS DEBUG: Exception type: {type(e).__name__}")
+        print(f"❌ OSS DEBUG: folder_path was: '{folder_path}'")
         return []
 
 def download_oss_file(oss_path: str, local_path: str):

simulation.py

@@ -204,22 +204,51 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
     local_image_dir = os.path.join(user_dir, task_id, "final_images")
     os.makedirs(local_image_dir, exist_ok=True)
     
+    # 添加调试信息
+    print(f"🔍 DEBUG: result_folder = {result_folder}")
+    print(f"🔍 DEBUG: task_id = {task_id}")
+    print(f"🔍 DEBUG: image_folder = {image_folder}")
+    print(f"🔍 DEBUG: user_dir = {user_dir}")
+    print(f"🔍 DEBUG: local_image_dir = {local_image_dir}")
+    
     try:
         # 1. 获取OSS上的所有图片文件
+        print(f"🔍 DEBUG: Listing files in OSS folder: {image_folder}")
         oss_files = list_oss_files(image_folder)
+        print(f"🔍 DEBUG: Total files found in OSS: {len(oss_files)}")
+        print(f"🔍 DEBUG: First 10 files: {oss_files[:10]}")
+        
         image_files = [f for f in oss_files if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+        print(f"🔍 DEBUG: Image files found: {len(image_files)}")
+        print(f"🔍 DEBUG: First 5 image files: {image_files[:5]}")
         
         if not image_files:
-            raise gr.Error("No images found in OSS for final video creation")
+            # 如果没有找到图片，尝试直接在result_folder中查找
+            print(f"🔍 DEBUG: No images in {image_folder}, trying {result_folder}")
+            oss_files_direct = list_oss_files(result_folder)
+            print(f"🔍 DEBUG: Files in result_folder: {len(oss_files_direct)}")
+            print(f"🔍 DEBUG: Direct files: {oss_files_direct[:10]}")
+            
+            # 查找所有子目录中的图片
+            all_image_files = [f for f in oss_files_direct if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+            print(f"🔍 DEBUG: All image files in result_folder: {len(all_image_files)}")
+            
+            if all_image_files:
+                image_files = all_image_files
+                print(f"🔍 DEBUG: Using image files from result_folder: {len(image_files)}")
+            else:
+                raise gr.Error("No images found in OSS for final video creation")
         
         # 2. 按文件名排序确保时间顺序正确
         image_files.sort(key=lambda x: os.path.splitext(os.path.basename(x))[0])
+        print(f"🔍 DEBUG: Sorted image files (first 5): {[os.path.basename(f) for f in image_files[:5]]}")
         
         # 3. 下载所有图片到本地
         frames = []
         width, height = 0, 0
         
-        for oss_path in image_files:
+        print(f"🔍 DEBUG: Starting to download {len(image_files)} images...")
+        for i, oss_path in enumerate(image_files):
             try:
                 filename = os.path.basename(oss_path)
                 local_path = os.path.join(local_image_dir, filename)
@@ -230,12 +259,18 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
                 if frame is not None:
                     if width == 0:  # 第一次获取图像尺寸
                         height, width = frame.shape[:2]
+                        print(f"🔍 DEBUG: Image dimensions: {width}x{height}")
                     frames.append(frame)
                     
+                    if (i + 1) % 10 == 0:  # 每10张图片输出一次进度
+                        print(f"🔍 DEBUG: Downloaded {i + 1}/{len(image_files)} images")
+                        
             except Exception as e:
-                print(f"Error processing {oss_path}: {e}")
+                print(f"❌ Error processing {oss_path}: {e}")
                 continue
         
+        print(f"🔍 DEBUG: Successfully loaded {len(frames)} frames")
+        
         if not frames:
             raise gr.Error("No valid images could be processed for final video")
         
@@ -244,6 +279,9 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
         os.makedirs(final_video_dir, exist_ok=True)
         final_video_path = os.path.join(final_video_dir, "final_video.mp4")
         
+        print(f"🔍 DEBUG: Creating video at: {final_video_path}")
+        print(f"🔍 DEBUG: Video parameters: {len(frames)} frames, {fps} fps, {width}x{height}")
+        
         # 使用OpenCV创建视频
         fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         out = cv2.VideoWriter(final_video_path, fourcc, fps, (width, height))
@@ -253,11 +291,15 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
         out.release()
         
         # 5. 转换为H.264格式
+        print(f"🔍 DEBUG: Converting to H.264...")
         h264_video_path = convert_to_h264(final_video_path)
         
-        print(f"Final video created: {h264_video_path} with {len(frames)} frames at {fps} fps")
+        print(f"✅ Final video created: {h264_video_path} with {len(frames)} frames at {fps} fps")
         return h264_video_path
         
     except Exception as e:
-        print(f"Error creating final video from OSS images: {e}")
+        print(f"❌ Error creating final video from OSS images: {e}")
+        print(f"❌ Exception type: {type(e).__name__}")
+        import traceback
+        print(f"❌ Traceback: {traceback.format_exc()}")
         raise gr.Error(f"Failed to create final video: {str(e)}")