add debug info

app.py

@@ -6,7 +6,7 @@ from backend_api import submit_to_backend, get_task_status, get_task_result
 from logging_utils import log_access, log_submission, is_request_allowed
 from simulation import stream_simulation_results, convert_to_h264, create_final_video_from_oss_images
 from ui_components import update_history_display, update_scene_display, update_log_display, get_scene_instruction
-from oss_utils import download_oss_file, get_user_tmp_dir, cleanup_user_tmp_dir, oss_file_exists
+from oss_utils import download_oss_file, get_user_tmp_dir, cleanup_user_tmp_dir, oss_file_exists, clean_oss_result_path
 import os
 from datetime import datetime
 
@@ -37,7 +37,7 @@ def run_simulation(scene, model, mode, prompt, history, request: gr.Request):
         time.sleep(5)
         status = get_task_status(task_id)
         # OSS上的结果文件夹路径，不再检查本地路径是否存在
-        result_folder = status.get("result", f"gradio_demo/tasks/{task_id}")
+        result_folder = clean_oss_result_path(status.get("result", f"gradio_demo/tasks/{task_id}"))
         
     except Exception as e:
         log_submission(scene, prompt, model, user_ip, str(e))

oss_utils.py

@@ -105,6 +105,20 @@ def get_user_tmp_dir(session_hash: str) -> str:
     os.makedirs(user_dir, exist_ok=True)
     return user_dir
 
+def clean_oss_result_path(result_folder: str, task_id: str) -> str:
+    """统一的OSS结果路径清理函数"""
+    cleaned_result_folder = result_folder.strip('/')
+    if cleaned_result_folder.startswith('oss-waic/'):
+        cleaned_result_folder = cleaned_result_folder[9:]  # 移除 'oss-waic/' 前缀
+    elif cleaned_result_folder.startswith('/oss-waic/'):
+        cleaned_result_folder = cleaned_result_folder[10:]  # 移除 '/oss-waic/' 前缀
+    
+    # 确保路径格式正确
+    if not cleaned_result_folder.startswith('gradio_demo/'):
+        cleaned_result_folder = f"gradio_demo/tasks/{task_id}"
+    
+    return cleaned_result_folder
+
 def test_oss_access(task_id: str = None):
     """测试OSS访问并查找特定任务的文件"""
     if bucket is None:

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, test_oss_access
+from oss_utils import list_oss_files, download_oss_file, get_user_tmp_dir, test_oss_access, clean_oss_result_path
 
 def stream_simulation_results(result_folder: str, task_id: str, request: gr.Request, fps: int = 6):
     """
@@ -198,16 +198,8 @@ def create_final_video_from_oss_images(result_folder: str, task_id: str, request
     返回:
         最终视频文件路径
     """
-    # 清理result_folder路径，移除OSS前缀
-    cleaned_result_folder = result_folder.strip('/')
-    if cleaned_result_folder.startswith('oss-waic/'):
-        cleaned_result_folder = cleaned_result_folder[9:]  # 移除 'oss-waic/' 前缀
-    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}"
+    # 使用统一的路径清理函数
+    cleaned_result_folder = clean_oss_result_path(result_folder)
     
     # 获取图片文件夹路径
     image_folder = os.path.join(cleaned_result_folder, "images").replace('\\', '/')