create user session folder

app.py

@@ -10,9 +10,14 @@ from typing import Optional, List
 import numpy as np
 from datetime import datetime, timedelta
 from collections import defaultdict
+import shutil
 
 # os.environ["SPACES_QUEUE_ENABLED"] = "true"
 
+from app_utils import (
+    TMP_ROOT,
+)
+
 # 后端API配置（可配置化）
 BACKEND_URL = os.getenv("BACKEND_URL", "http://47.95.6.204:51001/")
 API_ENDPOINTS = {
@@ -92,6 +97,13 @@ def log_submission(scene: str, prompt: str, model: str, max_step: int, user: str
     with open(SUBMISSION_LOG, "a") as f:
         f.write(json.dumps(log_entry) + "\n")
 
+# 记录访问
+def record_access(request: gr.Request):
+    user_ip = request.client.host if request else "unknown"
+    user_agent = request.headers.get("user-agent", "unknown")
+    log_access(user_ip, user_agent)
+    return update_log_display()
+
 def read_logs(log_type: str = "all", max_entries: int = 50) -> list:
     """读取日志文件"""
     logs = []
@@ -227,10 +239,13 @@ def stream_simulation_results(result_folder: str, task_id: str, fps: int = 30):
     if max_time <= 0:
         raise gr.Error("timeout 240s")
 
-def create_video_segment(frames: List[np.ndarray], fps: int, width: int, height: int) -> str:
+def create_video_segment(frames: List[np.ndarray], fps: int, width: int, height: int, req: gr.Request) -> str:
     """创建视频片段"""
-    os.makedirs("/opt/gradio_demo/tasks/video_chunk", exist_ok=True)
-    segment_name = f"/opt/gradio_demo/tasks/video_chunk/output_{uuid.uuid4()}.mp4"
+    user_dir = os.path.join(TMP_ROOT, str(req.session_hash))
+    os.makedirs(user_dir, exist_ok=True)
+    video_chunk_path = os.path.join(user_dir, "tasks/video_chunk")
+    os.makedirs(video_chunk_path, exist_ok=True)
+    segment_name = os.path.join(video_chunk_path, f"output_{uuid.uuid4()}.mp4")
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(segment_name, fourcc, fps, (width, height))
     
@@ -371,7 +386,7 @@ def run_simulation(
     max_step: int,
     history: list,
     request: gr.Request
-) -> dict:
+):
     """运行仿真并更新历史记录"""
     # 获取当前时间
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -619,6 +634,16 @@ custom_css = """
 }
 """
 
+def start_session(req: gr.Request):
+    user_dir = os.path.join(TMP_ROOT, str(req.session_hash))
+    os.makedirs(user_dir, exist_ok=True)
+    
+    
+def end_session(req: gr.Request):
+    user_dir = os.path.join(TMP_ROOT, str(req.session_hash))
+    shutil.rmtree(user_dir)
+
+
 # 创建Gradio界面
 with gr.Blocks(title="InternManip Model Inference Demo", css=custom_css) as demo:
     gr.HTML(header_html)
@@ -753,30 +778,23 @@ with gr.Blocks(title="InternManip Model Inference Demo", css=custom_css) as demo
     
     # 初始化场景描述和日志
     demo.load(
+        start_session
+    ).then(
         fn=lambda: update_scene_display("scene_1"),
         outputs=[scene_description, scene_preview]
     ).then(
-        fn=update_log_display,
-        outputs=logs_display
-    )
-    
-    # 记录访问
-    def record_access(request: gr.Request):
-        user_ip = request.client.host if request else "unknown"
-        user_agent = request.headers.get("user-agent", "unknown")
-        log_access(user_ip, user_agent)
-        return update_log_display()
-    
-    demo.load(
         fn=record_access,
         inputs=None,
         outputs=logs_display,
         queue=False
+    ).then(
+        fn=update_log_display,
+        outputs=logs_display
     )
     
     demo.queue(default_concurrency_limit=8)
 
-    demo.unload(fn=cleanup_session)
+    demo.unload(fn=cleanup_session).then(end_session)
 
 
 # 启动应用

app_utils.py

@@ -0,0 +1,6 @@
+import gradio as gr
+import os
+
+
+TMP_ROOT = os.path.join(os.path.dirname(os.path.abspath(__file__)), "tmp")
+os.makedirs(TMP_ROOT, exist_ok=True)