add debug info

oss_utils.py

@@ -25,6 +25,21 @@ try:
     auth = oss2.Auth(OSS_CONFIG["access_key_id"], OSS_CONFIG["access_key_secret"])
     bucket = oss2.Bucket(auth, OSS_CONFIG["endpoint"], OSS_CONFIG["bucket_name"])
     print(f"✅ OSS client initialized successfully")
+    
+    # 测试OSS连接
+    try:
+        # 尝试列出bucket根目录的文件来测试连接
+        test_files = []
+        for i, obj in enumerate(oss2.ObjectIterator(bucket, max_keys=5)):
+            test_files.append(obj.key)
+            if i >= 4:  # 只获取前5个
+                break
+        print(f"✅ OSS connection test successful, found {len(test_files)} test files")
+        if test_files:
+            print(f"   Sample files: {test_files[:3]}")
+    except Exception as test_e:
+        print(f"⚠️  OSS connection test failed: {test_e}")
+        
 except Exception as e:
     print(f"❌ OSS client initialization failed: {e}")
     bucket = None
@@ -90,6 +105,38 @@ def get_user_tmp_dir(session_hash: str) -> str:
     os.makedirs(user_dir, exist_ok=True)
     return user_dir
 
+def test_oss_access(task_id: str = None):
+    """测试OSS访问并查找特定任务的文件"""
+    if bucket is None:
+        print("❌ Cannot test OSS access - bucket not initialized")
+        return
+    
+    test_paths = [
+        "gradio_demo/",
+        "gradio_demo/tasks/",
+    ]
+    
+    if task_id:
+        test_paths.extend([
+            f"gradio_demo/tasks/{task_id}/",
+            f"gradio_demo/tasks/{task_id}/images/",
+            f"gradio_demo/tasks/{task_id}/image/",
+        ])
+    
+    for path in test_paths:
+        try:
+            print(f"🔍 Testing path: {path}")
+            files = []
+            for i, obj in enumerate(oss2.ObjectIterator(bucket, prefix=path, max_keys=10)):
+                files.append(obj.key)
+                if i >= 9:
+                    break
+            print(f"   Found {len(files)} files")
+            if files:
+                print(f"   Sample: {files[:3]}")
+        except Exception as e:
+            print(f"   ❌ Error: {e}")
+
 def cleanup_user_tmp_dir(session_hash: str):
     """清理用户临时目录"""
     user_dir = os.path.join(TMP_ROOT, str(session_hash))

simulation.py

@@ -8,7 +8,7 @@ import numpy as np
 from typing import List
 import gradio as gr
 from backend_api import get_task_status
-from oss_utils import list_oss_files, download_oss_file, get_user_tmp_dir
+from oss_utils import list_oss_files, download_oss_file, get_user_tmp_dir, test_oss_access
 
 def stream_simulation_results(result_folder: str, task_id: str, request: gr.Request, fps: int = 6):
     """
@@ -205,6 +205,10 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
     elif cleaned_result_folder.startswith('/oss-waic/'):
         cleaned_result_folder = cleaned_result_folder[10:]  # 移除 '/oss-waic/' 前缀
     
+    # 确保路径是正确的格式：gradio_demo/tasks/task_id
+    if not cleaned_result_folder.startswith('gradio_demo/'):
+        cleaned_result_folder = f"gradio_demo/tasks/{task_id}"
+    
     # 获取图片文件夹路径
     image_folder = os.path.join(cleaned_result_folder, "images").replace('\\', '/')
     user_dir = get_user_tmp_dir(request.session_hash)
@@ -220,6 +224,10 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
     print(f"🔍 DEBUG: local_image_dir = {local_image_dir}")
     
     try:
+        # 首先测试OSS访问
+        print(f"🔍 DEBUG: Testing OSS access for task {task_id}")
+        test_oss_access(task_id)
+        
         # 1. 获取OSS上的所有图片文件
         print(f"🔍 DEBUG: Listing files in OSS folder: {image_folder}")
         oss_files = list_oss_files(image_folder)
@@ -245,8 +253,11 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
             alternative_paths = [
                 f"gradio_demo/tasks/{task_id}/images",
                 f"gradio_demo/tasks/{task_id}",
-                f"tasks/{task_id}/images",
-                f"tasks/{task_id}"
+                f"tasks/{task_id}/images", 
+                f"tasks/{task_id}",
+                # 根据您确认的路径格式添加
+                f"gradio_demo/tasks/{task_id}/image",  # 可能是单数形式
+                f"gradio_demo/tasks/{task_id}/imgs",   # 可能是缩写
             ]
             
             for alt_path in alternative_paths: